codeforces1361B Johnny and Grandmaster

https://codeforces.com/contest/1361/problem/B

这题wa了好久好久好久。。。。天天掉分了，难顶

从大到小考虑幂次,a[n],a[n-1]...a[1]去填

求一个上限up，如果p^up>n，当a[i+1]+a[i]>=up且tmp>0的时候，说明tmp要乘以p^up以上，而且之后a[i-1]...a[1]都是<=a[i]的，就算他们全是a[i]，也无法把tmp减到0，那么就直接后面的全减就行了。

如果当前完成进制转换，也就是tmp*=p^(a[i+1]-a[i])之后，如果tmp>n，那么就算后面全是a[i]，也减不到0

如果tmp=0的时候，就把他变成1，我们保证这个差值始终是个非负整数

特判p=1的情况

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

const int maxl=1e6+10;
const int mod=1e9+7;
const int inf=1e6;

int n,m,cas,k,cnt,tot,up;
int a[maxl],b[maxl];
char s[maxl];
bool in[maxl]; 
ll p,num[maxl],ans;

inline void prework()
{
	scanf("%d%lld",&n,&p);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	if(p==1)
	{
		up=inf+1;
		return;
	}
	num[0]=1;
	for(int i=1;i<=inf;i++)
	{
		num[i]=num[i-1]*p;
		if(num[i]>n)
		{
			up=i;
			break;
		}
	}
} 

inline ll qp(ll a,ll b)
{
	ll ans=1,cnt=a;
	while(b)
	{
		if(b&1)
			ans=ans*cnt%mod;
		cnt=cnt*cnt%mod;
		b>>=1;
	}
	return ans;
}

inline void mainwork()
{
	if(p==1)
	{
		if(n&1)
			ans=1;
		else
			ans=0;
		return;
	}
	sort(a+1,a+1+n);ans=0;
	int lastk=a[n];__int128 tmp=1,one=1;
	for(int i=n-1;i>=1;i--)
	if(lastk-a[i]<=up || tmp==0)
	{
		tmp=tmp*num[lastk-a[i]];
		lastk=a[i];
		if(tmp>n)
		{
			ans=tmp%mod*qp(p,a[i])%mod;
			for(int j=i;j>=1;j--)
				ans=((ans-qp(p,a[j]))%mod+mod)%mod;
			return;
		}
		if(tmp==1)
			tmp=0;
		else if(tmp==0)
			tmp=1;
		else
			tmp=tmp-1;
	}
	else
	{
		ans=tmp%mod*qp(p,lastk)%mod;
		for(int j=i;j>=1;j--)
			ans=((ans-qp(p,a[j]))%mod+mod)%mod;
		return;
	}
	ans=tmp%mod*qp(p,lastk)%mod;
}

inline void print()
{
	printf("%lld\n",ans);
}

int main()
{
	int t=1;
	scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}