codeforces1361B 暴力

最新推荐文章于 2022-02-07 13:24:58 发布

原创最新推荐文章于 2022-02-07 13:24:58 发布 · 190 阅读

0 ·

CC 4.0 BY-SA版权

codeforces 同时被 2 个专栏收录

43 篇文章

订阅专栏

水题

11 篇文章

订阅专栏

本文解析了一道算法题1361B，目标是将数组元素分成两堆使它们的特定幂次和之差最小。通过排序和动态调整策略，结合两个模数验证，实现高效求解。

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;
}