LeetCode 61. 旋转链表(C++)

最新推荐文章于 2024-03-25 21:54:55 发布

原创最新推荐文章于 2024-03-25 21:54:55 发布 · 693 阅读

3 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

78 篇文章

订阅专栏

c++

75 篇文章

订阅专栏

该博客介绍了如何解决LeetCode 61题——旋转链表。当给定链表每个节点向右移动k个位置时，需要先计算链表长度，并根据链表周期性旋转特性进行操作。文章提到了特殊情况处理和具体步骤，并表示期待更简洁的解决方案。

题目：

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

思路：

这道题类似于LeetCode19。二者都是需要找出链表的倒数第k个节点，但是有所不同的是19题提供的数值是小于链表的长度，因此19题可以采用双索引只遍历一次链表，而本题需要先知晓链表的长度。具体步骤如下：
1.判断链表是否为空或只有一个节点，是的话直接返回head。
2.建立虚拟头结点，遍历链表，求出链表长度len。
3.由于题目中定义的旋转操作实际上是具有周期性的，因此k%len求出链表应该旋转的步数。步数为零直接返回head即可。
4.求链表的最后一个节点和倒数第k个节点前一节点（不求倒数第k个节点的原因是因为需要将倒数第k+1个节点指向NULL），然后让虚拟头结点指向第k个节点，最后一个节点指向head，第k—1个节点指向空。
ps:感觉代码可以更简洁一些，但是还没有找到方法，如果谁有办法，请留言。谢谢！

/**
 * 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 || head->next == NULL || k == 0)
            return head;
        ListNode* dummyHead = new ListNode(0);
        dummyHead->next = head;
        ListNode* p = dummyHead;
        ListNode* q = dummyHead;
        int len = 0; //链表的长度
        while(q->next != NULL){   //q为链表的最后一个节点
            q = q->next;
            ++len;
        }
        k = k % len;    //由于题目定义的旋转具有重复性
        if(k == 0)
            return head;
        //找到倒数第k -1个节点的
        int m = len + 1 - k -1;
        for(int i =0; i < m; i++){
            p = p->next;
        }
        dummyHead->next = p->next;
        q->next = head;
        p->next = NULL;
        return dummyHead->next;
    }
};