#leetcode刷题之路25- k个一组翻转链表
程序员文章站
2022-05-14 10:49:12
给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。 示例 :给定这个链表:1->2->3->4->5当 k = 2 时,应当返回: 2->1->4->3->5当 k = 3 时, ......
给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
示例 :
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明 :
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
思路:先把是不是从头开始区分一下,然后每一次翻转之前,都要判断数量够不够k
#include <iostream> using namespace std; struct listnode { int val; listnode *next; listnode(int x) : val(x), next(null) {} }; listnode* createlist(int n)//生成链表 { if (n == 0) return nullptr; listnode* head = (listnode*)malloc(sizeof(listnode)); cin >> head->val; listnode* pre = head; for (int i = 0; i < n - 1; i++) { listnode* p = (listnode*)malloc(sizeof(listnode)); cin >> p->val; pre->next = p; pre = pre->next; } pre->next = nullptr; return head; } listnode* reverselist(listnode* head, listnode *p, int k) { int tk = k;//暂时保存k的值 if (head == p)//是头 { listnode* temp = head;// listnode* endd = head;// listnode* ttemp=nullptr; listnode* pre = nullptr;//保存已反转的下一个 listnode* end = nullptr;// listnode* t = head;//保存反转过的最后一个 while (tk > 0)//说明从头节点开始有k个节点可供反转 { if (temp != nullptr) { temp = temp->next; tk--; } else return head; }//while循环后temp指针指向链表的第k+1个节点 //cout<<temp->val<<endl; //开始反转,让头指向空 pre = head->next; head->next = nullptr; //cout<<temp->val<<endl; while (pre != temp) { //t = pre; end = pre->next; pre->next = head; head = pre; pre = end; } //cout << t->val << endl; endd->next = temp;//连接后面剩下的 //带头的前k个处理完了,判断接下来的够不够k个 tk = k; //temp = t; while (tk != 0)//是否可继续处理 { if (temp != nullptr) { temp = temp->next; tk--; } else return head; } //cout << t->val <<t->next->val<< endl; //cout << head->val << head->next->val << endl; reverselist(head, t, k);//够的话就递归反转 } else { listnode* pre = p;//保存已经反转过的最后一个-------------------- listnode*cur = pre->next;//保存待反转的第一个 listnode* cur_next=cur->next ;//保存待反转的下一个 listnode* end = nullptr;//存放待反转的最后一个节点的下一个节点---------------------- listnode* thead=nullptr;//存放转后的头节点--------------- listnode* tcur = cur_next->next;//存放当前的cur_next的下一个节点,即下次要反转的那个------------------ listnode* temp = nullptr;//记录反转的链表尾 listnode* t = nullptr;// //-----------------------------先反转第一次 cur_next->next = cur; cur->next = nullptr; temp = cur; thead = cur_next; tk = k-2; while (tk > 0) { cur = tcur; tcur = tcur->next; cur->next = thead; thead = cur; tk--; } pre->next = thead; temp->next = tcur; t = temp->next; //带头的前k个处理完了,判断接下来的够不够k个 tk = k; //temp = t; while (tk != 0)//是否可继续处理 { if (t != nullptr) { t = t->next; tk--; } else return head; } reverselist(head, temp, k);//够的话就递归反转 } return head; } listnode* reversekgroup(listnode* head, int k) { if (head == nullptr || k == 0 || k == 1) return head; listnode *temp = head; head = reverselist(head, head, k); return head; } int main() { listnode* head = createlist(4); listnode*ans = reversekgroup(head, 2); while (ans != nullptr) { cout << ans->val << endl; ans = ans->next; } return 0; }