Codeforces Round #572 (Div. 2) E. Count Pairs（数学公式变形）

题目链接：http://codeforces.com/contest/1189/problem/E

题意：现在有$n$个数，你需要找出有多少对$(i,j)$满足$(a_i+a_j)\ast (a_i^2+a_j^2)\%p=k$。

解题心得：

• 刚开始看到这个题除了暴力没啥思路，但是$n$的范围是$3e5$，暴力是$n^2$的复杂度啊。
• 但是$(a_i+a_j)\ast (a_i^2+a_j^2)$这个公式真的是太熟悉了，情不自禁就想给它加个$(a_i-a_j)$，这样等式两边就变成了$$(a_i-a_j)\ast (a_i+a_j)\ast (a_i^2+a_j^2)\%p=k\ast (a_i-a_j)\%p$$再给它变一下就成了$$(a_i^4-a_j^4)\%p=(k\ast a_i+k\ast a_j)\%p$$再移项就变成了$$(a_i^4-k\ast a_i)\%p=(a_j^4-k\ast a_j)\%p$$这样问题变成了 O { n } O\{n\} O{n}统计$a_i^4-k\ast a_i$值的个数，将值相等的两两组合一下答案就出来了。

---

#include <bits/stdc++.h>
using namespace std;
typedef complex<double> cp;
typedef long long ll;
const ll maxn = 3e5+100;
const double pi = acos(-1);

ll n, num[maxn], p, k;
map <ll, ll> maps;

void init() {
    scanf("%lld%lld%lld", &n, &p, &k);
    for(int i=1;i<=n;i++) scanf("%lld", &num[i]);
}

int main() {
    //    freopen("1.in.txt", "r", stdin);
    init();

    for(int i=1;i<=n;i++) {
        ll ai4 = num[i] * num[i] % p *num[i] %p *num[i] % p;
        ll kai = k * num[i] %p;
        if(ai4 < kai) ai4 += p;
        maps[ai4-kai]++;
    }

    map <ll, ll> :: iterator iter;
    ll ans = 0;
    for(iter=maps.begin();iter!=maps.end();iter++) {
        ll cnt = iter->second;
        ans += cnt*(cnt-1)/2;
    }

    printf("%lld\n", ans);
    return 0;
}