61.旋轉鏈表

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

題解

右移k個位置，先遍歷一遍得到鏈表長度，同時將表尾與表頭鏈接形成環，在從頭開始遍歷n-k%n個位置，在這裏切斷鏈表，得到新的鏈表，再將新的鏈表表尾指針置空，返回新鏈表即可，代碼如下：

/**
 * 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) return null;
    if (head.next == null) return head;
//第一遍遍歷，得到鏈表長度，並連成環
    ListNode p = head;
    int n = 1;
    while(p.next != null){
      p = p.next;
      n++;
    }
    p.next = head;
//第二遍遍歷，找到切斷位置，打斷鏈表
    ListNode q = head;
    for (int i = 1; i < n - k % n; i++){
      q = q.next;
    }
//定義新的表頭，並將表尾指針置空
    ListNode newhead = q.next;
    q.next = null;
//返回新的鏈表
    return newhead;
  }
}

提交結果