LeetCode刷題筆記(鏈表):rotate-list



題目描述

Given a list, rotate the list to the right by k places, where k is non-negative.

For example:
Given1->2->3->4->5->NULLand k =2,
return4->5->1->2->3->NULL.

解題思路

  • 遍歷鏈表得到鏈表長度並定位fast到尾結點;
  • 根據鏈表長度,slow找到新的尾結點,slow的下一個結點便是新的頭結點因爲k可能比len大,所以需要取餘數;
  • 原先的尾結點與原先的頭節點相連,dummy的next指針更新指向新的頭結點,同時斷開新的尾節點與新的頭結點之間的聯繫。

C++版代碼實現

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode *rotateRight(ListNode *head, int k) {
        if(k == 0 || head == NULL || head->next == NULL)
            return head;

        ListNode *dummy = new ListNode(0);
        dummy->next = head;
        ListNode *fast = head;
        ListNode *slow = head;
        int len;
        //計算鏈表長度,同時fast指向尾結點
        for(len = 1; fast->next != NULL; ++len)
            fast = fast->next;
        //根據得到的鏈表長度,找到需要切開的結點
        for(int i = 1; i < len - k % len; ++i)
            slow = slow->next;
        fast->next = dummy->next;   //尾結點連接到頭節點,形成環狀鏈表
        dummy->next = slow->next;   //fast的下一個結點作爲新的頭結點
        slow->next = NULL;          //slow結點和新的頭結點斷開

        return dummy->next;
    }
};

系列教程持續發佈中,歡迎訂閱、關注、收藏、評論、點贊哦~~( ̄▽ ̄~)~

完的汪(∪。∪)。。。zzz

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章