CodeForces - 1189E Count Pairs

题目大意：
给定n个互不相同的数，一个k和一个质数p，问这n个数中有多少对数满足(ai+aj)(ai^2+aj2)≡k mod p

显然这是一道数学题

对柿子进行化简
(ai + aj)(ai ^2 + aj ^2)(ai - aj)≡k * (ai - aj) mod p
(ai ^2 - aj ^2)(ai ^2 + aj ^2)≡k * ai-k * aj mod p
(ai ^ 4 - aj ^4)≡k * ai-k * aj mod p
ai^4 - k * ai ≡ aj ^ 4 - k * aj mod p
所以我们可以使用快速幂（当然没有太大必要）求出每个ai ^ 4-k * ai，用map去存入数据，每次将答案加上map中第二关键字的值，即当前与它同余的数的个数

代码如下

#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll n,p,k;
inline ll read()
{
	ll w=1,s=0; char ch=getchar();
	while(ch<'0' || ch>'9'){if(ch=='-')w=-1; ch=getchar();}
	while(ch>='0' && ch<='9'){s=s*10+ch-'0'; ch=getchar();}
	return w*s;
}
ll qpow(ll a,ll b)
{
	ll ans=1;
	while(b)
	{
		if(b&1) ans=(ans*a)%p;
		a=(a*a)%p; b>>=1;
	}
	return ans;
}
map<ll,ll> m;
int main()
{
	n=read(); p=read(); k=read();
	ll ans=0;
	for(int i=1;i<=n;i++)
	{
		ll a=read();
		ll tmp=(qpow(a,4)-(k*a)%p+p)%p;
		ans+=m[tmp];
		m[tmp]++;
	}
	printf("%lld",ans);
	return 0;
}