题目大意:一个长度为n的数组a,取i和j,i<j,让ai+aj可以整除x,ai-aj可以整除y,输出有多少组i和j可以满足条件。
思路:ai-aj可以整除y,可以表示为ai%y=aj%y,ai+aj可以整除x,可以表示成ai%x=(x-aj%x)%x。也就是可以开一个map<pair<ll,ll>,ll>,用键存储ai%x和ai%y的,用值来存储数量,然后遍历map寻找可以满足条件的aj数量。
#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> pii;
const int N=1e6+10;
int main()
{
ios::sync_with_stdio(NULL);
cin.tie(0),cout.tie(0);
ll t;cin>>t;
while(t--)
{
map<pii,ll> op;
ll n,x,y;cin>>n>>x>>y;
for(int i=0;i<n;i++)
{
ll a;cin>>a;
op[{a%x,a%y}]++;//当前数模x和y对应的值
}
ll ans=0;
for(auto &[a,b]:op)
{
ll x1=(x-a.first)%x,y1=a.second%y;//x1和y1就是可以和当前遍历的map值组成满足条件的键
if(x1==a.first)//如果模数相同,如果那么就是b*b,但是题目要求i和j不能相等,所以不能直接相乘
{
ans=ans+b*(b-1);
}
else
{
ans=ans+b*op[{x1,y1}];
}
}
cout<<ans/2<<endl;//因为题目要求i<j,但是相加的时候没有考虑,就需要在输出的时候除2
}
return 0;
}