P2926 [USACO08DEC] Patting Heads S

思路分析:

首先使用一个b数组来存储奶牛手上数字出现的次数,比如数字是3,那么b[3]++,数字是2,就b[2]++.然后列举奶牛手上的所有数字i,如果他们的倍数j也在b数组中,就用一个cnt数组记录,即cnt[i*j]++,代表有数字i*j的奶牛可以拍手上数字为i的奶牛的头.举个例子,奶牛数字2,4,6,若i是2,那么它的整数倍为j为2,4,6,8,10......发现奶牛有2,4这两个数字,那么cnt[4]++,cnt[6]++.然后i=4,i=6不断往后推即可......最终输出cnt[奶牛数字]即可.

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10,M=1e6+10;
int a[N],b[M],n,mmaxm,cnt[N];
void solve() {
  cin>>n;
  for(int i=1;i<=n;i++){
  	cin>>a[i];
  	b[a[i]]++;
  	mmaxm=max(mmaxm,a[i]);
  }
  for(int i=1;i<=mmaxm;i++){
  	if(b[i]==0) continue;
  	for(int j=1;j*i<=mmaxm;j++){
  		cnt[i*j]+=b[i];
	  }
  }
  for(int i=1;i<=n;i++){
  	cout<<cnt[a[i]]-1<<"\n";
  }
}
int main() {

	ios::sync_with_stdio(false);
	int tt=1;
	//cin>>tt;
	while(tt--)
		solve();
	return 0;
}

over~