題目描述:給定一個(gè)鏈表,旋轉(zhuǎn)鏈表,使得每個(gè)節(jié)點(diǎn)向右移動(dòng)k個(gè)位置,其中k是一個(gè)非負(fù)數(shù)
樣例:給出鏈表1->2->3->4->5->null和k=2;返回4->5->1->2->3->null
首先,觀察一下這個(gè)題目要達(dá)到的目的,其實(shí),換一種說法,可以這樣來描述:給出一個(gè)k值,將鏈表從倒數(shù)第k個(gè)節(jié)點(diǎn)處起之后的部分移動(dòng)到鏈表前面,就樣例來說,其實(shí)是將4->5這一部分移動(dòng)到整個(gè)鏈表前面,變成4->5->1->2->3->null。不過,需要注意的是,題中沒有給出k的大小,當(dāng)k比鏈表的長度還大的時(shí)候,我們就需要先用k對鏈表的長度求余,比如,如果k = 7,那么上面的例子還是將4->5移動(dòng)到整個(gè)鏈表前面。
所以說,這個(gè)題的思路可以這樣來總結(jié):
1. 先求出整個(gè)鏈表的長度
2. 根據(jù)k值找到需要移動(dòng)的部分鏈表的前驅(qū)(樣例中的3)
3. 在前驅(qū)之后將鏈表斷開,移動(dòng)后半部分
代碼如下:
# 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 # 計(jì)算鏈表長度 while cur.next: cur = cur.next count += 1 # 為節(jié)省代碼量,這里是一個(gè)很有技巧的處理:用尾節(jié)點(diǎn)鏈接頭結(jié)點(diǎn) cur.next = head # 此處,k為cur從尾節(jié)點(diǎn)到要斷開部分的前驅(qū)需走的步數(shù) k = count - k % count # 找到前驅(qū) while k != 0: cur = cur.next k -= 1 # 斷開 head = cur.next cur.next = None # 因?yàn)槭孜惨呀?jīng)相連,所以直接返回前驅(qū)后面的那個(gè)節(jié)點(diǎn)即可,此處引用為head return head # write your code here
需要注意的是21行首尾相連的技巧,這大大節(jié)省了我們的代碼量,其實(shí),就按之前思路中所描述的一步步來,也沒問題。但是這個(gè)技巧確實(shí)很棒,值得學(xué)習(xí)。具體的細(xì)節(jié)我寫在了代碼注釋里。
感謝閱讀,希望能幫助到大家,謝謝大家對本站的支持!
新聞熱點(diǎn)
疑難解答
圖片精選