codeforces(E. Johnny and Grandmaster)贪心

在这里插入图片描述
排好序,从大到小贪心分配即可,比如当一个数K分配给了A集合,那么剩下的数如果凑不出K,那么他们的和就一定小于K,注意判断相等要再加一个模数。
代码:

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

ll mod1=1000000007,mod2=1610621741,mod3=1000000009,n,p;
ll a[1000005];
ll apow(ll a,ll b,ll mod)
{
    ll s=1;
    while(b)
    {
        if(b&1)
        {s=s*a%mod;}
        a=a*a%mod;
        b>>=1;
    }
    return s;
}
void solve()
{
    sort(a+1,a+n+1);
    int i=n-1,flag=0;
    ll p1=apow(p,a[n],mod1);
    ll p2=apow(p,a[n],mod2);
    ll p3=apow(p,a[n],mod3);
    ll p4=0,p5=0,p6=0;
    while(i>=1)
    {
        if(flag==0)
        {
            for(int j=i;j>=1;j--)
            {
                p4=(p4+apow(p,a[j],mod1))%mod1;
                p5=(p5+apow(p,a[j],mod2))%mod2;
                p6=(p6+apow(p,a[j],mod3))%mod3;
                if(p4==p1&&p5==p2&&p6==p3)
                {flag=1;i=j-1;p4=0;p5=0;p6=0;break;}
            }
            if(flag==0){break;}
        }
        else
        {
            p1=apow(p,a[i],mod1);
            p2=apow(p,a[i],mod2);
            p3=apow(p,a[i],mod3);
            flag=0;
            i--;
        }
    }
    if(flag==0)
    {
        ll ans=(p1-p4+mod1)%mod1;
        printf("%lld\n",ans);
    }
    else
    {printf("0\n");}
}
int main()
{
    ll t;
    scanf("%lld",&t);
    while(t--)
    {
        scanf("%lld %lld",&n,&p);
        for(int i=1;i<=n;i++)
        {scanf("%lld",&a[i]);}
        if(p==1)
        {
            if(n%2==0){printf("0\n");}
            if(n%2!=0){printf("1\n");}
            continue;
        }
        solve();
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值