SCU 4438 Censor (KMP)

題目大意：

就是現在給出兩個字符串A, B, 長度都不超過500萬, 要求每次將B中第一次A的出現刪除, 然後將B剩下的兩段前後合併, 然後重複這個過程直到沒有可以刪除的A位置, 問最後剩下的串是什麼

大致思路：

首先看這個字符串的長度不難想到用KMP處理出A的next數組

然後考慮在B上進行匹配, 這裏我用的是兩個數組來模擬鏈表, 表示剩下的串, 每次發生刪除時, 將當前位置的指針前移即可, 由於刪除的串不會在之後進行匹配, 整體的時間複雜度是O(|A| + |B|)

代碼如下：

Result  :  Accepted     Memory  :  51820 KB     Time  :  848 ms

/*
 * Author: Gatevin
 * Created Time:  2015/10/1 9:22:45
 * File Name: A.cpp
 */
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;
char a[5000010];
char b[5000010];
int next[5000010];
int len1, len2;
void getNext()
{
    for(int i = 0; i <= len1; i++)
        next[i] = 0;
    for(int i = 1; i < len1; i++)
    {
        int j = i;
        while(j > 0)
        {
            j = next[j];
            if(a[j] == a[i])
            {
                next[i + 1] = j + 1;
                break;
            }
        }
    }
    return;
}
int nex[5000010];
int bef[5000010];
int main()
{
    while(gets(a))
    {
<span style="white-space:pre">	</span>gets(b);
        len1 = strlen(a), len2 = strlen(b);
        if(len1 > len2)
        {
            printf("%s\n", b);
            continue;
        }
        getNext();
        for(int i = 0; i <= len2; i++)
        {
            nex[i] = i + 1;
            bef[i] = i - 1;
        }
        
        for(int i = 0, j = 0; i < len2;)
        {
            if(j < len1 && b[i] == a[j]) j++;
            else
            {
                while(j > 0)
                {
                    j = next[j];
                    if(b[i] == a[j])
                    {
                        j++;
                        break;
                    }
                }
            }
            if(j == len1)
            {
                int ti = i;
                for(int k = 0; k < len1; k++)
                {
                    i = bef[i];
                }
                if(i == -1)
                {
                    bef[nex[ti]] = -1;
                    i = nex[ti];
                    j = 0;
                    continue;
                }
                else
                {
                    nex[i] = nex[ti];
                    bef[nex[ti]] = i;
                    for(int k = 2; k < len1; k++)
                    {
                        if(bef[i] != -1)
                            i = bef[i];
                    }
                    j = 0;
                }
            }
            else i = nex[i];
        }
        int cnt = 0;
        int now = len2;
        while(bef[now] != -1)
        {
            //putchar(b[bef[now]]);
            a[cnt++] = b[bef[now]];
            now = bef[now];
        }
        for(int i = cnt - 1; i >= 0; i--)
            putchar(a[i]);
        putchar('\n');
        
    }
    return 0;
}