其他分享
首页 > 其他分享> > B. Johnny and Grandmaster 思维+hash

B. Johnny and Grandmaster 思维+hash

作者:互联网

B. Johnny and Grandmaster 思维+hash

题目大意:

题目意思就是给一个长度为n的序列k , 然后呢要求将这些数分为两个集合A、B,使得两个集合差值的绝对值最小,也就是 \(min|\sum{p^{k_i}}-\sum{p^{k_j}}|\)

题解:

这个想到了就是一个简单的思维题,再加一点点哈希,但是不经过仔细思考,对代码构成的思考、代码量的思考和难写程度,那么这个就会成为一个恶心的模拟题。

我就写成了模拟题,bug现在还没有找出来,所以我放弃了,看了题解。

题解的思路很简单也很巧妙:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+10;
const int mod1 = 1e9+7;
const int mod2 = 998244353;
int a[maxn];
typedef long long ll;
ll binpow(ll x,ll k,ll mod){
	ll ans = 1;
	x%=mod;
	while(k){
		if(k&1) ans = ans*x%mod;
		x=(x*x)%mod;
		k>>=1;
	}
	return ans;
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		int n,p;
		scanf("%d%d",&n,&p);
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		sort(a+1,a+1+n);
		ll ans1 = 0,ans2 = 0;
		for(int i=n;i>=1;i--){
			ll tmp1 = binpow(p,a[i],mod1),tmp2 = binpow(p,a[i],mod2);
			if(ans1==0&&ans2==0) ans1 = tmp1,ans2 = tmp2;
			else{
				ans1 = (ans1 - tmp1 + mod1)%mod1;
				ans2 = (ans2 - tmp2 + mod2)%mod2;
			}
		}
		printf("%lld\n", ans1);
	}
	return 0;
}

标签:mod1,mod2,hash,Johnny,int,ans1,ll,Grandmaster,mod
来源: https://www.cnblogs.com/EchoZQN/p/13618618.html