LeetCode 86. Partition List

stephen_wong

于 2014-07-23 04:54:26 发布

阅读量723

点赞数

分类专栏： LeetCode LeetCode Problems 文章标签： LeetCode C++

本文链接：https://blog.csdn.net/stephen_wong/article/details/38056381

版权

LeetCode 同时被 2 个专栏收录

186 篇文章 0 订阅

订阅专栏

LeetCode Problems

183 篇文章 2 订阅

订阅专栏

用两条链表维护两个partion, 最后把它们合并即可。

代码：

class Solution 
{
public:
    ListNode *partition(ListNode *head, int x) 
    {
    	ListNode *cur = head;
    	ListNode *p_head[2] = {NULL, NULL}, *p_cur[2] = {NULL, NULL};

    	while (cur != NULL)
    	{
    		int index = (cur->val<x)? 0: 1;
    		if (p_head[index] == NULL)
    		{
    			p_head[index] = p_cur[index] = cur;
    		} else
    		{
    			p_cur[index]->next = cur;
    			p_cur[index] = cur;
    		}
			cur = cur->next;
    	}
		if (p_cur[0] != NULL)
		{
	    	p_cur[0]->next = p_head[1];
		}
		if (p_cur[1] != NULL)
		{
	    	p_cur[1]->next = NULL;
		}

		return p_head[0]!=NULL? p_head[0]: p_head[1];
    }
};