本文共 1141 字,大约阅读时间需要 3 分钟。
给定一个数组 A A A,再给定一组询问,每次询问 A A A中小于等于 k k k的数有多少个。题目保证 A A A里只含非负整数。
先对 A A A排序,然后开一个数组 c c c, c [ i ] c[i] c[i]表示 A A A中等于 i i i的数有多少个,然后再求 c c c的前缀和数组,接着再询问的时候,每次就可以以 O ( 1 ) O(1) 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 ) O(l_A) O(lA)。
转载地址:http://mgcs.baihongyu.com/