题目描述
给出基数为 -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 组成的数组。
示例 1:
输入:arr1 = [1,1,1,1,1], arr2 = [1,0,1]
输出:[1,0,0,0,0]
解释:arr1 表示 11,arr2 表示 5,输出表示 16 。
示例 2:
输入:arr1 = [0], arr2 = [0]
输出:[0]
示例 3:
输入:arr1 = [0], arr2 = [1]
输出:[1]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/adding-two-negabinary-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
从尾到头遍历2个数组,k = arr1[i] + arr2[j] + jinwei;
如果k>=2,将产生进位,前面的一项为(-2)^ i 将会被抵消变成(-2)^(i-1),所以应该进-1;
如果k==-1,则只能将当前位置位1,然后向前面的一位借,由(-2) ^ i变成
(-2) ^ (i+1),进位=1;
代码
class Solution {
public int[] addNegabinary(int[] arr1, int[] arr2) {
List<Integer> list = new ArrayList<>();
int m = arr1.length - 1;
int n = arr2.length - 1;
int jinwei = 0;
while (m >= 0 || n >= 0 || jinwei!=0) {
int a = m >= 0 ? arr1[m] : 0;
int b = n >= 0 ? arr2[n] : 0;
int k = a + b + jinwei;
if (k >= 2) {
jinwei = -1;
list.add(k - 2);
} else if (k == -1) {
list.add(1);
jinwei = 1;
} else {
list.add(k);
jinwei=0;
}
m--;
n--;
}
Collections.reverse(list);
for(int i=0;i<list.size()-1;i++){
if(list.get(i)==0){
list.remove(i);
i=i-1;
}else{
break;
}
}
return list.stream().mapToInt(Integer::intValue).toArray();
}
}