Leetcode/AddBinary.cpp at master · Jessicahust/Leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
#include<iostream>
#include<string>
#include<vector>
using namespace std;

class Solution
{
public:
    string addBinary(string a, string b)
    {
        int size1=a.size()-1;
        int size2=b.size()-1;
        int size=max(size1,size2);
        string digits(size+1,'0');
        while(size1>=0&&size2>=0)
        {
            if((a[size1]=='1')&&(b[size2]=='1'))
            {
                digits[size-1]='1';
                if(size1==0&&size2==0)
                    digits.insert(digits.begin(),'1');
                size1--;
                size2--;
                size--;
            }
            else if(((a[size1]=='1')||(b[size2]=='1'))&&(digits[size]=='0'))
            {
                digits[size]='1';
                size1--;
                size2--;
                size--;
            }
            else if(((a[size1]=='1')||(b[size2]=='1'))&&(digits[size]=='1'))
            {
                digits[size]='0';
                digits[size-1]='1';
                if(size1==0&&size2==0)
                    digits.insert(digits.begin(),'1');
                size1--;
                size2--;
                size--;
            }
            else
            {
                digits[size]=digits[size];
                size1--;
                size2--;
                size--;
            }
        }
        cout<<digits<<endl;
        if(size1>=0)
        {
            while(size1>=0&&size>=0)
            {
                if((digits[size]=='1')&&(a[size1]=='1'))
                {
                    digits[size]='0';
                    digits[size-1]='1';
                    if(size==0)
                        digits.insert(digits.begin(),'1');
                    size1--;
                    size--;
                }
                else if((digits[size]=='1')&&(a[size1]=='0'))
                {
                    size1--;
                    size--;
                }
                else
                {
                    digits[size]=a[size1];
                    size--;
                    size1--;
                }
            }
        }
        if(size2>=0)
        {
            while(size2>=0&&size>=0)
            {
                if((digits[size]=='1')&&(b[size2]=='1'))
                {
                    digits[size]='0';
                    digits[size-1]='1';
                    //最前面两个数相加需要进位的时候，在前面插入1
                    if(size==0)
                        digits.insert(digits.begin(),'1');
                    size2--;
                    size--;
                }
                else if((digits[size]=='1')&&(b[size2]=='0'))
                {
                    size2--;
                    size--;
                }
                else
                {
                    digits[size]=b[size2];
                    size--;
                    size2--;
                }
            }
        }
        return digits;
    }
};
int main()
{
    string b="100";
    string a="110010";
    Solution s;
    cout<<"digits: "<<s.addBinary(a,b)<<endl;
}