链式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 {
    //链式A+B
public:
    ListNode* plusAB(ListNode* a, ListNode* b) {
        if(!a && !b)
            return NULL;
        
        if(!a)
            return b;
        
        if(!b)
            return a;
        
        ListNode *res = new ListNode(-1);
        ListNode *result = res;
        ListNode *pA = a;
        ListNode *pB = b;
        
        int carry = 0;
        int valA = 0;
        int valB = 0;
        int val = 0;
        
        while(pA!=NULL || pB!=NULL || carry){
            valA = (pA == NULL ? 0 : pA->val);
            valB = (pB == NULL ? 0 : pB->val);
            val = valA + valB + carry;
            
            carry = val/10;
            
            ListNode *node = new ListNode(val%10);
            res->next = node;
               res = res->next;
            
            pA = (pA == NULL ? NULL : pA->next);
            pB = (pB == NULL ? NULL : pB->next);
        }
        return result->next;
    };
};