LeetCode 61 旋轉連結串列

題目描述

給定一個連結串列，旋轉連結串列，將連結串列每個節點向右移動 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

題目分析

1. k可能很大，取模即可 k = k%n
2. 需要找一下規律

okok

程式碼

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head==null||k==0) return head;

        //取模
        int n = 0;
        ListNode p = head;
        while(p!=null){
            n++;
            p = p.next;
        }
        k = k % n;

        if(k==0) return head;

        ListNode dummy = new ListNode(-1);
        dummy.next = head;
        ListNode first = dummy;
        ListNode second = dummy;

        //1. 讓first先走k
        while(k-->0){
            first = first.next;
        }

        //2. 一同走到first.next = null
        while(first.next!=null){
            first = first.next;
            second = second.next;
        }

        //3. 交換
        //3.1 將first.next 指向 head
        first.next = dummy.next;
        //3.2 將dummy.next 指向 second.next;
        dummy.next = second.next;
        //3.3 將second.next 指向null
        second.next = null;

        return dummy.next;


    }
}
 

複雜度

此題遍歷一遍連結串列節點
所以複雜度為：O(n)