leetcode 5078.負の二進数が追加される（二Negabinary番号を追加します）

件名の説明：

基数2及びARR1 ARR2の二つの与えられた、2つの数値の結果を返します。

アレイの形で与えられる数値：最下位ビット配列に最上位ビットの順に0と1の数によって配列。たとえば、arr = [1,1,0,1]数を表します(-2)^3 + (-2)^2 + (-2)^0 = -3。また、先行ゼロなしでデジタルの配列：をarr、たとえば、これは、いずれかの意味arr == [0]又はarr[0] == 1。

同じを返し、ARR1 ARR2の形で結果が追加表示されます。2つの数の表現である：先行ゼロなし、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;
    }
};