牛客周赛 Round 56（F-不是烤串故事）（z函数）

F-不是烤串故事

Z函数做法

Z函数，也就是拓展kmp，可以在 O(n) 的时间处理出一个字符串a与另一个字符串b的每个后缀字符串的lcp（最长公共前缀）。这道题将字符串s在每个位置都进行了一次翻转，要怎么将z函数应用进去呢？

将s全部翻转一次后，就可以求出t与翻转后的s1的每个后缀的lcp。这个有什么用呢？如果将s的前i个字符翻转，那么翻转后的前i个字符就等于s1的后i个字符。如：s=abcdefg翻转3位得到cbadefg，s1=gfedcba。

所以在翻转i位时，我们就得到了t与翻转后s的前i位的lcp。如果lcp等于i，我们把从i+1位开始相等的字符数量加上就得到了题目所求的lcp。

预处理z函数复杂度O(n)，遍历一遍 i 复杂度 O(n) ，总体复杂度 O(n) 。

代码

int z[N], p[N];

void getz(string s, int n) {
    for (int i = 1; i <= n; i++) z[i] = 0;
    z[1] = n;
    for (int i = 2, l = 0, r = 0; i <= n; i++) {
        if (i <= r) z[i] = min(z[i - l + 1], r - i + 1);
        while (i + z[i] <= n && s[i + z[i]] == s[z[i] + 1]) ++z[i];
        if (i + z[i] - 1 > r) l = i, r = i + z[i] - 1;
    }
}

void exkmp(string a,int n, string b,int m) {
    getz(b, m);
    for (int i = 1; i <= n; i++) p[i] = 0;
    for (int i = 1, l = 0, r = 0; i <= n; i++) {
        if (i <= r) p[i] = min(z[i - l + 1], r - i + 1);
        while (i + p[i] <= n && a[i + p[i]] == b[p[i] + 1]) ++p[i];
        if (i + p[i] - 1 > r) l = i, r = i + p[i] - 1;
    }
}

string a, b;

void solve() {
    int n;cin>>n;
    int m=n;
    cin >> a >> b;
    vector<int> len(n + 2);//求位置i为起点有多少字符相等
    int  num=0;
    for (int i = n; i >= 1; i -- ) {
        if (a[i - 1] == b[i - 1])num++;
        else num = 0;
        len[i] = num;
    }
    reverse(a.begin(), a.end());
    a = ' ' + a, b = ' ' + b;
    exkmp(a,n, b,m);
    int ans=0,pos=1;
    for(int i=1;i<=n;i++){
        int ma = p[n - i + 1];
        if (ma == i) {//翻转的字符都是前缀，把后面相等的字符加上
            ma += len[p[n - i + 1] + 1];
        }
        if (ma > ans) ans = ma, pos = i;
    }
    cout<<ans<<' '<<pos<<endl;

}

Z函数模板

注意字符串的下标是从1开始的 

int z[N], p[N];

void getz(string s, int n) {
    for (int i = 1; i <= n; i++) z[i] = 0;
    z[1] = n;
    for (int i = 2, l = 0, r = 0; i <= n; i++) {
        if (i <= r) z[i] = min(z[i - l + 1], r - i + 1);
        while (i + z[i] <= n && s[i + z[i]] == s[z[i] + 1]) ++z[i];
        if (i + z[i] - 1 > r) l = i, r = i + z[i] - 1;
    }
}

void exkmp(string a,int n, string b,int m) {
    getz(b, m);
    for (int i = 1; i <= n; i++) p[i] = 0;
    for (int i = 1, l = 0, r = 0; i <= n; i++) {
        if (i <= r) p[i] = min(z[i - l + 1], r - i + 1);
        while (i + p[i] <= n && a[i + p[i]] == b[p[i] + 1]) ++p[i];
        if (i + p[i] - 1 > r) l = i, r = i + p[i] - 1;
    }
}