快捷導(dǎo)航

Python 數(shù)據(jù)結(jié)構(gòu)之旋轉(zhuǎn)鏈表

更新時間：2017年02月25日 08:57:05 作者：guoziqing506

這篇文章主要介紹了Python 數(shù)據(jù)結(jié)構(gòu)之旋轉(zhuǎn)鏈表的相關(guān)資料,需要的朋友可以參考下

題目描述：給定一個鏈表，旋轉(zhuǎn)鏈表，使得每個節(jié)點向右移動k個位置，其中k是一個非負(fù)數(shù)

樣例：給出鏈表1->2->3->4->5->null和k=2；返回4->5->1->2->3->null

首先，觀察一下這個題目要達(dá)到的目的，其實，換一種說法，可以這樣來描述：給出一個k值，將鏈表從倒數(shù)第k個節(jié)點處起之后的部分移動到鏈表前面，就樣例來說，其實是將4->5這一部分移動到整個鏈表前面，變成4->5->1->2->3->null。不過，需要注意的是，題中沒有給出k的大小，當(dāng)k比鏈表的長度還大的時候，我們就需要先用k對鏈表的長度求余，比如，如果k = 7，那么上面的例子還是將4->5移動到整個鏈表前面。

所以說，這個題的思路可以這樣來總結(jié)：

1. 先求出整個鏈表的長度
2. 根據(jù)k值找到需要移動的部分鏈表的前驅(qū)（樣例中的3）
3. 在前驅(qū)之后將鏈表斷開，移動后半部分

代碼如下：

# 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: 
      cur = cur.next 
      count += 1 
    # 為節(jié)省代碼量，這里是一個很有技巧的處理：用尾節(jié)點鏈接頭結(jié)點 
    cur.next = head 
    # 此處，k為cur從尾節(jié)點到要斷開部分的前驅(qū)需走的步數(shù) 
    k = count - k % count 
    # 找到前驅(qū) 
    while k != 0: 
      cur = cur.next 
      k -= 1 
    # 斷開 
    head = cur.next 
    cur.next = None 
    # 因為首尾已經(jīng)相連，所以直接返回前驅(qū)后面的那個節(jié)點即可，此處引用為head 
    return head 
    # write your code here

需要注意的是21行首尾相連的技巧，這大大節(jié)省了我們的代碼量，其實，就按之前思路中所描述的一步步來，也沒問題。但是這個技巧確實很棒，值得學(xué)習(xí)。具體的細(xì)節(jié)我寫在了代碼注釋里。

感謝閱讀，希望能幫助到大家，謝謝大家對本站的支持！

您可能感興趣的文章: