算法题2.两数相加

题目:
 给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
 如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
 您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例

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

所需类简单实现:

    public class ListNode {
        int val;
        ListNode next;

        ListNode(int x) {
            val = x;
        }
    }

我的思路:
 由于数是倒序存的,所以前面的数就是低位的数,所以通过设计一个进位标记,然后遍历然后依次累加即可。由于还得判断next所以代码显得比较冗长。

  public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
     boolean is = false; // 标记是否进位
        ListNode result = null, r2 = null; // result是结果,r2是最后一个next
        for (;;) {
            int iresult = l1.val + l2.val;

            if (is) {
                iresult += 1;  //加上进位
            }
            if (iresult >= 10) { //判断是否向高位进位
                is = true;
                iresult = iresult % 10;
            } else {
                is = false;
            }

            //存入结果
            if (result == null) {  
                result = new ListNode(iresult);
                r2 = result;
            } else {
                r2.next = new ListNode(iresult);
                r2 = r2.next;
            }

            if (l1.next == null && l2.next == null) {  //结束条件
                break;
            } else {
                l1 = l1.next == null ? new ListNode(0):l1.next;
                l2 = l2.next == null ? new ListNode(0):l2.next;
            }

        }
        if (is) {
            r2.next = new ListNode(1);
        }
        return result;
    }

官方题解:

  • 解题思路与我是一致的,不过代码上做了合并优化,特别是while循环里面的处理应该好好琢磨琢磨。特别注意的是这里返回的是dummyHead.next也蛮巧妙的,省去了链表开头或结尾的情况判断;以及carry进位标记的设置。
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
    ListNode dummyHead = new ListNode(0);
    ListNode p = l1, q = l2, curr = dummyHead;
    int carry = 0;
    while (p != null || q != null) {
        int x = (p != null) ? p.val : 0;
        int y = (q != null) ? q.val : 0;
        int sum = carry + x + y;
        carry = sum / 10;
        curr.next = new ListNode(sum % 10);
        curr = curr.next;
        if (p != null) p = p.next;
        if (q != null) q = q.next;
    }
    if (carry > 0) {
        curr.next = new ListNode(carry);
    }
    return dummyHead.next;
}

你可能感兴趣的:(算法题2.两数相加)