สมมติว่าเรามีรายการเชื่อมโยง เราต้องหมุนรายการไปทางขวาโดย k ตำแหน่ง ค่าของ k จะเป็นบวก ดังนั้นหากรายการเป็นแบบ [1 −> 2 −> 3 −> 4 −> 5 −> NULL] และ k =2 ผลลัพธ์จะเป็น [4 −> 5 −> 1 −> 2 −> 3 −> NULL]
ให้เราดูขั้นตอน -
-
หากรายการว่างเปล่า ให้คืนค่า null
-
เลน :=1
-
สร้างหนึ่งโหนดที่เรียกว่า tail :=head
-
ในขณะที่ next of tail ไม่เป็นโมฆะ
-
เพิ่มเลนขึ้น 1
-
หาง :=ถัดจากหาง
-
-
ข้างท้าย :=หัว
-
k :=k mod len
-
newHead :=null
-
สำหรับ i :=0 ถึง len − k
-
หาง :=ถัดจากหาง
-
-
newHead :=ถัดจากหาง
-
ข้างท้าย :=null
-
กลับหัวใหม่
ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -
ตัวอย่าง
#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, ]
