Partition List,拆分链表

问题描述:

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

算法分析:开始我只是想在原来链表上进行操作,但是无法返回头结点。这道题区别链表反转,链表反转不用新建链表,只用在原有的链表上操作就行了。这道题,要新建两个新的链表,一个链表的元素全部小于目标值,另一个链表的元素全部大于目标值。然后把这两个链表连接起来。

public ListNode partition(ListNode head, int x)
	{
		if(head == null || head.next == null)
		{
			return head;
		}
		ListNode lessHead = new ListNode(0);
		ListNode greaterHead = new ListNode(0);
		ListNode less = lessHead, greater = greaterHead;
		ListNode node = head;
		while(node != null)
		{
			ListNode temp = node.next;
			if(node.val < x)
			{
				less.next = node;
				less = less.next;
				less.next = null;
			}
			else
			{
				greater.next = node;
				greater = greater.next;
				greater.next = null;
			}
			node = temp;
		}
		less.next = greaterHead.next;
		return lessHead.next;
	}

 

posted @ 2016-08-24 20:11  32ddd  阅读(225)  评论(0编辑  收藏  举报