题目地址:
https://www.lintcode.com/problem/english-software/description
给定一个数组 A A A代表每个人的分数,有一系列询问,问下标 i i i的同学在班里超过了百分之多少的同学(分数一样也视为超过)。如果某个同学分数是 s s s,那么这个百分比可以这样算出: ( ∣ { i ∣ A [ i ] ≤ s } ∣ − 1 ) / l A × 100 % (|\{i|A [i]\le s\}|-1)/l_A\times 100\% (∣{i∣A[i]≤s}∣−1)/lA×100%返回每个询问的答案,省略百分号。为了保持精度,先算乘法。题目保证每个人的分数小于等于 100 100 100。
思路是前缀和。先算一下每个得分的人数,做成一个数组 s s s,使得 s [ i ] s[i] s[i]是得 i i i分的人数,然后算一下 s s s的前缀和,这样就能 O ( 1 ) O(1) O(1)算出得分小于等于某个数 c c c的人数了。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
/**
* @param score: Each student's grades
* @param ask: A series of inquiries
* @return: Find the percentage of each question asked
*/
public List<Integer> englishSoftware(List<Integer> score, List<Integer> ask) {
// write your code here
int[] count = new int[101];
for (int s : score) {
count[s]++;
}
int[] preSum = new int[count.length + 1];
for (int i = 0; i < count.length; i++) {
preSum[i + 1] = preSum[i] + count[i];
}
List<Integer> res = new ArrayList<>();
for (int a : ask) {
// 先得到第a个人的分数是多少
int s = score.get(a - 1);
res.add((preSum[s + 1] - 1) * 100 / score.size());
}
return res;
}
}
时空复杂度 O ( l A ) O(l_A) O(lA)。