LeetCode(61):旋转链表 Rotate Array 旋转数组

Medium!

题目描述:

给定一个链表,旋转链表,将链表每个节点向右移动 个位置,其中 是非负数。

示例 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

解题思路:

这道旋转链表的题和之前那道 Rotate Array 旋转数组 很类似,但是比那道要难一些,因为链表的值不能通过下标来访问,只能一个一个的走,我们可能首先想到的就是用快慢指针来解,快指针先走k步,然后两个指针一起走,当快指针走到末尾时,慢指针的下一个位置是新的顺序的头结点,这样就可以旋转链表了,自信满满地写完程序,放到OJ上跑,以为能一次通过,结果就跪在了各种特殊情况,首先一个就是当原链表为空时,直接返回NULL,还有就是当k大于链表长度和k远远大于链表长度时该如何处理,我们需要首先遍历一遍原链表得到链表长度n,然后k对n取余,这样k肯定小于n,就可以用上面的算法了。

C++解法一:

 1 class Solution {
 2 public:
 3     ListNode *rotateRight(ListNode *head, int k) {
 4         if (!head) return NULL;
 5         int n = 0;
 6         ListNode *cur = head;
 7         while (cur) {
 8             ++n;
 9             cur = cur->next;
10         }
11         k %= n;
12         ListNode *fast = head, *slow = head;
13         for (int i = 0; i < k; ++i) {
14             if (fast) fast = fast->next;
15         }
16         if (!fast) return head;
17         while (fast->next) {
18             fast = fast->next;
19             slow = slow->next;
20         }
21         fast->next = head;
22         fast = slow->next;
23         slow->next = NULL;
24         return fast;
25     }
26 };

这道题还有一种解法,跟上面的方法类似,但是不用快慢指针,一个指针就够了,原理是先遍历整个链表获得链表长度n,然后此时把链表头和尾链接起来,再往后走n - k % n个节点就到达新链表的头结点前一个点,这时断开链表即可。

C++解法二:

 1 class Solution {
 2 public:
 3     ListNode *rotateRight(ListNode *head, int k) {
 4         if (!head) return NULL;
 5         int n = 1;
 6         ListNode *cur = head;
 7         while (cur->next) {
 8             ++n;
 9             cur = cur->next;
10         }
11         cur->next = head;
12         int m = n - k % n;
13         for (int i = 0; i < m; ++i) {
14             cur = cur->next;
15         }
16         ListNode *newhead = cur->next;
17         cur->next = NULL;
18         return newhead;
19     }
20 };

猜你喜欢

转载自www.cnblogs.com/ariel-dreamland/p/9149729.html