题目链接:http://codeforces.com/problemset/problem/691/F
题目大意:给定n个数,再给m个询问,每个询问给一个p,求n个数中有多少对数的乘积≥p
数据范围:2≤n≤10^6, 1≤ai≤3*10^6,1≤m≤10^6, 1≤p≤3*10^6
解题思路:比赛的时候比较naive的思路是把n中的数字排序去了重之后,对于每个p,最多枚举√p步,就能得到答案。而这个naive的思路是O(p√p)的,结果T了。
后来百思不得其解,去看了官方的解答。感觉是一种很有必要总结的思路。思路的模型是埃氏筛素数。
筛素数中枚举到一个素数pr,我们就把MAX范围内的pr的倍数依次搭上标记。这样做看似暴力,实际上复杂度近乎O(n)(其实是O(MAX/p1+MAX/p2..MAX/pk))
回到这道题目,完全可以借鉴上述思路,从1-MAX枚举a,再从1-MAX/a枚举另一半b,记n个数中乘积等于i的对数ans[i],那么就有ans[a*b] += cnt[a]*cnt[b];其中cnt[i]表示n个数中i这个数出现了多少次。仔细分析复杂度的话是O(MAX/1+MAX/2+MAX/3+...+MAX/MAX),事实上,这个东西是接近O(MAXlogMAX)的,这种思想在处理一些数论问题中同样很有借鉴意义。我认为这种思路复杂度的支撑点在于他有一个上限MAX,并且内部的操作是相乘。
最后得到了ans[i]之后取个前缀和就得到了n个数中乘积<p的对数,用总对数一减就得到了答案。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long LL; const int MaxN = 3e6;
int n, m, MaX;
int a[MaxN + ], p[MaxN + ];
LL cnt[MaxN + ], sum[MaxN + ]; int main()
{
while (~scanf("%d", &n)) {
memset(cnt, , sizeof(cnt));
memset(sum, , sizeof(sum));
for (int i = ; i <= n; i++)
scanf("%d", &a[i]), cnt[a[i]]++;
scanf("%d", &m); MaX = -( << );
for (int i = ; i <= m; i++)
scanf("%d", &p[i]), MaX = max(MaX, p[i]);
for (int i = ; i <= MaX; i++)
for (int j = ; i * j <= MaX; j++)
if (i != j) sum[i * j] += cnt[i] * cnt[j];
else sum[i * j] += cnt[i] * cnt[i] - cnt[i];
for (int i = ; i <= MaX; i++) sum[i] += sum[i - ];
for (int i = ; i <= m; i++)
printf("%I64d\n", (LL)n * (n - ) - sum[p[i] - ]);
}
}