Sum the linked lists-优快云博客

Sum the linked lists
1->2->3
4->5

result is 1->6->8

1->5->9

5->3

result 2->1->2

A: 把链表的值加起来，注意，根据sample，是从链表的最后一位开始相加，而且，要考虑进位的问题

pseudo code如下（通过编译，但没有测试用例）

View Code

LinkNode* SumLinkLists(LinkNode *p1, LinkNode *p2)
{
if (NULL == p1 && NULL == p2)
return NULL;

if (NULL == p1 && NULL != p2)
return p2;

if (NULL != p1 && NULL == p2)
return p1;

    std::stack<LinkNode*> s1;
    std::stack<LinkNode*> s2;

    LinkNode* t1 = p1;
    LinkNode* t2 = p2;

while (t1)
    {
        s1.push(t1);    
        t1 = t1->next;
    }

while (t2)
    {
        s2.push(t2);
        t2 = t2->next;
    }

int sum = 0, carry = 0;
    LinkNode* p = new LinkNode();
    LinkNode* h = p;

while (!s1.empty() && !s2.empty())
    {
        t1 = s1.top();
        t2 = s2.top();

        sum = t1->data + t2->data + carry;    // 得到当前结点的和，同时加上进位

if (sum > 10)    // 计算需要存储到结点上的'和'及'进位'
        {
            sum = sum%10;
            carry = sum/10;
        }

        LinkNode* n = new LinkNode();
        n->data = sum;

        p->next = n;
        p = p->next;        

        s1.pop();
        s2.pop();
    }

// 将剩余的数据也添加进去
    std::stack<LinkNode*> &refL = (!s1.empty())? s1:s2;
while (!refL.empty())
    {
        sum = refL.top()->data + carry;
if (sum > 10)
        {
            sum = sum%10;
            carry = sum/10;
        }

        LinkNode* n = new LinkNode();
        n->data = sum;
        p->next = n;
        p = p->next;
        refL.pop();
    }

    p = h->next; // 得到真正的head
    delete h;
    h = NULL;

// reverse the link list p
// 反转链表
    LinkNode* pre = p;
    p = p->next;
    pre->next = NULL;

while (p)
    {
// pre -> p -> nex
        LinkNode* n = p->next;

        p->next = pre;
        pre = p;
        p = n;
    }

return pre;
}