leetcode解题: Rotate List (61)

发表于 | 分类于

Given a list, rotate the list to the right by k places, where k is non-negative.

For example:
Given 1->2->3->4->5->NULL and k = 2,
return 4->5->1->2->3->NULL.

解法1：Two pointers, O(N)

这题考的是Linkedlist的基本操作。可以容易看到只要把最后k个元素作为字串挪到list的头部即可。
实际操作可以维护两个指针，同时向右移动，让左指针指向新的头部，右指针指向衔接的地方。
要注意的操作就是k可能会大于linkedlist的长度，那么需要多一个取余操作。
还有对输入的数据范围也需要进行判断，NULL linkedlist直接返回，取余后为0的k也直接返回原list。
C++

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        int len = getLength(head);
        k = k % len;
        ListNode left = head;
        ListNode right = head;
        int count = 0;
        while (count < k) {
            right = right->next;
            ++count;
        }
        while (right->next != NULL) {
            left = left->next;
            right = right->next;
        }
        ListNode* res = left->next;
        left->next = NULL;
        right->next = head;
        return res;
    }
    int getLength(ListNode* head) {
        int count = 0;
        while (head != NULL) {
            ++count;
            head = head->next;
        }
        return count;
    }
};

Java

1