LeetCode-25. K 个一组翻转链表（Reverse Nodes in k-Group）

最新推荐文章于 2023-01-29 14:27:23 发布

Stillsings

最新推荐文章于 2023-01-29 14:27:23 发布

阅读量236

点赞数

分类专栏： leetcode 文章标签： leetcode 25 K个一组翻转链表 Reverse Nodes in k-Group C++

本文链接：https://blog.csdn.net/Listen_heart/article/details/103099319

版权

leetcode 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

和24题差不多，用一个指针记录前一个节点，需要特判，若count结束后并不等于k则翻转回来

class Solution {
public:
    ListNode* reverseKGroup(ListNode* head, int k) {
    	ListNode *newHead = new ListNode(0);
    	ListNode *relativeHead = newHead;
    	ListNode *tail = newHead;
    	ListNode *p = head;
    	ListNode *pre = head;
    	int count = 0;
    	while(p) {
    		if(count == k || count == 0) {
    			relativeHead = tail;
    			tail = p;
    			count = 0;
    		}
    		pre = p;
    		p = p->next;
    		pre->next = relativeHead->next;
    		relativeHead->next = pre;
    		count++;
    	}
    	if(count != k){
    		p = relativeHead->next;
    		relativeHead->next = NULL;
    		while(p) {
    			pre = p;
    			p = p->next;
    			pre->next = relativeHead->next;
    			relativeHead->next = pre;
    		}
    	}
    	return newHead->next;      
    }
};