ย้อนกลับโหนดใน k-Group ใน C ++

สมมติว่าเรามีรายการที่เชื่อมโยง เราต้องย้อนกลับโหนดของรายการที่เชื่อมโยง k ทีละรายการและส่งคืนรายการที่แก้ไข โดยที่ k เป็นจำนวนเต็มบวกและน้อยกว่าหรือเท่ากับความยาวของรายการที่เชื่อมโยง ดังนั้นหากจำนวนโหนดไม่ใช่ทวีคูณของ k โหนดที่แยกออกไปในท้ายที่สุดก็ควรเป็นเหมือนเดิม

ดังนั้นหากรายการที่เชื่อมโยงเป็นเช่น [1,2,3,4,5,6,7] และ k เป็น 3 ผลลัพธ์จะเป็น [3,2,1,6,5,4,7]

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

• กำหนดวิธีการที่เรียกว่า Solve() ซึ่งจะใช้ส่วนหัวของรายการที่เชื่อมโยง ใช้ partCount และ k

• ถ้า partCount เป็น 0 ให้กลับหัว

• newHead :=head, ก่อนหน้า :=null, x :=k

• ในขณะที่ newHead ไม่เป็นค่าว่างและ x ไม่ใช่ 0

  • temp :=ถัดไปของ newHead ถัดไปของ nextHead :=ก่อนหน้า

  • ก่อนหน้า :=newHead, newHead :=ชั่วคราว

• ถัดไปของหัว :=แก้ (newHead, partCount – 1, k)

• ย้อนกลับก่อนหน้า

• จากวิธีหลัก ให้ทำดังต่อไปนี้ −

• กลับแก้(หัวหน้าของรายการที่เชื่อมโยง ความยาวของรายการ / k, k)

ตัวอย่าง

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -

#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << v[i] << ", ";
   }
   cout << "]"<<endl;
}
void print_vector(vector<vector<auto>> v){
   cout << "[";
   for(int i = 0; i<v.size(); i++){
      cout << "[";
      for(int j = 0; j <v[i].size(); j++){
         cout << v[i][j] << ", ";
      }
      cout << "],";
   }
   cout << "]"<<endl;
}
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){
      cout << ptr->val << ", ";
      ptr = ptr->next;
   }
   cout << "]" << endl;
}
class Solution {
public:
   ListNode* solve(ListNode* head, int partitionCount, int k){
      if(partitionCount == 0)return head;
      ListNode *newHead = head;
      ListNode* prev = NULL;
      ListNode* temp;
      int x = k;
      while(newHead && x--){
         temp = newHead->next;
         newHead->next = prev;
         prev = newHead;
         newHead = temp;
      }
      head->next = solve(newHead, partitionCount - 1, k);
      return prev;
   }
   int calcLength(ListNode* head){
      int len = 0;
      ListNode* curr = head;
      while(curr){
         len++;
         curr = curr->next;
      }
      return len;
   }
   ListNode* reverseKGroup(ListNode* head, int k) {
      int length = calcLength(head);
      return solve(head, length / k, k);
   }
};
main(){
   vector<int> v = {1,2,3,4,5,6,7};
   ListNode *head = make_list(v);
   Solution ob;
   print_list(ob.reverseKGroup(head, 3));
}

อินพุต

1,2,3,4,5,6,7
3

ผลลัพธ์

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