面试题：m个长度为n的ordered array，求top k 个 数字

package com.sinaWeibo.interview;



import java.util.Comparator;

import java.util.Iterator;

import java.util.TreeSet;



/**

 * @Author: weblee

 * @Email: [email protected]

 * @Blog: http://www.cnblogs.com/lkzf/

 * @Time: 2014年10月25日下午5:22:58

 * 

 *************        function description *************** m个长度为n的ordered array，求top

 *        k 个 数字 *

 ****************************************************

 */



public class Top1000From2000Arrays {

    /*

     * 从m个数组中各取一个数，并记录每个数的来源数组，建立一个含k个元素的大根堆。

     * 此时堆顶就是最大的数，取出堆顶元素，并从堆顶元素的来源数组中取下一个数加入堆，再取最大值，一直这样进行k次即可。

     * 时间复杂度：k*log2(20)

     */

    public static int[] maxK(final int k, int[][] arrays) {

    int count = 0;

    Node max = null;

    int[] result = new int[k];



    TreeSet<Node> set = new TreeSet<Node>(new NodeComparator());



    set.add(new Node(arrays[0][0], 0, 0));



    for (int i = 0; i < arrays.length; i++) {

        if (set.first().value < arrays[i][arrays[i].length - 1]) {

        set.add(new Node(arrays[i][arrays[i].length - 1], i,

            arrays[i].length - 1));



        if (set.size() > k) {

            

            

            max = set.first();

            result[count++] = max.value;

            set.remove(set.first());

            set.add(new Node(arrays[max.originX][max.originY - 1], max.originX,

                max.originY - 1));

            

         // testing

            Iterator<Node> iter = set.iterator();

            while (iter.hasNext()) {

            System.out.print(iter.next().value + "\t");

            }

            System.out.println();

            // testing

        }

        }

    }



    while (true) {

        max = set.first();

        result[count++] = max.value;



        if (count == k) {

        break;

        }

        

        set.remove(set.first());

        set.add(new Node(arrays[max.originX][max.originY - 1], max.originX,

            max.originY - 1));



        // testing

        Iterator<Node> iter = set.iterator();

        while (iter.hasNext()) {

        System.out.print(iter.next().value + "\t");

        }

        System.out.println();

        // testing

    }



    return result;

    }



    public static void main(String[] args) {

    int[][] arrays = { { 4, 5, 6, 12 }, { 1, 2, 3, 45 }, { 3, 4, 5, 89 },

        { 101, 102, 103, 189 },{ 11, 12, 13, 289 } };



    int[] result = maxK(4, arrays);



    for (int num : result) {

        System.out.print(num + "\t");

    }

    }



}



class Node {

    int value;

    int originX;

    int originY;



    public Node(int value, int originX, int originY) {

    this.value = value;

    this.originX = originX;

    this.originY = originY;

    }

}



class NodeComparator implements Comparator<Node> {



    @Override

    public int compare(Node o1, Node o2) {

    // TODO Auto-generated method stub

    return -(o1.value - o2.value);

    }



}