首页 > 代码库 > Codility上的练习 (9)
Codility上的练习 (9)
半质数的定义是恰好两个质数(可以相同)乘积的数,例如 4, 6, 9, 10, 14, 15, 21, 22, 25, 26,都是半质数。给定N,长度为M的等长整数数组P和Q,满足1 ≤ P[K] ≤ Q[K] ≤ N, 求每个区间[P[k], Q[k]]之间有多少个半质数。
函数头部:vector<int> solution(int N, vector<int> &P, vector<int> &Q);
数据范围: N[1..50000] 数组长度M [1..30000]。
要求时间复杂度 O(NlogN + M), 空间复杂度O(N + M)。
分析: 我们可以用筛法筛出质数。 简单筛质数筛法的复杂度为N / 2 + N /3 + N / 5 + .... + N / (不超过N的最大质数) 就是n除以质数的和,由于调和级数的渐近复杂度,知道筛法的时间复杂度是O(NlogN),空间复杂度为O(N)。 筛出质数后,我们同样利用筛法的思想,对每个不超过sqrt(N)的质数,算出判断它的每个倍数是不是半质数……当然这里可以优化,可以用两个质数乘积求出所有半质数。 这部分复杂度可以用O(NlogN)界定。 接下来,我们利用前缀和思想,打出一张O(N)的表tab,表示0..i中有多少个半质数。这个时间复杂度是O(N),然后对于P[i], Q[i]。我们直接用tab[Q[i]] - tab[P[i] - 1]得到结果,M个pair的时间复杂度为O(M)。
代码:
// you can also use includes, for example: // #include <algorithm> vector<int> solution(int N, vector<int> &P, vector<int> &Q) { // write your code in C++98 vector<bool> mark; mark.resize(N + 1, true); mark[0] = mark[1] = false; for (int i = 2; i <= N; ++i) { if (mark[i] && (N / i >= i)) { for (int j = i * i; j <= N; j += i) { mark[j] = false; } } } vector<bool> is; is.resize(N + 1, false); for (int i = 2; N / i >= i; ++i) { if (mark[i]) { for (int j = i * i; j <= N; j += i) { if (mark[j / i]) { is[j] = true; } } } } vector<int> num; num.resize(N + 1); num[0] = 0; for (int i = 1; i <= N; ++i) { num[i] = num[i - 1] + (is[i]?1:0); } vector<int> answer; for (int i = 0; i < P.size(); ++i) { answer.push_back(num[Q[i]] - num[P[i] - 1]); } return answer; }
(2) CountNonDivisible
给定一个整数数组,求每个元素在数组中非约数的个数。
例如:
A[0] = 3
A[1] = 1
A[2] = 2
A[3] = 3
A[4] = 6
A[0] = 3在数组中的非约数有2,6
A[1] = 1在数组中的非约数有3,2,3,6
A[2] = 2在数组中的非约数有3,3,6
A[3] = 3在数组中的非约数有2,6
A[4] = 6在数组中的没有非约数
因此返回数组[2,4,3,2,6]。
函数头部: vector<int> solution(vector<int> &A);
数据范围: 数组长度N [1..50000]
数组元素范围: [1..2 * N]
要求时间复杂度 O(NlogN) 空间复杂度O(N)
这题的关键在于数据范围和数组长度的数量级相同,都是O(N)。我们可以用2 * N的空间统计数组A中每个数出现多少次。我们在用一张表tab[1..2 * N],tab[i]记录数组A中出现的i的非约数有几个。起初我们认为tab[i] = N,即A中所有的数都不是i的约数。然后对于每个A中出现的数,类似筛法,我们求它所有的倍数j,从tab[j]中减掉i在A中出现的次数,因为i是j的约数。时间复杂度主要在筛法那里 是O(2Nlog(2N)) = O(NlogN)。空间复杂度主要是两张表,一张统计数在A出现多少次,一张是tab,因为数据范围是2 * N,所以这个空间还是O(N)的。
代码:
// you can also use includes, for example: // #include <algorithm> vector<int> solution(vector<int> &A) { // write your code in C++98 int n = A.size(), m = (n << 1) | 1; vector<int> num; num.resize(m + 1, 0); for (int i = 0; i < n; ++i) { ++num[A[i]]; } vector<int> answer; answer.resize(m + 1, n); for (int i = 1; i <= m; ++i) { if (num[i]) { for (int j = i; j <= m; j += i) { answer[j] -= num[i]; } } } vector<int> r; for (int i = 0; i < n; ++i) { r.push_back(answer[A[i]]); } return r; }
Codility上的练习 (9)