LeetCode - 旋轉鏈表

題目鏈接：https://leetcode-cn.com/problems/rotate-list/

題目描述

給定一個鏈表，旋轉鏈表，將鏈表每個節點向右移動 k 個位置，其中 k 是非負數。

示例1：

輸入: 1->2->3->4->5->NULL, k = 2
輸出: 4->5->1->2->3->NULL
解釋:
向右旋轉 1 步: 5->1->2->3->4->NULL
向右旋轉 2 步: 4->5->1->2->3->NULL

示例2：

輸入: 0->1->2->NULL, k = 4
輸出: 2->0->1->NULL
解釋:
向右旋轉 1 步: 2->0->1->NULL
向右旋轉 2 步: 1->2->0->NULL
向右旋轉 3 步: 0->1->2->NULL
向右旋轉 4 步: 2->0->1->NULL

思路

• 這道題其實可以看成是字符串拼接問題，將後 k%len 個元素移到字符串前面
• 有了這樣的思路實現起來就很容易，先計算出鏈表長度，然後將temp1指針指向鏈表最後一個元素，temp2指針指向新鏈表的最後一個元素
• 將temp1指針指向原鏈表表頭
• 將temp2指針指向的元素置空

代碼如下：

/**
 * 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) {
        if(head == NULL) return NULL;
        ListNode* temp1 = head;
        int len = 1; 
        while(temp1->next != NULL) {
            len++;
            temp1 = temp1->next;
        }
        if(len <= 1) return head;
        k = (k >= len) ? (k % len) : k;
        if(k == 0) return head;
        ListNode* temp2 = head;
        for(int i = 0; i < len-k-1; i++) {
            temp2 = temp2->next;
        }
        ListNode* newHead = temp2->next;
        temp1->next = head;
        temp2->next = NULL;
        return newHead;
    }
};