leetcode Partition List

最新推荐文章于 2019-07-16 09:33:43 发布

原创最新推荐文章于 2019-07-16 09:33:43 发布 · 262 阅读

0 ·

CC 4.0 BY-SA版权

leetCode 专栏收录该内容

199 篇文章

订阅专栏

本文介绍了一个链表分区算法的实现，该算法将链表中小于给定值x的所有节点放到大于等于x的所有节点之前。通过调整节点连接顺序，实现链表的重新组织。

题目链接

package newcoder;

class ListNode {
      int val;
      ListNode next;
      ListNode(int x) { val = x; }
}

public class Solution {
    public static void main(String args[])
    {
        Solution
        obj=new Solution();
        ListNode h=new ListNode(2);

        h.next=new ListNode(1);
        obj.partition(h, 1);

    }


    public ListNode partition(ListNode head, int x) {
        if(head==null)
        {
            return head;
        }
        ListNode result=null;
        ListNode current=head;
        ListNode bigList=null;


        ListNode resultTal=null;
        ListNode bigTal=null;
        if(head.val>=x)
        {
            bigList=head;
            while(current.next!=null&&current.next.val>=x)
            {

                    current=current.next;

            }
            resultTal=result=current.next;
            bigTal=current;
        }
        else
        {
            result=head;
            while(current.next!=null&&current.next.val<x)
            {

                    current=current.next;

            }
            bigTal=bigList=current.next;
            resultTal=current;
        }
        if(current.next!=null)//current.next 已经处理
        {
            current=current.next.next;
        }
        else
        {
            current=null;
        }


        while(current!=null)
        {
            if(current.val<x)
            {
                resultTal.next=current;
                resultTal=current;
            }
            else
            {
                bigTal.next=current;
                bigTal=current;
            }
            current=current.next;
        }
        if(bigTal!=null)
        {
            bigTal.next=null;
        }
        if(result==null)
        {
            return bigList;
        }
        resultTal.next=bigList;
        return result;
    }
}