链式A+B

题目描述

有两个用链表表示的整数，每个结点包含一个数位。这些数位是反向存放的，也就是个位排在链表的首部。编写函数对这两个整数求和，并用链表形式返回结果。

给定两个链表ListNode* A，ListNode* B，请返回A+B的结果(ListNode*)。

测试样例：

{1,2,3},{3,2,1}

返回：{4,4,4}

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};*/
class Plus {
public:
    ListNode* plusAB(ListNode* a, ListNode* b) {
        bool flag=false;
        ListNode *ans=new ListNode((a->val+b->val)%10);
        ListNode *p=ans;
        if(a->val+b->val>=10) flag=true;
        while(a->next!=nullptr&&b->next!=nullptr)
        {
            a=a->next;
            b=b->next;
            if(flag)
            {
                ListNode *temp=new ListNode((a->val+b->val+1)%10);
                p->next=temp;
                p=temp;
                if(a->val+b->val+1>=10) flag=true;
                else flag=false;
            }
            else{
                ListNode *temp=new ListNode((a->val+b->val)%10);
                p->next=temp;
                p=temp;
                if(a->val+b->val>=10) flag=true;
                else flag=false;
            }
        }
        if(!flag)
        {
            if(a->next!=nullptr)
                p->next=a->next;
            if(b->next!=nullptr)
                p->next=b->next;
            return ans;
        }
        else
        {
            while(a->next!=nullptr)
            {
                a=a->next;
              if(flag)
              {
                ListNode *temp=new ListNode((a->val+1)%10);
                p->next=temp;
                p=temp;
                if(a->val+1>=10) flag=true;
                else flag=false;
              }
             else
              {
                ListNode *temp=new ListNode((a->val)%10);
                p->next=temp;
                p=temp;
                if(a->val>=10) flag=true;
                else flag=false;
              }
            }
            while(b->next!=nullptr)
            {
                b=b->next;
              if(flag)
              {
                ListNode *temp=new ListNode((b->val+1)%10);
                p->next=temp;
                p=temp;
                if(b->val+1>=10) flag=true;
                else flag=false;
              }
             else
              {
                ListNode *temp=new ListNode((b->val)%10);
                p->next=temp;
                p=temp;
                if(b->val>=10) flag=true;
                else flag=false;
              }
            }
            if(flag)
            {
                ListNode *temp=new ListNode(1);
                p->next=temp;
            }
        }
        return ans;
    }
};