給定一個鏈表,旋轉(zhuǎn)鏈表,將鏈表每個節(jié)點(diǎn)向右移動 k 個位置,其中 k 是非負(fù)數(shù)。
示例 1:
輸入: 1->2->3->4->5->NULL, k = 2
輸出: 4->5->1->2->3->NULL
解釋:
向右旋轉(zhuǎn) 1 步: 5->1->2->3->4->NULL
向右旋轉(zhuǎn) 2 步: 4->5->1->2->3->NULL
示例 2:
輸入: 0->1->2->NULL, k = 4
輸出: 2->0->1->NULL
解釋:
向右旋轉(zhuǎn) 1 步: 2->0->1->NULL
向右旋轉(zhuǎn) 2 步: 1->2->0->NULL
向右旋轉(zhuǎn) 3 步: 0->1->2->NULL
向右旋轉(zhuǎn) 4 步: 2->0->1->NULL
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/rotate-list
解題思路
- 求出鏈表長度, 同時記錄尾節(jié)點(diǎn)
- k %= length, 取模, k = length 也就是完整一圈, 相當(dāng)于沒移動
- k == 0直接返回
- 循環(huán)遍歷找出新的頭節(jié)點(diǎn)的上一個節(jié)點(diǎn)
- 將該節(jié)點(diǎn)的 next 記錄為新頭節(jié)點(diǎn), 然后再將 next 置為 null 斷開
- 將第一步記錄的尾節(jié)點(diǎn)的 next 置為原頭節(jié)點(diǎn)
代碼
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode rotateRight(ListNode head, int k) {
if (head == null || head.next == null || k == 0) {
return head;
}
ListNode cur = head;
ListNode tail; // 優(yōu)化: 提前找出尾節(jié)點(diǎn)
// 求出鏈表長度
int length = 1;
while (cur.next != null) {
length++;
cur = cur.next;
}
tail = cur; // 記錄尾節(jié)點(diǎn)
k %= length; // 原先用while去減去length, 傻逼
if (k == 0) {
return head; // k=0直接返回
}
ListNode newHead; // 新頭節(jié)點(diǎn)
cur = head;
// 循環(huán)找出新的頭節(jié)點(diǎn)的上一個節(jié)點(diǎn)
for (int i = 0; i < length - k - 1; i++) {
cur = cur.next;
}
newHead = cur.next; // 新頭節(jié)點(diǎn)
cur.next = null; // 斷開
tail.next = head; // 尾節(jié)點(diǎn)連上原來的頭節(jié)點(diǎn)
return newHead;
}
}