牛客挑战赛45——C 友人 题解

题目链接:https://ac.nowcoder.com/acm/contest/8563/C

题解:

首先我们要知道z^k=z+k-2*(z&k);

则只进行一次操作的代价:(r-l+1)*((z^k)+k-z)=2*(r-l+1)(k-z&k).
对于k-z&k,我们要令其尽可能的小,则z要是k的较大的子集。那么我们枚举k的子集并保存下来,对其排序。

然后我们枚举i=0->n,直到等差数列前i个数之和大于y

对于每一个i,我们计算其前i个数之和s[i],为了满足整个数列之和小于y,对于剩下的每一个数,要满足小于等于(y-s[i])/(n-i),即[0,(y-s[i])/(n-i)]

那么我们要在这个区间寻找最大的k的子集,即可对之前存下的k子集的数组进行二分查找,然后计算此时的答案。

具体见代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
typedef long long ll;
ll n,a,d,k,y;
ll s[maxn];
ll si[maxn];
int cnt;
signed main() {
	cin>>n>>a>>d>>k>>y;
	ll x=a;
	ll sub=k;
	ll ans=1e18;
        //枚举k的子集
	do{
	    si[++cnt]=sub;
	    sub=(sub-1)&k;
	} while(sub!=k);
	sort(si+1,si+cnt+1);
        //i=0的情况
	ll minx=(y-s[0])/(n-0);
	int p=upper_bound(si+1,si+1+cnt,minx)-si-1;
	ans=ans=min(ans,1LL*2*(n-0)*(k-si[p]));
        //1-n的情况
	for(int i=1;i<=n;++i) {
		if(i==1) s[i]=x;
		else x+=d,s[i]=s[i-1]+x;
		if(s[i]>y) break;
		ll minx=(y-s[i])/(n-i);
		int p=upper_bound(si+1,si+1+cnt,minx)-si-1;
		ans=ans=min(ans,1LL*2*(n-i)*(k-si[p]));
	}
	cout<<ans<<endl;
}

猜你喜欢

转载自blog.csdn.net/qq_44132777/article/details/109684940