facebook 面经

FB 面经 数组中等概率随机取K个

2018-09-16  本文已影响0人  Anseis

水池抽样, reservoir sample

static int[] randomPick(int[] array, int k) { 
    int[] res = new int[k];
    Random r = new Random();
    for (int i = 0; i < k; i++) {
      res[i] = array[i];
    }
    for (int i = k; i < array.length; i++) {
      int index = r.nextInt(i);
      if (index < k) {
        res[index] = array[i];
      }
    }
    return res;
  }
上一篇 下一篇

猜你喜欢

热点阅读