赞
踩
题目链接:链表中的节点每k个一组翻转
解题思路1:递归
代码如下:
ListNode* reverseKGroup(ListNode* head, int k) { //找到每次反转的尾部,尾部即是反转后的头部 ListNode* tail = head; //遍历k次找到尾部,一组有k个,从当前节点往后走k部,就走到了第k个,也就是该组的尾部 //本for循环的目的就是找到每一组的头部,进行拼接 for(int i=0; i<k; i++){ if(tail == nullptr){ //如果不足k步走到了链表尾部,直接返回head,不反转,该组顺序不变 return head; } tail = tail->next; } //链表进行反转的时候需要前序节点和当前节点 //当前节点的next要指向前序节点,当前节点要指向下一个节点 ListNode* pre = nullptr; ListNode* cur = head; //在到达当前段尾节点前 while(cur != tail){ ListNode* temp = cur->next; cur->next = pre; pre = cur; cur = temp; } //head是原来的头,现在是反转后的尾,用来和下一组进行连接 //为什么返回pre呢?pre是每一个当前组的头节点 head->next = reverseKGroup(tail, k); return pre; }
解题思路2:非递归
代码如下:
ListNode* reverseKGroup(ListNode* head, int k) { //创建一个虚拟节点 ListNode* dummy = new ListNode(0); //虚拟节点指向头结点 dummy->next = head; ListNode* pre = dummy;//pre是每组的前一个节点,要和下一组进行连接 ListNode* end = dummy; while(end->next != nullptr){ //每k个一组进行反转,end是每k个链表的最后一个 for(int i=0; i<k&&end!=nullptr; i++){ end = end->next; } //如果end是空的,则说明不够k个,凑不成一组,直接退出循环即可 if(end == nullptr) break; //反转开始的节点是pre的下一个节点 ListNode* start = pre->next; //next是下一次反转的头结点 ListNode* next = end->next; //end是本组的最后一个节点 end->next = nullptr; //将pre和反转后的k个一组的链表连接起来 pre->next = reverse(start); start->next = next;//start是下一次反转的头,更新为next pre = start; end = start; } return dummy->next; } //链表的反转 ListNode* reverse(ListNode* head){ ListNode* pre = nullptr; ListNode* cur = head; while (cur != nullptr) { ListNode* temp = cur->next; cur->next = pre; pre = cur; cur = temp; } return pre; }
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。