LeetCode-2. 兩數相加(連結串列+大數加法模擬)

kewlgrl發表於2018-04-30

2. 兩數相加


給定兩個非空連結串列來表示兩個非負整數。位數按照逆序方式儲存,它們的每個節點只儲存單個數字。將兩數相加返回一個新的連結串列。

你可以假設除了數字 0 之外,這兩個數字都不會以零開頭。

示例:

輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807

C



#include<bits/stdc++.h>
using namespace std;
//Definition for singly-linked list.
struct ListNode
{
    int val;
    struct ListNode *next;
};
/********************提交程式碼********************/
struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2)
{
    int i,j=0,ca=0,cb=0,cnt=0;
    int *a=(int*)malloc(10000*sizeof(int));
    int *b=(int*)malloc(10000*sizeof(int));
    int *c=(int*)malloc(10000*sizeof(int));
    int *res=(int*)malloc(10000*sizeof(int));
    struct ListNode *ans;
    struct ListNode *p=l1;
    struct ListNode *q=l2;
    while(p)//將連結串列中的數存入陣列
    {
        a[ca++]=p->val;
        p=p->next;
    }
    while(q)
    {
        b[cb++]=q->val;
        q=q->next;
    }
    bool flag=false;
    if(ca>cb)//第一個串比第二個串位數多
    {
        for(i=0; i<cb; ++i)//處理前面相同位數下的相加
        {
            int res=a[i]+b[i];
            if(flag)//有進位要加
            {
                ++res;
                flag=false;
            }
            if(res>=10)//需要進位
            {
                flag=true;
                c[cnt++]=res%10;
            }
            else
                c[cnt++]=res%10;
        }
        j=cnt;
        for(i=cb; i<ca; ++i)//處理多餘位數
            c[cnt++]=a[i];
        while(flag)//處理相同位數下的進位
        {
            if(j==ca-1)
            {
                ++c[j];
                break;
            }
            ++c[j];
            flag=false;
            if(c[j]>=10)
            {
                c[j]%=10;
                flag=true;
                ++j;
            }
        }
    }
    else if(ca<cb)//第一個串比第二個串位數少
    {
        for(i=0; i<ca; ++i)
        {
            int res=a[i]+b[i];
            if(flag)
            {
                ++res;
                flag=false;
            }
            if(res>=10)
            {
                flag=true;
                c[cnt++]=res%10;
            }
            else
                c[cnt++]=res%10;
        }
        j=cnt;
        for(i=ca; i<cb; ++i)
            c[cnt++]=b[i];
        while(flag)
        {
            if(j==cb-1)
            {
                ++c[j];
                break;
            }
            ++c[j];
            flag=false;
            if(c[j]>=10)
            {
                c[j]%=10;
                flag=true;
                ++j;
            }
        }
    }
    else//相等位數
    {
        for(i=0; i<ca; ++i)
        {
            int res=a[i]+b[i];
            if(flag)
            {
                ++res;
                flag=false;
            }
            if(res>=10)
            {
                flag=true;
                c[cnt++]=res%10;
            }
            else
                c[cnt++]=res%10;
        }
        if(flag)//處理最後一位的進位
        {
            c[cnt++]=1;
            flag=false;
        }
    }
    j=0;
    for(i=0; i<cnt; ++i)
    {
        if(c[i]>=10)//處理二位數
        {
            res[j++]=c[i]%10;
            res[j++]=c[i]/10;
        }
        else
            res[j++]=c[i];
    }
    struct ListNode *s,*r;
    ans=(struct ListNode *)malloc(sizeof(struct ListNode));
    ans->val=res[0];
    ans->next=NULL;
    r=ans;
    for(i=1; i<j; ++i)//尾插
    {
        s=(struct ListNode *)malloc(sizeof(struct ListNode));
        s->val=res[i];
        r->next=s;
        r=s;
    }
    r->next=NULL;
    return ans;
}
/***************************************************/
void Create(ListNode *&h,int a[],int n)//建立一個不帶頭節點的單連結串列
{
    int i;
    ListNode *s,*r;
    h=(ListNode *)malloc(sizeof(ListNode));
    h->val=a[0];
    h->next=NULL;
    r=h;
    for(i=1; i<n; ++i)
    {
        s=(ListNode *)malloc(sizeof(ListNode));
        s->val=a[i];
        r->next=s;
        r=s;
    }
    r->next=NULL;
}
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("F:/cb/read.txt","r",stdin);
    //freopen("F:/cb/out.txt","w",stdout);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    int m,n;
    while(cin>>m>>n)
    {
        struct ListNode *ans;
        ListNode *l1,*l2;
        int a[100],b[100];
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(int i=0; i<m; ++i)
            cin>>a[i];
        Create(l1,a,m);//尾插法將a陣列中的值插入連結串列
        for(int i=0; i<n; ++i)
            cin>>b[i];
        Create(l2,b,n);
        ans=addTwoNumbers(l1,l2);
        struct ListNode *p=ans;
        while(p)
        {
            cout<<p->val;
            p=p->next;
        }
        cout<<endl;
    }
    return 0;
}

資料位數會比較長,所以不能先相加再存入新連結串列。

所以這其實就是兩個大數相加的模擬題啦。

手生的很(;д;),這種函式連結串列也忘了(;′⌒`),我賊菜哇QAQ。

我是分成兩個數位數長度的三種情況來討論的,相同位上的數依次相加考慮進位就OK了。

測試用例:

2 2
9 9
9 9

2 2
3 7
9 2

3 3
2 4 3
5 6 4

1 3
1
9 9 9

3 1
8 9 9
2

3 1
9 9 9
1

1 3
0
2 7 8

1 2
1
9 9

2 1
9 8
1

2 1
1 8
0

1 2
0
1 8

3 3
2 4 3
5 6 4

相關文章