codeforces1464B. Grime Zoo

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

草啊这题也没猜出结论,i vegetable exploded

当x<y的时候,显然就要01多点,10少点,那么对于?来说,肯定是前一部分全放0,后一部分全放1

具体证明挺难想到的,题解里有

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

const int maxl=3e5+10;

int n,m,k,tot,cas;ll ans;
ll x,y;
int a[maxl];
ll cnt1[maxl],cnt0[maxl],pre[maxl],suf[maxl];
bool vis[maxl];
char s[maxl];

inline void prework()
{
	scanf("%s",s+1);
	n=strlen(s+1);
	scanf("%lld%lld",&x,&y);
	if(x>y)
	{
		swap(x,y);
		for(int i=1;i<=n;i++)
		if(s[i]!='?')
			s[i]='1'+'0'-s[i];
	}
}

inline void mainwork()
{
	for(int i=1;i<=n;i++)
	if(s[i]=='1')
		cnt1[i]=cnt1[i-1]+1,pre[i]=pre[i-1]+x*(i-cnt1[i]);
	else
		cnt1[i]=cnt1[i-1],pre[i]=pre[i-1]+y*cnt1[i];
	for(int i=n;i>=1;i--)
	if(s[i]=='0')
		cnt0[i]=cnt0[i+1]+1,suf[i]=suf[i+1]+x*(n-i+1-cnt0[i]);
	else
		cnt0[i]=cnt0[i+1],suf[i]=suf[i+1]+y*cnt0[i];
	ans=1ll<<61;ll tmp;
	for(int i=0;i<=n;i++)
	{
		tmp=pre[i]+suf[i+1];
		tmp+=x*(i-cnt1[i])*(n-i-cnt0[i+1]);
		tmp+=y*cnt1[i]*cnt0[i+1];
		ans=min(ans,tmp);
	}
}

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

猜你喜欢

转载自blog.csdn.net/liufengwei1/article/details/114424296
zoo