标签：Adding digit idx res Two 5078 arr2 arr1 carry

目录

• 题目描述：
• 示例：
• 解法：

题目描述：

给出基数为 -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

解法：

class Solution {
public:
    vector<int> addNegabinary(vector<int>& arr1, vector<int>& arr2) {
        //   10101
        //     101
        // 1101110
        int sz1 = arr1.size();
        int sz2 = arr2.size();
        int max_sz = max(sz1, sz2);
        vector<int> res(max_sz+4, 0);
        int i = sz1-1, j = sz2-1, idx = max_sz+3;
        int carry = 0;
        while(i >= 0 && j >= 0){
            int digit = carry + arr1[i] + arr2[j];
            if(digit >= 2){
                carry = -1;
                digit -= 2;
            }else if(digit == -1){
                digit = 1;
                carry = 1;
            }else{
                carry = 0;
            }
            res[idx] = digit;
            idx--;
            i--;
            j--;
        }
        while(i >= 0){
            int digit = carry + arr1[i];
            if(digit >= 2){
                carry = -1;
                digit -= 2;
            }else if(digit == -1){
                digit = 1;
                carry = 1;
            }else{
                carry = 0;
            }
            res[idx] = digit;
            idx--;
            i--;
        }
        
        while(j >= 0){
            int digit = carry + arr2[j];
            if(digit >= 2){
                carry = -1;
                digit -= 2;
            }else if(digit == -1){
                digit = 1;
                carry = 1;
            }else{
                carry = 0;
            }
            res[idx] = digit;
            idx--;
            j--;
        }
        assert(carry == 0 || carry == 1 || carry == -1);
        if(carry == -1){
            res[idx--] = 1;
            res[idx--] = 1;
        }else if(carry == 1){
            res[idx--] = 1;
        }
        
        while(!res.empty() && res.front() == 0){
            res.erase(res.begin());
        }
        if(res.empty()){
            res.push_back(0);
        }
        
        return res;
    }
};

标签：Adding,digit,idx,res,Two,5078,arr2,arr1,carry
来源： https://www.cnblogs.com/zhanzq/p/10973933.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。