码迷,mamicode.com
首页 > 其他好文 > 详细

OUC 假期

时间:2019-07-13 22:45:20      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:src   space   return   cout   syn   its   define   ace   mic   

技术图片

技术图片

一道数学题,乘以ai - aj 用平方差,O(n)解决

#include<bits/stdc++.h>
#define MAX 300010
using namespace std;
typedef long long ll;
 
ll a[MAX];
 
int main()
{
	ios::sync_with_stdio(false);
    ll n,p,k;
    cin >> n >> p >> k;
    for(int i=1; i<=n; i++){
        cin >> a[i];
        a[i]=(a[i]*a[i]%p*a[i]%p*a[i]%p-k*a[i]%p+p)%p;
    }
    sort(a + 1, a + n + 1);
    ll temp = 0, ans = 0;
    for(int i = 2; i <= n; i++){
        if(a[i-1] == a[i]){
            temp++;
            ans+=temp;
        }
        else temp=0;
    }
    cout << ans << endl;
    return 0;
}

 

OUC 假期

标签:src   space   return   cout   syn   its   define   ace   mic   

原文地址:https://www.cnblogs.com/lightac/p/11182373.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!