Giả sử chúng ta có một danh sách liên kết, chúng ta phải đảo ngược các nút của danh sách liên kết k tại một thời điểm và trả về danh sách đã sửa đổi của nó. Ở đây k là một số nguyên dương và nhỏ hơn hoặc bằng độ dài của danh sách liên kết. Vì vậy, nếu số lượng nút không phải là bội số của k thì cuối cùng, các nút bên trái sẽ vẫn như cũ.
Vì vậy, nếu danh sách liên kết giống như [1,2,3,4,5,6,7] và k là 3, thì kết quả sẽ là [3,2,1,6,5,4,7].
Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -
-
Xác định một phương thức có tên là giải quyết (), phương thức này sẽ lấy phần đầu của danh sách được liên kết, lấy partCount và k
-
nếu partCount là 0, thì trả về head
-
newHead:=đầu, trước:=null, x:=k
-
trong khi newHead không phải là null và x không phải là 0
-
temp:=next of newHead, next of nextHead:=trước
-
trước:=newHead, newHead:=temp
-
-
next of head:=giải quyết (newHead, partCount - 1, k)
-
trả lại trước
-
Từ phương thức chính, thực hiện như sau -
-
trả về giải quyết (đầu danh sách được liên kết, độ dài của danh sách / k, k)
Ví dụ
Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -
#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));
} Đầu vào
1,2,3,4,5,6,7 3
Đầu ra
[3, 2, 1, 6, 5, 4, 7, ]