hdu-6128 Inverse of sum 推公式

jerans

于 2017-08-19 21:34:15 发布

阅读量400

点赞数

CC 4.0 BY-SA版权

分类专栏： hdu 思维数学规律

本文链接：https://blog.youkuaiyun.com/jerans/article/details/77417369

hdu 同时被 3 个专栏收录

55 篇文章

订阅专栏

思维

55 篇文章

订阅专栏

数学规律

2 篇文章

订阅专栏

本文提供了一道 HDU 6128 的算法题解，通过使用 map 来处理重复元素，并结合自定义的幂运算函数 powadd 实现题目要求。该算法题解适用于 ACM 竞赛及类似算法挑战。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

http://acm.hdu.edu.cn/showproblem.php?pid=6128

很好懂得博客：http://blog.youkuaiyun.com/dormousenone/article/details/77340852

想到乘一个ai-aj真的太强了。。。。。

然后用map注意下重复的元素，搞一搞就可以了

#include<bits/stdc++.h>
using namespace std;
long long p;
long long powadd(long long a,long long b)
{
    long long ans=0;
    while(b)
    {
        if(b&1)
        {
            ans=(ans+a)%p;
        }
        a=(a*2)%p;
        b>>=1;
    }
    return ans;
}
long long a[100005];
map<long long,int>p1,p2,p3;
int main()
{
    int T;
    cin>>T;
    int n,i;
    while(T--)
    {
        p1.clear();
        p2.clear();
        p3.clear();
        cin>>n>>p;
        for(i=0;i<n;i++)
        {
            scanf("%lld",&a[i]);
        }
        long long ans1=0,ans2=0,ans3=0;
        for(i=0;i<n;i++)
        {
            if(a[i])
            {
                long long now=powadd(powadd(a[i],a[i]),a[i]);
                if((now*3)%p==0)
                    ans1+=p1[a[i]]++;
                ans3+=p3[a[i]]++;
            }
        }
        for(i=0;i<n;i++)
        {
            if(a[i])
            {
                long long now=powadd(powadd(a[i],a[i]),a[i]);
                ans2+=p2[now]++;
            }
        }
        ans2-=ans3;
        if(p<=3)
            cout<<ans2+ans1<<endl;
        else
            cout<<ans2<<endl;
    }
}