[LeetCode] 369. Plus One Linked List

Given a non-negative integer represented as a linked list of digits, plus one to the integer.

The digits are stored such that the most significant digit is at the head of the list.

Example 1:

Input: head = [1,2,3]
Output: [1,2,4]

Example 2:

Input: head = [0]
Output: [1]

Constraints:

• The number of nodes in the linked list is in the range [1, 100].
• 0 <= Node.val <= 9
• The number represented by the linked list does not contain leading zeros except for the zero itself. 

给单链表加一。

用一个 非空 单链表来表示一个非负整数，然后将这个整数加一。你可以假设这个整数除了 0 本身，没有任何前导的 0。这个整数的各个数位按照 高位在链表头部、低位在链表尾部 的顺序排列。

这个题有多种思路，我这里给出的是双指针的解法。思路是创建一个dummy节点，还是连到head之前（dummy.next = head）。此时再创建两个指针 i 和 j ，也都从dummy的位置开始往后走。j 只做一件事，就是看整个list里面是否有9，若遍历到当前位置不是9的话，也将i移动到该位置，直到j遍历完整个list。此时 i 停下的地方只会是两种情况

• list的最后一个节点，此时只要对这个数字+1即可
• 如果最后一个节点是9的话，i停下的位置会在9之前的一个节点。比如[1, 2, 9]好了，i会停在2的位置。

第一种情况很好处理；第二种情况，因为i停下的位置是在2，所以只要把2变成3，将剩下的digits全部改为0即可。注意一个corner case是如果只有一个digit怎么办，这也就是为什么指针i 和 j 都从 dummy 开始而不是从 dummy.next 开始遍历的原因。

时间O(n)

空间O(1)

Java实现

class Solution {
    public ListNode plusOne(ListNode head) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;
        ListNode i = dummy;
        ListNode j = dummy;
        while (j.next != null) {
            j = j.next;
            if (j.val != 9) {
                i = j;
            }
        }
        i.val++;
        i = i.next;
        while (i != null) {
            i.val = 0;
            i = i.next;
        }
        if (dummy.val == 0) {
            return dummy.next;
        }
        return dummy;
    }
}

JavaScript实现

/**
 * Definition for singly-linked list.
 * function ListNode(val) {
 *     this.val = val;
 *     this.next = null;
 * }
 */
/**
 * @param {ListNode} head
 * @return {ListNode}
 */
var plusOne = function(head) {
    let dummy = new ListNode(0);
    dummy.next = head;
    let i = dummy;
    let j = dummy;
    while (j.next !== null) {
        j = j.next;
        if (j.val !== 9) {
            i = j;
        }
    }

    // i is the last non-9 digit
    i.val++;
    i = i.next;
    while (i !== null) {
        i.val = 0;
        i = i.next;
    }
    return dummy.val === 0 ? dummy.next : dummy;
};

LeetCode 题目总结