用 C++ 旋轉一個連結列表 k 位


假設我們有一個連結串列。我們必須將該連結串列向右旋轉 k 位。k 的值為正值。因此,如果連結串列如下:[1 −> 2 −> 3 −> 4 −> 5 −> NULL],且 k = 2,則輸出為:[4 −> 5 −> 1 −> 2 −> 3 −> NULL]

讓我們看看這些步驟 −

  • 如果連結串列為空,則返回 null

  • len := 1

  • 建立一個名為 tail := head 的節點

  • 當 tail 的下一個節點不為 null 時

    • 將 len 加 1

    • tail := tail 的下一個節點

  • tail 的下一個節點 := head

  • k := k mod len

  • newHead := null

  • 對於 i := 0 到 len − k

    • tail := tail 的下一個節點

  • newHead := tail 的下一個節點

  • tail 的下一個節點 := null

  • 返回 newHead

讓我們看看以下實現程式碼,以便更好地理解 −

示例

 即時演示

#include <bits/stdc++.h>
using namespace std;
class ListNode{
   public:
   int val;
   ListNode *next;
   ListNode(int data){
      val = data;
      next = NULL;
   }
};
ListNode *make_list(vector<int> v){
   ListNode *head = new ListNode(v[0]);
   for(int i = 1; i<v.size(); i++){
      ListNode *ptr = head;
      while(ptr->next != NULL){
         ptr = ptr->next;
      }
      ptr->next = new ListNode(v[i]);
   }
   return head;
}
void print_list(ListNode *head){
   ListNode *ptr = head;
   cout << "[";
   while(ptr->next){
      cout << ptr->val << ", ";
      ptr = ptr->next;
   }
   cout << "]" << endl;
}
class Solution {
   public:
   ListNode* rotateRight(ListNode* head, int k) {
      if(!head) return head;
      int len = 1;
      ListNode* tail = head;
      while(tail->next){
         len++;
         tail = tail->next;
      }
      tail->next = head;
      k %= len;
      ListNode* newHead = NULL;
      for(int i = 0; i < len - k; i++){
         tail = tail->next;
      }
      newHead = tail->next;
      tail->next = NULL;
      return newHead;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,2,3,4,5,6,7,8,9};
   ListNode *head = make_list(v);
   print_list(ob.rotateRight(head, 4));
}

輸入

[1,2,3,4,5,6,7,8,9], 4

輸出

[6, 7, 8, 9, 1, 2, 3, 4, ]

於以下時間更新: 2020-10-21

277 次瀏覽

開始你的 職業生涯

完成課程獲得認證

開始
廣告