1. 题目
给出基数为 -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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
0+0 = 0
1+0 = 1
0+1 = 1
1+1 = 1*(-2)0+1*(-2)0= 2 = 1*(-2)2+1*(-2)1+0*(-2)0= (1 1 0)-2
11+1 = 1*(-2)1+1*(-2)0 + 1*(-2)0= 0
class Solution {
public:
vector<int> addNegabinary(vector<int>& arr1, vector<int>& arr2) {
int i = arr1.size()-1, j = arr2.size()-1, carry = 0;
vector<int> ans;
while(i>=0 || j>=0 || carry)
{
if(i >= 0)
carry += arr1[i--];
if(j >= 0)
carry += arr2[j--];
ans.push_back((carry+2)%(-2));
//carry 0, push_back 0
//carry 1, push_back 1
//carry 2, push_back 0
//carry -1, push_back 1
if(carry == -1)
carry = 1;
else
carry /= -2;
}
while(ans.size()>1 && ans.back() == 0)
ans.pop_back();//去除前导零
reverse(ans.begin(),ans.end());
return ans;
}
};