[LeedCode OJ]#86 Partition List

【 声明:版权所有,转载请标明出处,请勿用于商业用途。  联系信箱:[email protected]


题目链接:https://leetcode.com/problems/partition-list/


题意:

给定一个链表和一个x,要求在不改变其在原本链表中相对位置的情况下,将小于x的结点放在新链表的左边,大于等于x的结点放在新链表的右边


思路:

思路很简单,新建两个链表,一个存放大于等于x的结点,一个存放小于x的结点,左后再合并两个链表即可。


/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution
{
	public:
		ListNode* partition(ListNode* head, int x)
		{
			if(head==nullptr) return nullptr;
			ListNode *big = new ListNode(0);
			ListNode *small = new ListNode(0);
			ListNode *pa = big;
			ListNode *pb = small;
			ListNode *cur = head;
			while(cur)
			{
				ListNode *pnext = cur->next;
				if(cur->val>=x)
				{
					pa->next = cur;
					pa = pa->next;
					pa->next = nullptr;
				}
				if(cur->val<x)
				{
					pb->next = cur;
					pb = pb->next;
					pb->next = nullptr;
				}
				cur = pnext;
			}
			pb->next = big->next;
			return small->next;
		}
};


你可能感兴趣的:(leedcode)