1. 程式人生 > >【LeetCode】61. Rotate List(C++)

【LeetCode】61. Rotate List(C++)

地址:https://leetcode.com/problems/rotate-list/

題目:

Given a linked list, rotate the list to the right by k k places, where k k

is non-negative.

Example 1:

Input: 1->2->3->4->5->NULL, k = 2
Output: 4->5->1->2->3->NULL
Explanation:
rotate 1 steps to the right: 5->1->2->3->4->NULL
rotate 2 steps to the right: 4->5->1->2->3->NULL

Example 2:

Input

: 0->1->2->NULL, k = 4
Output: 2->0->1->NULL
Explanation:
rotate 1 steps to the right: 2->0->1->NULL
rotate 2 steps to the right: 1->2->0->NULL
rotate 3 steps to the right: 0->1->2->NULL
rotate 4 steps to the right: 2->0->1->NULL

理解:

把一個連結串列向右旋轉k次,也就是把倒數第k個結點當作頭結點。需要改變的有尾結點的next域,倒數第k+1個結點的next域和head。

實現:

下面的實現非常優雅。
使用tail進行操作,tail就是head的前一個,因此可以寫for(int i=0;i<cnt-k;++i)
tail->next=head;就直接改變了尾節點的next

class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(!head) return nullptr;
        int cnt=1;
        ListNode* tail=head;
        while(tail->next){
            tail=tail->next;
            ++cnt;
        }
        tail->next=head;
        if(k%=cnt){
            for(int i=0;i<cnt-k;++i){
                tail=tail->next;
            }
        }
        head=tail->next;
        tail->next=nullptr;
        return head;
    }
};