leetcode 2 两数相加

题目描述：

给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。

如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外，这两个数都不会以 0 开头。

示例：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

这个题只有两个地方需要思路：进位加法、边界判断。

我第一遍的思路很直接，有一个空的就返回另一个就行了，两个都不空就进位加法，然后空一个换个while继续直到都加完了。

最后我漏掉的一个边界就是，两个都空了还有最高位的进位。

第一遍思路很直，所以就直接写了三个while，每个while内容差不多粘贴也不用手打，然后觉得代码很丑……就统一判断写了份短的。但是我还是把长的记下来，因为三while执行24ms，统一判断需要36ms，差得很多了。

数据结构：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

统一判断代码简洁：

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        if(l1==NULL) return l2;
        if(l2==NULL) return l1;
        ListNode* res=new ListNode(0);
        ListNode* cur=res;
        int flag=0;
        while(l1!=NULL || l2!=NULL)
        {
            int ans=(l1!=NULL?l1->val:0)+(l2!=NULL?l2->val:0)+flag;//链表空了就当加个0，统一处理
            if(ans>=10)//处理进位
            {
                ans-=10;
                flag=1;
            }else flag=0;
            ListNode* tmp=new ListNode(ans);
            cur->next=tmp;
            l1=(l1==NULL?NULL:l1->next);//非空就往后移一位
            l2=(l2==NULL?NULL:l2->next);
            cur=cur->next;
        }
        if(flag==1)
        {
            ListNode* tmp=new ListNode(1);
            cur->next=tmp;
        }
        return res->next;
    }
};

三while代码效率高：

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        if(l1==NULL) return l2;
        if(l2==NULL) return l1;
        ListNode* res=new ListNode(0);
        ListNode* cur=res;

        int flag=0;
        while(l1!=NULL && l2!=NULL)//先把两个都非空的加了
        {
            int ans=l1->val+l2->val+flag;
            if(ans>=10)//取进位
            {
                ans-=10;
                flag=1;
            }else flag=0;
            ListNode* tmp=new ListNode(ans);
            cur->next=tmp;
            l1=l1->next;
            l2=l2->next;
            cur=cur->next;
        }
        while(l1!=NULL)//再分别加没加完的l1和l2
        {
            int ans=l1->val+flag;
            if(ans>=10)
            {
                ans-=10;
                flag=1;
            }else flag=0;
            ListNode* tmp=new ListNode(ans);
            cur->next=tmp;
            l1=l1->next;
            cur=cur->next;
        }
        while(l2!=NULL)
        {
            int ans=l2->val+flag;
            if(ans>=10)
            {
                ans-=10;
                flag=1;
            }else flag=0;
            ListNode* tmp=new ListNode(ans);
            cur->next=tmp;
            l2=l2->next;
            cur=cur->next;
        }
        if(flag==1)//链表都空时的最高位进位。
        {
            ListNode* tmp=new ListNode(1);
            cur->next=tmp;
        }
        return res->next;//这是我为了方便，头结点随便建了一个直接从next开始链正常的结果。
    }
};