bzoj4641 基因改造 [kmp]

Description:
定义两个字符串匹配，存在一种排列的对应关系使得同一位置字符相等，问相等的起始位置。

---

Solution:
看上去就很kmp，但是最先开始还统计了字符出现次数，导致写了很长时间还不对，发现旁边的人写的很短，想了想发现只维护上次出现的位置就行了。结果文件名打爆零了。

两个关键字相同当且仅当上一个该字符在正在匹配的串中出现的位置和当前匹配到的位置的距离相同，预处理一下位置即可。

---

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
int n, m, C;
int a[maxn], b[maxn], nxt[maxn], cnt1[maxn], cnt2[maxn], last1[maxn], last2[maxn], ap1[maxn], ap2[maxn];
vector<int> ans;
int main() {
//  freopen("xjz.in", "r", stdin);
//  freopen("xjz.out", "w", stdout);
    int T;
    scanf("%d%d", &T, &C);
    while(T--) {
        scanf("%d%d", &n, &m);
        ans.clear();
        memset(nxt, 0, sizeof(nxt));
        memset(ap1, 0, sizeof(ap1));
        memset(ap2, 0, sizeof(ap2));
        for(int i = 1; i <= n; ++i) {
            scanf("%d", &a[i]);
            last1[i] = ap1[a[i]];
            ap1[a[i]] = i; 
        }
        for(int i = 1; i <= m; ++i) {
            scanf("%d", &b[i]);
            last2[i] = ap2[b[i]];
            ap2[b[i]] = i;
        }
        for(int i = 2, j = 0; i <= m; ++i) {
            while(j) {
                int l1 = last2[i] >= i - j ? last2[i] : i - j - 1, l2 = last2[j + 1];
                if(i - l1 == j + 1 - l2) {
                    break;
                }
                j = nxt[j];
            }
            ++j;
            nxt[i] = j;
        }
        for(int i = 1, j = 0; i <= n; ++i) {
            ++cnt1[a[i]];
            while(j) {
                int l1 = last1[i] >= i - j ? last1[i] : i - j - 1, l2 = last2[j + 1];
                if(i - l1 == j + 1 - l2) {
                    break;
                }
                j = nxt[j];
            }
            ++j;            
            if(j == m) {
                ans.push_back(i - m + 1);
                j = nxt[j];
            }
        }
        printf("%d\n", ans.size());
        for(int i = 0; i < ans.size(); ++i) {
            printf("%d ", ans[i]);
        }
        puts("");
    }
//  fclose(stdin);
//  fclose(stdout);
    return 0;
}