找找第k个数在第几堆里

简介:
/**
     * 找找第k个数在第几堆里,例如:
     * int k = 25;
     * int[] array = new int[]{10, 12, 14, 15};
     * 答案是 3 .
     *
     * @param heaps
     * @param k
     * @return
     */
    public static int whereAmI(int[] heaps, int k) {
        if (null == heaps || 0 == heaps.length || 0 >= k) {
            return -1;
        }
        int[] heapRecord = new int[heaps.length];
        int sum = 0;
        for (int i = 0; i < heaps.length; i++) {
            sum += heaps[i];
            heapRecord[i] = sum;
        }
        if (k > sum) {
            return -1;
        }
        return 1 + whereAmI(heapRecord, 0, heaps.length - 1, k);
    }

    private static int whereAmI(int[] heapRecord, int start, int end, int k) {
        if (start >= end) {
            return start;
        }
        int mid = start + (end - start) / 2;
        if (heapRecord[mid] == k) {
            return mid;
        }
        if (heapRecord[mid] < k) {
            return whereAmI(heapRecord, mid + 1, end, k);
        }
        return whereAmI(heapRecord, start, mid - 1, k);
    }
目录
相关文章
|
7月前
和最小的K个数对
和最小的K个数对
|
7月前
|
Java 编译器 C++
位1的个数(C++)
位1的个数(C++)
48 0
输入2个数,计算这2个数的,和商积差余,
输入2个数,计算这2个数的,和商积差余,
92 0
|
机器学习/深度学习 算法
第k个数
第k个数
128 0
|
算法 Java 编译器
20天刷题计划-191. 位1的个数
编写一个函数,输入是一个无符号整数(以二进制串的形式),返回其二进制表达式中数字位数为 '1' 的个数(也被称为汉明重量)。