Codeforces Round #545 (Div. 2)D(KMP,最长公共前后缀,贪心)

Posted ldudxy

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Codeforces Round #545 (Div. 2)D(KMP,最长公共前后缀,贪心)相关的知识,希望对你有一定的参考价值。

#include<bits/stdc++.h>
using namespace std;
const int N=1000007;
char s1[N],s2[N];
int len1,len2;
int nex[N];
int cnt1[7],cnt2[7];
int main(){
    scanf("%s %s",s1+1,s2+1);
    len1=strlen(s1+1);
    len2=strlen(s2+1);
    for(int i=1;i<=len1;i++)
        cnt1[s1[i]-‘0‘]++;
    for(int i=1;i<=len2;i++)
        cnt2[s2[i]-‘0‘]++;
    if(cnt1[1]>=cnt2[1]&&cnt1[0]>=cnt2[0]){//如果个数足够先输出s2
     for(int i=1;i<=len2;i++)
            putchar(s2[i]);
     cnt1[0]-=cnt2[0];
     cnt1[1]-=cnt2[1];
    }
    for(int i=2,j=0;i<=len2;i++){//kmp寻找最长公共前后缀
        while(s2[i]!=s2[j+1]&&j)//这一位不相等就把模式串返回到标记
            j=nex[j];
        if(s2[i]==s2[j+1])//相等就让模式串位置前进,模式串可能为空此时不需要使模式串标记前进,因为模式串里面没有能和主串相等的字符
            j++;
        nex[i]=j;//记录标记,以便以后返回
    }
    while(cnt1[1]&&cnt1[0]){
     int flag=0;
     for(int i=nex[len2]+1;i<=len2;i++){//贪心地从最长公共前后缀的下一位开始输出
      if(cnt1[s2[i]-‘0‘]){
       putchar(s2[i]);
       cnt1[s2[i]-‘0‘]--;
      }
      else{
       flag=1;
                break;
      }
     }
     if(flag)
            break;
    }
    for(int i=1;i<=cnt1[0];i++)//输出剩余的0或1
        putchar(‘0‘);
    for(int i=1;i<=cnt1[1];i++)//输出剩余的0或1
        putchar(‘1‘);
}

以上是关于Codeforces Round #545 (Div. 2)D(KMP,最长公共前后缀,贪心)的主要内容,如果未能解决你的问题,请参考以下文章

Codeforces Round #545 Div. 1自闭记

Codeforces Round #545 (Div. 2) 掉分记

Codeforces Round #545 (Div. 2)D(KMP,最长公共前后缀,贪心)

Codeforces Round #545 (Div. 2)A. Sushi for Two

「日常训练」Woodcutters(Codeforces Round 303 Div.2 C)

Codeforces Round #436 E. Fire(背包dp+输出路径)