1361B
题意:给你一个长度为n的数组,然后将这些数分为两堆,使得两堆数的差值最小,和的定义为p^a[i]。
思路:先进行排序,然后从大到小直接暴力求每个数的p次方的值,先把最大的数放在一堆,然后减去下一个数,其结果只会大于0,或者等于0,可以证明,或者说想一下就ok,然后进行分类讨论
如果这个数是大于0的,那么减去下一个数,
如果这个数是等于0的,那么就加上下一个数,
之后的就是黑科技了,因为你这个数是有可能是大于0,但是mod之后变为1了,举个例子比如现在的数是1e9+8,mod之后为1,下一个数为0,减去这个数之后你现在的数为0,但实际上是1e9+7,所以要用到另外一个模数来证明当前数为0或者不为0,另外一个模数最好为一个素数,然后尽可能大就行了。
代码如下:
#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
#define rep(i, a, n) for(int i = a; i <= n; i++)
#define per(i, a, n) for(int i = n; i >= a; i--)
#define IOS std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
#define fopen freopen("file.in","r",stdin);freopen("file.out","w",stdout);
#define fclose fclose(stdin);fclose(stdout);
const int inf = 1e9;
const ll onf = 1e18;
const int maxn = 1e6+10;
const int mod = 1e9+7;
const int MOD = 1e9+3;
inline int read(){
int x=0,f=1;char ch=getchar();
while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
while (isdigit(ch)){x=(x<<3)+(x<<1)+ch-48;ch=getchar();}
return x*f;
}
ll qpow(ll a, ll b, ll p){
ll res = 1;
while(b>0){
if(b&1) res = (res*a)%p;
b>>=1;
a = (a*a)%p;
}
return res;
}
int a[maxn];
inline void cf(){
int t = read();
while(t--){
int n = read(), p = read();
rep(i,1,n) a[i]=read();
sort(a+1, a+1+n);
ll ans1 = 0, ans2 = 0;
per(i,1,n){
if(!ans1&&!ans2){
ans1 = qpow(p, a[i], mod);
ans2 = qpow(p, a[i], MOD);
}else {
ans1 = (ans1 - qpow(p, a[i], mod)+mod)%mod;
ans2 = (ans2 - qpow(p, a[i], MOD)+MOD)%MOD;
}
}
printf("%lld\n", ans1);
}
return ;
}
signed main(){
cf();
return 0;
}