Codeforces Round #545 (Div. 2) D 贪心 + kmp

https://codeforces.com/contest/1138/problem/D

题意

两个01串s和t,s中字符能相互交换,问最多能得到多少个(可交叉)的t

题解

• 即将s中的01塞进t中,预处理出next(tlen),然后每次填完移到next(tlen)继续填即可

代码

#include<bits/stdc++.h>

using namespace std;
int sl,pl,i,j,a,b,e,f,nt[500005];
string s,p;

void get_nt(){
	int plen=p.size(),j=0,k=-1;
	nt[0]=-1;
	while(j<plen){
		if(k==-1||p[k]==p[j]){
			k++;j++;
			nt[j]=k;
		}else k=nt[k];
	}
}
int main(){
	cin>>s>>p;
	sl=s.size();pl=p.size();
	get_nt();
	for(i=0;i<pl;i++){
		if(p[i]=='0')a++;
		else b++;
	}
	for(i=0;i<sl;i++){
		if(s[i]=='0')e++;
		else f++;
	}
	if(e>=a&&f>=b){
		e-=a;f-=b;
		for(i=0;i<pl;i++)s[i]=p[i];
		j=nt[pl];
	}else{
		cout<<s<<endl;
		return 0;
	}
	while(1){
		if(!e||!f){
			if(e){
				for(j=0;j<e;j++)s[i++]='0';
				break;
			}else{
				for(j=0;j<f;j++)s[i++]='1';
				break;
			}
		}
		if(p[j]=='0'){
			s[i++]='0';
			j++;
			e--;
		}else{
			s[i++]='1';
			j++;
			f--;
		}
		if(j==pl)j=nt[j];
	}
	cout<<s<<endl;
}