LeetCode 2--两数相加 ( Add Two Numbers ) ( C语言版 )

题目描述 : 

代码如下   ( 附有解析 )

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* addTwoNumbers(struct ListNode* l1, struct ListNode* l2) {
    //head用于返回
    struct ListNode* head=(struct ListNode*)malloc(sizeof(struct ListNode));
    struct ListNode* tail=head;
    int count=0,sum=0;
    //sum用来判断是否需要进位,如果需要进位sum自减10,count值1用于和下一位相加
    for(;l1&&l2;l1=l1->next,l2=l2->next){
        sum=l1->val+l2->val+count;
        //判断是否进位
        if(sum>9){
            sum-=10;
            count=1;
        }
        else
            count=0;
        tail->next=(struct ListNode*)malloc(sizeof(struct ListNode));
        tail=tail->next;
        tail->val=sum;
    }
    //求出两个链表中没有走完的链表,如果有将链表剩余部分加在新链表的后边
    //注意也需要判断进位问题
    l1 = l1?l1:l2;
    for(;l1;l1=l1->next){
        sum=l1->val+count;
        if(sum>9){
            sum-=10;
            count=1;
        }
        else
            count=0;
        tail->next=(struct ListNode*)malloc(sizeof(struct ListNode));
        tail=tail->next;
        tail->val=sum;
    }
    //判断最后一位是否需要进位
    if(count){
        tail->next=(struct ListNode*)malloc(sizeof(struct ListNode));
        tail=tail->next;
        tail->val=count;
    }
    //返回链表的尾节点
    tail->next=NULL;
    tail=head;
    head=head->next;
    free(tail);
    return head;
}

猜你喜欢

转载自blog.csdn.net/ds19980228/article/details/82720649