2. Add Two Numbers/两数相加

2019-04-28  本文已影响0人  蜜糖_7474

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

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

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

示例:

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

AC代码

vector<int> add(vector<int> v1, vector<int> v2) {
    int t, c = 0;
    vector<int> res;
    if (v1.size() < v2.size()) v1.swap(v2);
    auto it1 = v1.begin(), it2 = v2.begin();
    for (; it2 != v2.end(); ++it1, ++it2) {
        t = (*it1 + *it2 + c) % 10;
        c = (*it1 + *it2 + c) / 10;
        res.push_back(t);
    }
    while (it1 != v1.end()) {
        t = (*it1 + c) % 10;
        c = (*it1 + c) / 10;
        res.push_back(t);
        ++it1;
    }
    if (c != 0) res.push_back(c);
    reverse(res.begin(), res.end());
    while (res[0] == 0 && res.size() != 1) res.erase(res.begin());
    return res;
}

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        vector<int> v1, v2;
        while (l1) {
            v1.push_back(l1->val);
            l1 = l1->next;
        }
        while (l2) {
            v2.push_back(l2->val);
            l2 = l2->next;
        }
        vector<int> res = add(v1, v2);
        reverse(res.begin(), res.end());
        ListNode* ans = new ListNode(res[0]);
        ListNode* rt = ans;
        for (int i = 1; i < res.size(); ++i) {
            ListNode* t = new ListNode(res[i]);
            ans->next = t;
            ans = t;
        }
        return rt;
    }
};

总结

类字符串两数加法还是挺模板化操作的,就是代码有点长

上一篇下一篇

猜你喜欢

热点阅读