LeetCode:61. 旋转链表(中等)

题目描述

给定一个链表，旋转链表，将链表每个节点向右移动 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

题解

题解1:快慢指针法

首先计算链表长度length, 实际需要旋转的次数为k= k % length。
快指针fast先向右移动k次, 然后快慢指针同时移动, 当快指针达到链表末尾时,说明慢指针slow后面的结点均需要移到链表头部再往后移动。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(!head || !head->next) return head;
        int length = 0;
        ListNode* temp = head;//头结点就是链表中的第一个结点
        ListNode* slow = head;
        ListNode* fast = head;
        while(temp){
            length++;
            temp = temp->next;
        } 
        k %= length;
        //cout << "k=" << k << endl;
        //cout << "length=" << length << endl;
        if(k == 0) return head;
        while(k--) fast = fast->next;
        while(fast->next){
            fast = fast->next;
            slow = slow->next;
        }
        fast->next = head;//快指针指向原头指针
        head = slow->next;//新头指针为慢指针的下一个指针
        slow->next = NULL;
        return head;
    }
};

复杂度分析

• 时间复杂度: O(n), n为链表长度。
• 空间复杂度: O(1)。

题解2:

由题意得: 移动k个位置就相当于把原链表最后k个节点切割下来，并重新拼接到原链表的head的前面即可。

/**
 * 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 || !head->next) {//当链表为空或者只有一个结点
            return head;
        }
        int length = 1;//计算长度的同时得到末尾结点tail
        auto tail = head;
        while (tail->next) {
            ++length;
            tail = tail->next;
        }
        k %= length;// 右移k位（k = k % length), 代表将最右边的k的结点放到第一个结点的前面
        if (k == 0) return head;// 不需要旋转
        // 找到最后k个节点起始节点 (倒数第k个节点，就是正数第length-k+1个节点)
        // newHead作为最后k个节点的开头，移动到head前面后，就会成为新的头部节点
        auto newHead = head;
        auto pre = newHead;// pre保存newHead的前一个节点
        // 第（length-k+1）个节点，就是从第一个节点往右右移（length-k）次
        for (int i = 0; i < length-k; ++i) {
            pre = newHead;
            newHead = newHead->next;
        }
        // 将后k个节点移动到head前面
        // 将链表从中间分开，前一半是 head到pre， 后一半是 newHead 到tail
        pre->next = nullptr;//NULL
        tail->next = head;   // 将原链表连接到tail的后面
        // newHead 成为新的链表头
        return newHead;
    }
};

复杂度分析

• 时间复杂度: O(n), n为链表长度。
• 空间复杂度: O(1)。

参考

【手绘漫画】图解LeetCode之旋转链表
61. 旋转链表（图文详解）
【快慢指针+迭代直接确定新头尾】61.旋转链表-图解思路Python
动图算法旋转链表-四种解法
C++ 利用取余切割并重新合并链表