【BZOJ3942】Censoring [KMP]

Censoring

Time Limit: 10 Sec  Memory Limit: 128 MB
[Submit][Status][Discuss]

Description

　　有一个S串和一个T串，长度均小于1,000,000，设当前串为U串，然后从前往后枚举S串一个字符一个字符往U串里添加，若U串后缀为T，则去掉这个后缀继续流程。

Input

　　第一行是S串，第二行是T串。

Output

　　输出一行，表示按照操作后得到的串。

Sample Input

　　whatthemomooofun
　　moo

Sample Output

　　whatthefun

HINT

　　串长小于1000000。

Main idea

　　按照S串的顺序加入S串中的字符，一旦出现了一段和T串一样，则删去这一段，求最后得到的串。

Solution

　　运用KMP，我们显然只要先把T串加入到Stack里面，然后再按照S的顺序加入字符，每次求next（next[i]表示s[1…i]中最长的公共前后缀），显然next==T串长度的话删去相应长度即可。

Code

 #include<iostream>
 #include<string>
 #include<algorithm>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<cmath>
 #include<bitset>
 using namespace std;  

 const int ONE=;

 int n,m;
 char a[ONE],b[ONE],Stack[ONE];
 int next[ONE],j,top;

 int get()
 {
         int res=,Q=;char c;
         while( (c=getchar())< || c> )
         if(c=='-')Q=-;
         res=c-;
         while( (c=getchar())>= && c<= )
         res=res*+c-;
         return res*Q;
 }

 void Deal(int n,char a[],int PD)
 {
         for(int i=PD;i<=n;i++)
         {
             Stack[++top] = a[i];
             j=next[top-];
             while(j && Stack[j+] != Stack[top]) j=next[j];
             if(Stack[j+] == Stack[top]) j++;
             next[top] = j;
             if(PD== && next[top]==m) top-=m;
         }
 }

 int main()
 {
         scanf("%s",a+);    n=strlen(a+);
         scanf("%s",b+);    m=strlen(b+);
         Stack[++top]=b[];
         Deal(m,b,);    Deal(n,a,);

         for(int i=m+;i<=top;i++)
             printf("%c",Stack[i]);

 }