Patting Heads~基础数论问题

题目描述

AC Code

#include <iostream>
using namespace std;
typedef long long ll;
const ll maxn = 1000005;
ll a[maxn] = {0},b[maxn] = {0},cnt[maxn] = {0},MAX = -65535,n;
int main()
{
	cin>>n;
	for(ll i=1;i<=n;i++){
		cin>>a[i];
		b[a[i]]++;
		MAX = max(MAX,a[i]);
	}
	for(ll i=1;i<=MAX;i++){
		if(b[i] == 0)	continue;
		for(ll j=1;j<=MAX/i;j++){
			if(i*j == i)	cnt[i]--;
			if(b[i*j] > 0)	cnt[i*j] += b[i];
		}
	}
	for(ll i=1;i<=n;i++){
		cout<<cnt[a[i]]<<endl;
	}
	return 0;
}

解释

〇本题思路参考与洛谷博主 zhukewen123 。
①使用常规的暴力解法会出现多个测试点TLE，因此我们应该变换策略，使用其他方法。本篇给出的方法如下：

• a数组记录输入数据。
• b数组记录输入数据中某个数据出现了几次。
• cnt数组记录合法的“拍头”次数，即cnt[i]的i位置上的数字可以拍其他牛的头的次数。
• 注意，不能拍自己的头，因此需要加入特判。
• 循环的上限：对于第一层循环遍历b数组，上限应为a中的最大值，对于第二层循环更新“拍头”次数，上限应为最大值MAX/i，因 i*j = MAX 是可能满足的最大界限。

②实现：

• 输入部分：

for(ll i=1;i<=n;i++){
	cin>>a[i];
	b[a[i]]++;
	MAX = max(MAX,a[i]);
}

👆完成三件事情，输入，更新输入数字的出现次数和寻找最大值。

• 算法部分：

for(ll i=1;i<=MAX;i++){
	if(b[i] == 0)	continue;
	for(ll j=1;j<=MAX/i;j++){
		if(i*j == i)	cnt[i]--;
		if(b[i*j] > 0)	cnt[i*j] += b[i];
	}
}

👆举个例子，如果存在数字2于a中，那么b数组的2位置＞0，而4，6，8，10，……均为2的倍数，如果4，6，8，10，……位置上也有数字存在，则 cnt[4，6，8，10，……] += b[i]，意味着有几个2，就可以拍几次头。注意，当i*j == i时，不能拍自己的头，但是仍可以拍其他牛头。

• 输出部分：

for(ll i=1;i<=n;i++){
		cout<<cnt[a[i]]<<endl;
	}

注意，输出的索引是a[i]，而不直接是i，因为a[i]记录的是输入的数据，即每个牛拿到的数字，而cnt记录的是拿着某个数字的牛可以拍其他满足条件的牛的头几次。