Problem 2 : Add Two Numbers

You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8


简单题,但是很久不做指针相关的练习,写的异常难受,各种问题。甚至一开始写的错误代码运行时瞬间占据90+的CPU,内存和100%的磁盘,直接死机。写的时候要注意细节。

解法:

同时遍历两个链表,相加存在新链表。遍历完成后若有剩余则把剩余项全部加入新链表;判断新链表每一项是否超过10,超过则做进位处理;最后单独判断最后一个数字是否进位。

public class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        if (l1 == null || l2 == null) {
            return null;
        }
        ListNode head = new ListNode(0);
        ListNode p = head;
        int flag = 0;// 记录是否进位
        int value = 0;// 求和值
        // 同时遍历两个链表
        while (l1 != null && l2 != null) {
            value = l1.val + l2.val + flag;
            p.next = new ListNode(value);
            p = p.next;
            l1 = l1.next;
            l2 = l2.next;
        }
        // l1或l2没有遍历完
        ListNode l3 = (l1 != null) ? l1 : l2;
        while (l3 != null) {
            p.next = new ListNode(l3.val);
            l3 = l3.next;
            p = p.next;
        }
        // 统一处理进位问题
        ListNode q = head.next;
        // 取最后一个节点
        ListNode end = null;
        while (q != null) {
            value = q.val + flag;
            if (value >= 10) {
                value -= 10;
                flag = 1;
            } else {
                flag = 0;
            }
            q.val = value;
            end = q;
            q = q.next;
        }
        // 最后一位可能进位
        if (flag == 1)
            end.next = new ListNode(1);
        return head.next;
    }
}

发布者

VC-Robot

游戏爱好者,动漫迷,C++修炼中,编程菜鸟,随性

发表评论

电子邮件地址不会被公开。 必填项已用*标注

This site uses Akismet to reduce spam. Learn how your comment data is processed.