2017ACM暑期多校联合训练

题目链接

Problem Description
There are n nonnegative integers a1…n which are less than p. HazelFan wants to know how many pairs i,j(1≤i<j≤n) are there, satisfying 1ai+aj≡1ai+1aj when we calculate module p, which means the inverse element of their sum equals the sum of their inverse elements. Notice that zero element has no inverse element.

Input
The first line contains a positive integer T(1≤T≤5), denoting the number of test cases.
For each test case:
The first line contains two positive integers n,p(1≤n≤105,2≤p≤1018), and it is guaranteed that p is a prime number.
The second line contains n nonnegative integers a1...n(0≤ai<p).

Output
For each test case:
A single line contains a nonnegative integer, denoting the answer.

Sample Input
2
5 7
1 2 3 4 5
6 7
1 2 3 4 5 6

Sample Output
4
6

题意：

给定一个数组a，找出数组a里面所有的满足当(1≤i<j≤n)是，1/(ai+aj)≡1/ai+1/aj的关系有多少对。

分析：

如果暴力遍历整个a数组的话，因为i，j的位置都需要确定，时间复杂度相当于n^2，肯定会超时，所以想办法将上面的式子进行变形，使之变为在O(n)的时间之内可以确定出来结果。

将式子通分后化简可得（ai^2+aj^2+ai*aj）%p=0 。
然后等式两边同时乘上（ai-aj),化简可得(ai^3-aj^3)%p=0。现在的问题就转换为求满足这个关系的对数。

但是直接计算满足这个等式的pair的对数就可以了吗？不是。我们还要考虑到a[i]=a[j]的时候。

当a[i]=a[j]时，(ai^2+aj^2+aiaj)%p=0 可以转换为(a[i]a[i]+a[i]a[i]+a[i]a[i])%p=0%p(因为p是素数)是不满足条件的，但是我们直接计算上面那个式子会把满足这个关系的式子也算进去，所以我们需要把满足a[i]=a[j]即 3a[i]a[j]>0的这些对数减掉。 这样求出来的才是最终的结果。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N=1e5+7;
int t,n;
ll p,a[N];
map<ll,int>hsh;
map<ll,int>cnt;

ll mul(ll a,ll b)///注意这里并不是整数幂，作用时将a连加b次，返回加后的结果
{
    ll an=0;
    while(b)
    {
        if(b&1)an=(an+a)%p;
        b>>=1,a=(a+a)%p;
    }
    return an;
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%lld",&n,&p);
        hsh.clear();
        cnt.clear();
        ll ans=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",a+i);
            if(!a[i])continue;
            if(mul(mul(a[i],a[i]),3))ans-=cnt[a[i]];///判断3*a[i]*a[i]的值是否大于0,大于的话要把之前加上的全部减去
            ll tp=mul(mul(a[i],a[i]),a[i]);///求出的是a[i]^3
            ans+=hsh[tp]++;///当前求出的这个tp值可以于之前的所有的相匹配，匹配过后个数再加，下次匹配时的方案数就是这次加过之后的
            ++cnt[a[i]];///a[i]所对应的值也要更新
        }
        printf("%lld
",ans);
    }
    return 0;
}