leetcode - 链表两数相加

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

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

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

示例:

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

来源:力扣(LeetCode) 

 

/**
 * 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) {
        ListNode *head = new ListNode(0);
        ListNode *cur = head;
        int jinwei = 0;
        int n1 = 0 , n2 = 0 , sum = 0;
        while(1) {
            if(l1 == NULL && l2 == NULL) {
                //循环跳出条件 -- 考虑到最后一位的进位问题
                if(jinwei == 1) {
                    cur->next = new ListNode(jinwei);
                    cur = cur->next;
                }
                break;
            } 
            if(l1 == NULL) {
                n1 = 0;
                n2 = l2->val;
            } else if(l2 == NULL) {
                n2 = 0;
                n1 = l1->val;
            } else {
                n1 = l1->val;
                n2 = l2->val;
            } 
            sum = n1+n2;
            //取数
            cur->next = new ListNode((sum+jinwei) % 10);
            //判断是否有进位
            if((sum+jinwei)>=10) {  
                jinwei = 1;
            } else {  
                jinwei = 0; 
            }  
            //如果链表为空,就向前了.避免空指针
            if(l1 != NULL) l1 = l1->next;
            if(l2 != NULL) l2 = l2->next; 
            cur = cur->next;
        }
        return head->next;
    }
};

 

你可能感兴趣的:(C++,数据结构)