LeetCode-061-旋轉(zhuǎn)鏈表

給定一個鏈表,旋轉(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

解題思路

  1. 求出鏈表長度, 同時記錄尾節(jié)點(diǎn)
  2. k %= length, 取模, k = length 也就是完整一圈, 相當(dāng)于沒移動
  3. k == 0直接返回
  4. 循環(huán)遍歷找出新的頭節(jié)點(diǎn)的上一個節(jié)點(diǎn)
  5. 將該節(jié)點(diǎn)的 next 記錄為新頭節(jié)點(diǎn), 然后再將 next 置為 null 斷開
  6. 將第一步記錄的尾節(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;
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容