洛谷 P2926 [USACO08DEC] Patting Heads S 信息学奥赛一本通1621：轻拍牛头 算法提高篇 质数 USACO08DEC

1621：轻拍牛头

时间限制: 1000 ms         内存限制: 524288 KB
提交数: 1481     通过数: 598

【题目描述】

原题来自：USACO 2008 Dec. Silver

今天是贝茜的生日，为了庆祝自己的生日，贝茜邀你来玩一个游戏。

贝茜让 N 头奶牛坐成一个圈。除了 1 号与 N 号奶牛外，i 号奶牛与 i−1 号和 i+1 号奶牛相邻，N 号奶牛与 1 号奶牛相邻。农夫约翰用很多纸条装满了一个桶，每一张包含了一个 1 到 106 的数字。

接着每一头奶牛 ii 从桶中取出一张纸条 Ai ，每头奶牛轮流走一圈，同时拍打所有「抽到字条数字是Ai的因数」的牛，然后走回到原来的位置。牛们希望你帮助他们确定，每一头奶牛需要拍打的牛。

【输入】

第一行包含一个整数 N；

接下来第二到第 N+1 行每行包含一个整数 Ai 。

【输出】

第一到第 N 行，第 i 行的输出表示第 i头奶牛要拍打的牛数量。

【输入样例】

5
2
1
2
3
4

【输出样例】

2
0
2
1
3

【提示】

数据范围与提示：

对于全部数据，1≤N≤10^5 。

这道题本以为可以直接暴力，没想到数据太大过不了。。。。。

好吧，那就筛吧

​
#include <bits/stdc++.h>
using namespace std;
int n;
int a[1000001];
int cnt[1000001],ans[1000001];
int maxn;
int main() {
	cin >> n;
	for (int i=1;i<=n;i++) {
		cin >> a[i];
		cnt[a[i]]++;
		maxn=max(maxn,a[i]);
	}
	for(int i=1;i<=maxn;i++){        
		if(cnt[i]>0){
            for(int j=i;j<=maxn;j+=i) {
            	ans[j]+=cnt[i];
			}
        }
    }
    for (int i=1;i<=n;i++) {
    	cout << ans[a[i]]-1 << endl;
	}
	return 0;
}


​

经过“艰苦卓绝”奋斗后，写出了这个代码，但是。。。。。。

超时？？？

我优化了半天，debug也跟踪了n遍，就是过不了。。。。。。

终于，正在slowly be mad的我，发现数据量太大，输入输出就会爆掉，我只好搬出了封存许久的快读快写模板

#include <bits/stdc++.h>
using namespace std;
int n;
int a[1000001];
int cnt[1000001],ans[1000001];
int maxn;
inline int read(){
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-') f=-1;
        ch=getchar();
    }
     while(ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
inline void write(int x){
    if(x<0) {
        putchar('-');
        x=-x;
    }
    if(x>9) write(x/10);
     	putchar(x%10+'0');
 	}
int main() {
	n=read();
	for (int i=1;i<=n;i++) {
		a[i]=read();
		cnt[a[i]]++;
		maxn=max(maxn,a[i]);
	}
	for(int i=1;i<=maxn;i++){        
		if(cnt[i]>0){
            for(int j=i;j<=maxn;j+=i) {
            	ans[j]+=cnt[i];
			}
        }
    }
    for (int i=1;i<=n;i++) {
    	write(ans[a[i]]-1);
    	putchar('\n');
	}
	return 0;
}

不容易啊，终于AC了（泪流满面）

这件事告诉我们，一定要尽量使用快读快写模板。。。。。。。。。。。