传说中的谷歌招聘委员会成员之一,从幕后走出来,教你学系统设计!


一亩三分地论坛

 找回密码
 获取更多干活,快来注册
天天打游戏、照样领工资、还办H1B
这份工作你要不要?
把贵司招聘信息放这里
查看: 2555|回复: 8
收起左侧

linkedin的高频题mergeKSortedArray, 要求用iterator写,

[复制链接] |试试Instant~ |关注本帖
yuxrose 发表于 2015-3-15 09:54:11 | 显示全部楼层 |阅读模式

2015(1-3月) 码农类 博士 全职@Linkedin - 内推 - 技术电面 |Other

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看,没有帐号?获取更多干活,快来注册

x
似乎是经典题。我不是太熟悉iterator的操作,写了一下,不知有没有错误,不知有没有人能帮我看看?
先谢谢大家了!

public Iterable<Integer> mergeKSortedIterators(Iterator<Integer>[] iters){. 1point 3acres 璁哄潧
        Queue<Iterator<Integer>> minHeap = new PriorityQueue<Iterator<Integer>>(iters.length);
        List<Integer> result = new ArrayList<Integer>();
        for(Iterator<Integer> iter : iters){
            if(iter.hasNext()){. 涓浜-涓夊垎-鍦帮紝鐙鍙戝竷
                minHeap.add(iter);
            }
        }. 涓浜-涓夊垎-鍦帮紝鐙鍙戝竷
        while(!minHeap.isEmpty()){.1point3acres缃
            Iterator<Integer> curr = minHeap.poll();
            result.add(curr.next());
            if(curr.hasNext()){
                minHeap.add(curr);
            }
        }
        return result;
    }. 鍥磋鎴戜滑@1point 3 acres
   
    public class iteratorComp implements Comparator<Iterator<Integer>>{
        public int compare(Iterator<Integer> a, Iterator<Integer> b){
            int a1 = 0; 鏉ユ簮涓浜.涓夊垎鍦拌鍧.
            int b1 = 0;
            if(a.hasNext()){
                a1 = a.next();
            }
            if(b.hasNext()){
                b1 = b.next();. 鍥磋鎴戜滑@1point 3 acres
            }
            return a1 - b1;
        }
    }. 鐗涗汉浜戦泦,涓浜╀笁鍒嗗湴
. 1point 3acres 璁哄潧


补充内容 (2015-3-15 09:57):
第一行应该是
Queue<Iterator<Integer>> minHeap = new PriorityQueue<Iterator<Integer>>(iters.length,new iteratorComp());

评分

1

查看全部评分

本帖被以下淘专辑推荐:

peace 发表于 2015-3-16 06:13:04 | 显示全部楼层
写comparator的时候如果直接用a.next()的话总感觉好像有点问题,就像stack.pop()了一样,我也不太熟悉iterator的操作,还得大神们来指导啊

评分

1

查看全部评分

回复 支持 反对

使用道具 举报

sodasun 发表于 2015-3-16 06:19:16 | 显示全部楼层
peace 发表于 2015-3-16 06:13.1point3acres缃
写comparator的时候如果直接用a.next()的话总感觉好像有点问题,就像stack.pop()了一样,我也不太熟悉itera ...

直接在minheap里面存所有iterator的next就行了吧,不用比较iterator的大小
回复 支持 反对

使用道具 举报

peace 发表于 2015-3-16 06:49:05 | 显示全部楼层
sodasun 发表于 2015-3-16 06:19. 涓浜-涓夊垎-鍦帮紝鐙鍙戝竷
直接在minheap里面存所有iterator的next就行了吧,不用比较iterator的大小

这样也不行,这样minheap弹出当前最小值以后无法判断它是从哪个iterator出来的,然后就不知道该把哪个iterator的next放入minheap了。我把next值和对应的iterator放在一个class里了,不知道这样可以么
  1.    public Iterable<Integer> mergeKSortedIterators(Iterator<Integer>[] iters){
  2.         Queue<IntIter> minHeap = new PriorityQueue<IntIter>(iters.length, new iteratorComp());
  3.         List<Integer> result = new ArrayList<Integer>();
  4.         for(Iterator<Integer> iter : iters){. 鍥磋鎴戜滑@1point 3 acres
  5.             if(iter.hasNext()){.鏈枃鍘熷垱鑷1point3acres璁哄潧
  6.                 minHeap.add(new IntIter(iter));
  7.             }
  8.         }
  9.         while(!minHeap.isEmpty()){
  10.             IntIter curr = minHeap.poll();.鐣欏璁哄潧-涓浜-涓夊垎鍦
  11.             result.add(curr.next); 鏉ユ簮涓浜.涓夊垎鍦拌鍧.
  12.             if(curr.iter.hasNext()){
  13.                 minHeap.add(new IntIter(curr.iter));
  14.             }
  15.         }. From 1point 3acres bbs
  16.         return result;. 1point 3acres 璁哄潧
  17.     }
  18.     .1point3acres缃
  19.     public class iteratorComp implements Comparator<IntIter>{
  20.         public int compare(IntIter a, IntIter b){
  21.             return a.next - b.next;
  22.         }
  23.     }

  24.     public class IntIter {
  25.         int next;
  26.         Iterator<Integer> iter;.1point3acres缃
  27.         public IntIter(Iterator<Integer> iter) {
  28.             this.next = iter.next();
  29.             this.iter = iter;
  30.         }
  31.     }
复制代码

补充内容 (2015-3-16 06:52): 鏉ユ簮涓浜.涓夊垎鍦拌鍧.
13行可以换成这样:. from: 1point3acres.com/bbs
curr.next = curr.iter.next();
minHeap.add(curr);
回复 支持 反对

使用道具 举报

eval 发表于 2015-3-16 07:22:17 | 显示全部楼层
哈哈,我gg面试被问的这个题,我用python generator写的
回复 支持 反对

使用道具 举报

 楼主| yuxrose 发表于 2015-3-16 07:29:01 | 显示全部楼层
eval 发表于 2015-3-16 07:22
哈哈,我gg面试被问的这个题,我用python generator写的

狗家也面这题?好吧,看来有一箭双雕的效果
回复 支持 反对

使用道具 举报

 楼主| yuxrose 发表于 2015-3-16 07:30:38 | 显示全部楼层
peace 发表于 2015-3-16 06:49
这样也不行,这样minheap弹出当前最小值以后无法判断它是从哪个iterator出来的,然后就不知道该把哪个ite ...

我觉得写个wrapper class可以,我写了一个和你这个非常像的,只是没再Implement的comparator, 在IntIter里面写了个comparable class。
回复 支持 反对

使用道具 举报

parasitew 发表于 2015-3-17 02:33:23 | 显示全部楼层
yuxrose 发表于 2015-3-16 07:30
我觉得写个wrapper class可以,我写了一个和你这个非常像的,只是没再Implement的comparator, 在IntIter ...

用IntIter类挺好的,比我写的简洁多了
回复 支持 反对

使用道具 举报

hli42 发表于 2015-9-12 07:41:19 | 显示全部楼层
为什么还要用一个list来存呢?直接返回PriorityQueue不行吗?
回复 支持 反对

使用道具 举报

本版积分规则

关闭

一亩三分地推荐上一条 /5 下一条

手机版|小黑屋|一亩三分地论坛声明

custom counter

GMT+8, 2017-9-25 03:42

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

快速回复 返回顶部 返回列表