Codeforces1189 E. Count Pairs（数学）

题意：

给定长度为n的序列a，和一个质数p，一个整数k
要求计算有多少个数对(i,j)，满足：

数据范围：n<=3e5，p<=2e9，0<=k<p，保证p是质数

解法：

两层枚举不可取，考虑如何将两个变量拆到两边，然后分别计算再合并

等式两边同时乘上(a_i-a_j)，构造平方差，
式子最后变为：
a(i)⁴-a(j)⁴=a(i)k-a(j)k (mod p)
移项得a(i)⁴-a(i)k=a(j)⁴-a(j)k (mod p)

用map记录一下每种的数量就行了

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=3e5+5;
int a[maxm];
int n,mod,k;
signed main(){
    ios::sync_with_stdio(0);
    cin>>n>>mod>>k;
    for(int i=1;i<=n;i++)cin>>a[i];
    map<int,int>mark;
    for(int i=1;i<=n;i++){
        int temp=((a[i]*a[i]%mod)*(a[i]*a[i]%mod)%mod-a[i]*k%mod+mod)%mod;
        mark[temp]++;
    }
    int ans=0;
    for(auto i:mark){
        int x=i.second;
        ans+=x*(x-1)/2;
    }
    cout<<ans<<endl;
    return 0;
}

---