本文共 1141 字，大约阅读时间需要 3 分钟。

题目地址：

给定一个数组$A$，再给定一组询问，每次询问$A$中小于等于$k$的数有多少个。题目保证$A$里只含非负整数。

先对$A$排序，然后开一个数组$c$，$c[i]$表示$A$中等于$i$的数有多少个，然后再求$c$的前缀和数组，接着再询问的时候，每次就可以以$O(1)$的时间询问出来了。代码如下：

public class Solution {       /**     * @param nums:     * @param sub:     * @return: return a Integer array     */    public int[] SimpleQueries (int[] nums, int[] sub) {           // write your code here        int[] res = new int[sub.length];                int max = 0;        for (int num : nums) {               max = Math.max(max, num);        }                int[] count = new int[max + 1];        for (int num : nums) {               count[num]++;        }                int[] preSum = new int[count.length + 1];        for (int i = 0; i < count.length; i++) {               preSum[i + 1] = preSum[i] + count[i];        }            for (int i = 0; i < sub.length; i++) {           	// 要特判询问的数大于最大值的情况            if (sub[i] > max) {                   res[i] = nums.length;            } else {                   res[i] = preSum[sub[i] + 1];            }        }                return res;    }}

时空复杂度$O(l_A)$。

转载地址：http://mgcs.baihongyu.com/