Facebook面试题：Sort partial Array

package com.Facebook;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.PriorityQueue;

/**
 * Created by hongruzh on 10/5/16.
 * 改程序的大体思想就是
 * 1. 创建一个对象arrayPair,遍历数组，将每一个segment array(即从小到大的一个片段加入到堆里面，start是起始点，end是结束点)加入到小顶堆里面
 * 2. 每次poll出堆顶元素,nums[pq.poll().start]就是当前未排序的元素最小的一个
 * 3. poll出的arrayPair必须去除start节点，重新构造一个新的array Pair加入到堆里面去
 * 4. 时间复杂度是O(logm) m是树的高度
 */
public class SortPartialArray {
    public int[] mergeKSortedArraySegment(int[] nums){
        int n = nums.length;
        if(n<=1) return nums;
        int start = 0;
        ArrayList<arrayPair> list = new ArrayList<arrayPair>();

        for(int i=1;i<n;i++){
            if(nums[i]<nums[i-1]){
                list.add(new arrayPair(start,i));
                start = i;
            }
        }

        list.add(new arrayPair(start,n));
        if(list.size()==1) return nums;//如果segment只有一个的话，就返回原数组

        // 1,3,5,4,7,9,8,10,12
        PriorityQueue<arrayPair> pq = new PriorityQueue<arrayPair>(new Comparator<arrayPair>() {
            @Override
            public int compare(arrayPair o1, arrayPair o2) {
                return nums[o1.start]-nums[o2.start];//重新写比较器方法，每次是按照arrayPair前一个的nums[start]的大小来排序
            }
        });

        for(int i=0;i<list.size();i++){
            pq.offer(list.get(i));
        }
        System.out.println(pq.size());

        int[] result = new int[n];//构造结果集

        int i = 0;
        while(!pq.isEmpty()){
            arrayPair segment = pq.poll();
            int segmentStart = segment.start;//当时犯了一个愚蠢错误,居然写成了 pq.poll().start，这样的话，不就是把元素给poll出去了吗
            int segmentEnd = segment.end;
//            System.out.println(segment.start);
            result[i++] = nums[segment.start];
            if(segment.addNumberStart(segmentStart, segmentEnd)!=null){//当start = end的时候。这时候创建的segment其实是为null的
                //具体见arrayPair里面的方法，所以只有不为空话，才重新构造一个segment加到堆里面去
                pq.offer(segment.addNumberStart(segmentStart, segmentEnd));
            }

        }


        return result;
    }

    class arrayPair{
        int start;
        int end;
        public arrayPair(int start,int end){
            this.start = start;
            this.end = end;
        }

        public arrayPair addNumberStart(int start,int end){
            if(start+1<end){
                return new arrayPair(start+1,end);
            }
            else return null;
        }
    }
}