作者: 碼蹄疾 畢業於哈爾濱工業大學。 小米廣告第三代廣告引擎的設計者、開發者； 負責小米應用商店、日曆、開屏廣告業務線研發； 主導小米廣告引擎多個模組重構； 關注推薦、搜尋、廣告領域相關知識;
給定一個連結串列，旋轉連結串列，將連結串列每個節點向右移動 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
複製程式碼題解
昨晚吃火鍋吃撐了回來這道題，還算順利~~ 連結串列的題目,其實就是在考指標交換,這個題目先讓連結串列連成一個環，然後再切開就可以完成了。
python版本
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def rotateRight(self, head, k):
"""
:type head: ListNode
:type k: int
:rtype: ListNode
"""
if head is None or head.next is None:
return head
# 連結串列的節點個數
count = 1
cur = head
while cur.next:
count += 1
cur = cur.next
# 如果恰好走了一個環,就直接返回
k = k % count
if k == 0:
return head
cur.next = head
dummy = ListNode(-1)
dummy.next = head
prev = dummy
# 需要走count-k個,然後把連結串列切斷
for _ in range(count - k):
prev = prev.next
# 重新組成新的連結串列
cur = prev.next
new_head = cur
prev.next = None
return new_head
複製程式碼java版本
public class Solution {
public ListNode rotateRight(ListNode head, int k) {
if (head == null || head.next == null) {
return head;
}
int count = 1;
ListNode cur = head;
while (cur.next != null) {
count++;
cur = cur.next;
}
k = k % count;
if (k == 0) {
return head;
}
cur.next = head;
ListNode dummy = new ListNode(-1);
dummy.next = head;
ListNode prev = dummy;
for (int i = 0; i < count - k; i++) {
prev = prev.next;
}
cur = prev.next;
prev.next = null;
return cur;
}
}
複製程式碼熱門閱讀
