373. Find K Pairs with Smallest Sums

    /*
     * 373. Find K Pairs with Smallest Sums
     * 2016-7-16 by Mingyang
     * heap来做，记住Queue<int[]>不是Queue<Integer[]>因为array不是primitive variable
     * 另外一点就是k万一很大也不行，大于heap的size就只能输出那么多
     * 判断空，不要想太多，什么一边为空一边不为空的
     */
     public static  List<int[]> kSmallestPairs(int[] nums1, int[] nums2, int k) {
            int len1=nums1.length;
            int len2=nums2.length;
            List<int[]> res=new ArrayList<int[]>();
             if (len1 == 0 || len2 == 0 || k == 0) {
                return res;
            }
            Queue<int[]> queue = new PriorityQueue<int[]>(len1*len2,
                    new Comparator<int[]>() {
                        @Override
                        public int compare(int[] i1, int[] i2) {
                            return i1[0]+i1[1] - i2[0]-i2[1];
                        }
            });
           for(int i=0;i<nums1.length;i++){
               for(int j=0;j<nums2.length;j++){
                   int[] s=new int[]{nums1[i],nums2[j]};
                   queue.add(s);
               }
           }
           if(k>queue.size())
           k=queue.size();
           while(k>0){
               res.add(queue.poll());
               k--;
           }
           return res;
        }