java 自习室 day 25

leetcode 1073题 负二进制数相加

负二进制数相加

给出基数为 -2 的两个数 arr1 和 arr2，返回两数相加的结果。

数字以 数组形式 给出：数组由若干 0 和 1 组成，按最高有效位到最低有效位的顺序排列。例如，arr = [1,1,0,1] 表示数字 (-2)^3 + (-2)^2 + (-2)^0 = -3。数组形式 的数字也同样不含前导零：以 arr 为例，这意味着要么 arr == [0]，要么 arr[0] == 1。

返回相同表示形式的 arr1 和 arr2 相加的结果。两数的表示形式为：不含前导零、由若干 0 和 1 组成的数组。

示例：

输入：arr1 = [1,1,1,1,1], arr2 = [1,0,1]
输出：[1,0,0,0,0]
解释：arr1 表示 11，arr2 表示 5，输出表示 16 。

提示：

1 <= arr1.length <= 1000
1 <= arr2.length <= 1000
arr1 和 arr2 都不含前导零
arr1[i] 为 0 或 1
arr2[i] 为 0 或 1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/adding-two-negabinary-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

在加法或者减法中，借位可以看成-1的进位

两个M进制数相加（M>0）

当对应位结果小于0或者大于M时就需要进位
以carry表示进位，remain表示最后当前位的结果，temp表示当前位两个数与上一位进位相加的结果
temp = remain + carry * M
remain = temp % M
carry = temp / M
当temp大于M时，直接往前进位
当temp小于0时，此时remain小于0，由于M进制数不能出现小于0的数字，因此将remain加上M，此时为了保证temp不变，carry需要相应减1

解法

由于题目要求的是进制数的数字只能为0，1，M为-2，因此当remain小于0时，需要将remain加2，相应的carry也加一

import java.util.Stack;
class Solution {
    public int[] addNegabinary(int[] arr1, int[] arr2) {
        if (arr1[0] == 0) {
            return arr2;
        }
        if (arr2[0] == 0) {
            return arr1;
        }
        int i = arr1.length-1;
        int j = arr2.length-1;
        Stack<Integer> result = new Stack<>();
        int remain = 0;
        int carry = 0;
        while(i>=0 | j>=0 | carry != 0){
            carry += (i<0?0:arr1[i--]) + (j<0?0:arr2[j--]);
            remain = carry % (-2);
            carry /= -2;
            if (remain < 0){
                remain += 2;
                carry++;
            }
            result.push(remain);
        }
        //去除多余的0
        while (result.peek() == 0){
            result.pop();
            if(result.isEmpty()){
                return new int [1];
            }
        }
        int[] resultArr = new int[result.size()];
        for (int k = 0; k < resultArr.length; k++) {
            resultArr[k] = result.pop();
        }
        return resultArr;
    }
}

执行结果：通过
执行用时 :9 ms, 在所有 Java 提交中击败了27.00%的用户
内存消耗 :39.6 MB, 在所有 Java 提交中击败了100.00%的用户