一亩三分地论坛

 找回密码
 获取更多干货,去instant注册!

一亩三分地官方iOS手机应用下载
查看: 2180|回复: 8
收起左侧

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

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

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

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

您需要 登录 才可以下载或查看,没有帐号?获取更多干货,去instant注册!

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

public Iterable<Integer> mergeKSortedIterators(Iterator<Integer>[] iters){
        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);
            }. From 1point 3acres bbs
        }
        while(!minHeap.isEmpty()){
            Iterator<Integer> curr = minHeap.poll();
            result.add(curr.next());
            if(curr.hasNext()){
                minHeap.add(curr);
            }
        }
        return result;.鐣欏璁哄潧-涓浜-涓夊垎鍦
    }
   
    public class iteratorComp implements Comparator<Iterator<Integer>>{
        public int compare(Iterator<Integer> a, Iterator<Integer> b){
            int a1 = 0;. from: 1point3acres.com/bbs
            int b1 = 0;
            if(a.hasNext()){
                a1 = a.next();
            }. visit 1point3acres.com for more.
            if(b.hasNext()){
                b1 = b.next();.1point3acres缃
            }. visit 1point3acres.com for more.
            return a1 - b1;. 鍥磋鎴戜滑@1point 3 acres
        }
    }



补充内容 (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 | 显示全部楼层
关注一亩三分地公众号:
Warald_一亩三分地
写comparator的时候如果直接用a.next()的话总感觉好像有点问题,就像stack.pop()了一样,我也不太熟悉iterator的操作,还得大神们来指导啊

评分

1

查看全部评分

回复 支持 反对

使用道具 举报

sodasun 发表于 2015-3-16 06:19:16 | 显示全部楼层
关注一亩三分地微博:
Warald
peace 发表于 2015-3-16 06:13
写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){. 1point3acres.com/bbs
  5.             if(iter.hasNext()){
  6.                 minHeap.add(new IntIter(iter));
  7.             }
  8.         }
  9.         while(!minHeap.isEmpty()){
  10.             IntIter curr = minHeap.poll();. more info on 1point3acres.com
  11.             result.add(curr.next);. 1point 3acres 璁哄潧
  12.             if(curr.iter.hasNext()){
  13.                 minHeap.add(new IntIter(curr.iter));
  14.             }
  15.         }
  16.         return result;
  17.     }
  18.    
  19.     public class iteratorComp implements Comparator<IntIter>{
  20.         public int compare(IntIter a, IntIter b){
  21.             return a.next - b.next;.1point3acres缃
  22.         }
  23.     }
  24. . Waral 鍗氬鏈夋洿澶氭枃绔,
  25.     public class IntIter {
  26.         int next;
  27.         Iterator<Integer> iter;
  28.         public IntIter(Iterator<Integer> iter) {
  29.             this.next = iter.next();
  30.             this.iter = iter;
  31.         }
  32.     }
复制代码
. visit 1point3acres.com for more.
补充内容 (2015-3-16 06:52):.鐣欏璁哄潧-涓浜-涓夊垎鍦
13行可以换成这样:
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不行吗?
回复 支持 反对

使用道具 举报

本版积分规则

请点这里访问我们的新网站:一亩三分地Instant.

Instant搜索更强大,不扣积分,内容组织的更好更整洁!目前仍在beta版本,努力完善中!反馈请点这里

关闭

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

手机版|小黑屋|一亩三分地论坛声明 ( 沪ICP备11015994号 )

custom counter

GMT+8, 2017-4-26 07:49

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

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