Python數(shù)據(jù)結構之旋轉鏈表
題目描述:給定一個鏈表,旋轉鏈表,使得每個節(jié)點向右移動k個位置,其中k是一個非負數(shù)
樣例:給出鏈表1->2->3->4->5->null和k=2;返回4->5->1->2->3->null
首先,觀察一下這個題目要達到的目的,其實,換一種說法,可以這樣來描述:給出一個k值,將鏈表從倒數(shù)第k個節(jié)點處起之后的部分移動到鏈表前面,就樣例來說,其實是將4->5這一部分移動到整個鏈表前面,變成4->5->1->2->3->null。不過,需要注意的是,題中沒有給出k的大小,當k比鏈表的長度還大的時候,我們就需要先用k對鏈表的長度求余,比如,如果k = 7,那么上面的例子還是將4->5移動到整個鏈表前面。
所以說,這個題的思路可以這樣來總結:
1. 先求出整個鏈表的長度
2. 根據(jù)k值找到需要移動的部分鏈表的前驅(樣例中的3)
3. 在前驅之后將鏈表斷開,移動后半部分
代碼如下:
- # Definition for singly-linked list.
- # class ListNode:
- # def __init__(self, x):
- # self.val = x
- # self.next = None
- class Solution:
- # @param head: the list
- # @param k: rotate to the right k places
- # @return: the list after rotation
- def rotateRight(self, head, k):
- if head is None:
- return head
- cur = head
- count = 1
- # 計算鏈表長度
- while cur.next:
- curcur = cur.next
- count += 1
- # 為節(jié)省代碼量,這里是一個很有技巧的處理:用尾節(jié)點鏈接頭結點
- cur.next = head
- # 此處,k為cur從尾節(jié)點到要斷開部分的前驅需走的步數(shù)
- k = count - k % count
- # 找到前驅
- while k != 0:
- curcur = cur.next
- k -= 1
- # 斷開
- head = cur.next
- cur.next = None
- # 因為首尾已經(jīng)相連,所以直接返回前驅后面的那個節(jié)點即可,此處引用為head
- return head
- # write your code here
需要注意的是21行首尾相連的技巧,這大大節(jié)省了我們的代碼量,其實,就按之前思路中所描述的一步步來,也沒問題。但是這個技巧確實很棒,值得學習。具體的細節(jié)我寫在了代碼注釋里。