LeetCode 61. 旋转链表
题目描述
🔥 61. 旋转链表
思路分析
- 计算链表长度:遍历链表计算其长度。
- 连接成环:将链表尾部连接到头部,形成一个环。
- 找到新的头节点:根据 k 值计算新的头节点位置,并断开环。
参考代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
func rotateRight(head *ListNode, k int) *ListNode {
if head == nil || head.Next == nil || k == 0 {
return head
}
// 1. 计算链表长度
length := 1
tail := head
for tail.Next != nil {
tail = tail.Next
length++
}
// 2. 处理 k 的值
k = k % length
if k == 0 {
return head
}
// 3. 连接成环
tail.Next = head
// 4. 找到新的头节点
newTail := tail
for i := 0; i < length-k; i++ {
newTail = newTail.Next
}
newHead := newTail.Next
// 5. 断开环
newTail.Next = nil
return newHead
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
class Solution {
public ListNode rotateRight(ListNode head, int k) {
if (head == null || head.next == null || k == 0) {
return head;
}
int n = 1;
ListNode cur = head;
while (cur.next != null) {
n++;
cur = cur.next;
}
k = k % n;
if (k == 0) {
return head;
}
cur.next = head;
for (int i = 0; i < n - k; i++) {
cur = cur.next;
}
ListNode newHead = cur.next;
cur.next = null;
return newHead;
}
}
CC BY-NC-SA 4.0
许可协议,转载请注明出处!
本博客所有文章除特别声明外,均采用