hdu6128 Inverse of sum 2017多校第七场1009 数学

最新推荐文章于 2020-08-21 23:29:05 发布

ACZone

最新推荐文章于 2020-08-21 23:29:05 发布

阅读量317

点赞数

分类专栏：其他算法文章标签： acm 多校 hdu 数学

本文链接：https://blog.csdn.net/weixin_36571742/article/details/77428671

版权

其他算法专栏收录该内容

6 篇文章 0 订阅

订阅专栏

http://acm.hdu.edu.cn/showproblem.php?pid=6128

题意：给出数组，求数组中有多少数对满足1/(a+b)=1/a+1/b(两边模p)。

代码我是参考：http://blog.csdn.net/ddarkdawn/article/details/77248602

题解：化简有：a*a+a*b+b*b=0(mod p).两边同乘(a-b)，有a^3-b^3=0(mod p)。所以记录每个数字x^3%p的结果，相加即可。但是需要排除a==b的情况，这个在第一个式子中判断即可。

代码：

#include<bits/stdc++.h>
#define debug cout<<"aaa"<<endl
#define d(a) cout<<a<<endl
#define mem(a,b) memset(a,b,sizeof(a))
#define LL long long
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
#define MIN_INT (-2147483647-1)
#define MAX_INT 2147483647
#define MAX_LL 9223372036854775807i64
#define MIN_LL (-9223372036854775807i64-1)
using namespace std;

const int N = 100000 + 5;
const int mod = 1000000000 + 7;
const double eps = 1e-8;
map<LL,int> num;
map<LL,int> cnt;

LL quick(LL a,LL b,LL m){
	LL ans=0;
	while(b){
		if(b&1){
			ans=(ans+a)%m;
		}
		a=(a+a)%m;
		b>>=1;
	}
	return ans;
}

int main(){
	int t,n;
	LL ans,p,x,val;
	scanf("%d",&t);
	while(t--){
		scanf("%d%lld",&n,&p);
		num.clear(),cnt.clear(),ans=0;
		for(int i=1;i<=n;i++){
			scanf("%lld",&x);
			if(!x) continue;
			//排除a[i]==a[j]时，如果(3*a*a)%p!=0，需要排除多算的部分 
			if(quick(quick(x,x,p),3,p)!=0){
				ans-=cnt[x];
			} 
			val=quick(quick(x,x,p),x,p);
			ans+=num[val],num[val]++;
			cnt[x]++;
		}
		printf("%lld\n",ans);
	}
	return 0;
}