排好序,从大到小贪心分配即可,比如当一个数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;
}