<—点个赞吧QwQ
宣传一下算法提高课整理
今天是贝茜的生日,为了庆祝自己的生日,贝茜邀你来玩一个游戏.
贝茜让 $N$ 头奶牛(编号 $1$ 到 $N$)坐成一个圈。
除了 $1$ 号与 $N$ 号奶牛外,$i$ 号奶牛与 $i-1$ 号和 $i+1$ 号奶牛相邻,$N$ 号奶牛与 $1$ 号奶牛相邻。
农夫约翰用很多纸条装满了一个桶,每一张纸条中包含一个 $1$ 到 $1000000$ 之间的数字。
接着每一头奶牛 $i$ 从桶中取出一张纸条,纸条上的数字用 $A_i$ 表示。
所有奶牛都选取完毕后,每头奶牛轮流走上一圈,当走到一头奶牛身旁时,如果自己手中的数字能够被该奶牛手中的数字整除,则拍打该牛的头。
牛们希望你帮助他们确定,每一头奶牛需要拍打的牛的数量。
即共有 $N$ 个整数 $A_1,A_2,…,A_N$,对于每一个数 $A_i$,求其他的数中有多少个是它的约数。
输入格式
第一行包含整数 $N$。
接下来 $N$ 行,每行包含一个整数 $A_i$。
输出格式
共 $N$ 行,第 $i$ 行的数字为第 $i$ 头牛需要拍打的牛的数量。
数据范围
$1 \le N \le 10^5$,
$1 \le A_i \le 10^6$
输入样例:
5
2
1
2
3
4
输出样例:
2
0
2
1
3
思路
如果我们直接考虑因数暴力枚举,时间复杂度是$O(n^2)$的,肯定会$TLE$。
那我们考虑倍数,如果存在一个数$x$,那么$2\times x,3\times x\cdots \lfloor \dfrac{max\_num}{x}\rfloor\times x$都是$x$的倍数,所以那些牛都会拍$x$的头。
所以我们从值域出发,不断累加,最终得出答案
代码
#include <iostream>
#include <algorithm>
#include <unordered_map>
using namespace std;
const int N = 100010,M = 1000010;
int n;
int a[N];
int cnt[M];
int ans[M];
int main () {
cin >> n;
for (int i = 1;i <= n;i++) {
cin >> a[i];
cnt[a[i]]++;
}
for (int i = 1;i < M;i++) {
if (cnt[i]) {
for (int j = i;j < M;j += i) ans[j] += cnt[i];
}
}
for (int i = 1;i <= n;i++) cout << ans[a[i]] - 1 << endl; //不包括自己
return 0;
}
大佬 Orz