【C语言刷LeetCode】86. 分隔链表(M)

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/partition-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

构造一个临时链表。

struct ListNode* partition(struct ListNode* head, int x){
	struct ListNode *temp;
	struct ListNode *find = head;
	struct ListNode *last = head;
	struct ListNode *tend;
	
	temp = (struct ListNode*)malloc(sizeof(struct ListNode));
	tend = temp;
	
	if(head == NULL) return head;
	
	find = find->next;
	while(find) {
		if(find->val < x) {
			last->next = find->next;
			tend->next = find;
			find->next = NULL;
			tend = find;
			find = last->next;
		} else {
			last = find;
			find = find->next;
		}
	}
	
	if(tend == temp) return head;
	
	if(head->val < x) {
		tend->next = head->next;
		head->next = temp->next;
	} else {
		tend->next = head;
		head = temp->next;
	}
	
	free(temp);
	
	return head;
}
发布了102 篇原创文章 · 获赞 17 · 访问量 11万+

猜你喜欢

转载自blog.csdn.net/jin615567975/article/details/104324815
今日推荐