poj3167

這道題看別人代碼看了一天，我去……功夫不行啊！

#include <iostream>
#include <stdio.h>
#include <vector>
#include <string.h>

using namespace std;
const int maxn=25010;
int n,m,s;
vector<int> ans;
int sum[maxn];
int next[maxn];
int tot1[maxn],tot2[maxn];
int a[100010],b[maxn];
int query(int x)
{
    int tmp=0;
    for(int i=1;i<=x;i++) tmp+=sum[i];
    return tmp;
}
int main()
{
    scanf("%d%d%d",&n,&m,&s);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&b[i]);
        tot1[i]=query(b[i]-1);
        tot2[i]=query(b[i]);
        sum[b[i]]++;
    }
    memset(sum,0,sizeof sum);
    int j=0;
    for(int i=2;i<=m;i++)
    {
        while(j&&(tot1[j+1]!=query(b[i]-1) || tot2[j+1]!=query(b[i])))//while：清除sum部分值，直到j=0（匹配到第一個了）或匹配
        {
            for(int k=i-j;k<i-next[j];k++)//(i-next[i]-(i-j))就是已經匹配的長度-1，-1是爲了給下一段匹配留一個（形成匹配至少需要2個）；i-j是這段匹配的開始
                sum[b[k]]--;
            j=next[j];
        }
        if(tot1[j+1]==query(b[i]-1) && tot2[j+1]==query(b[i])) j++;//如果匹配，j++
        next[i]=j;//這個+1後的i=+1後的j，+1前的j或者是不+1的j是next[i-1]
        sum[b[i]]++;//讀取b[i]了，對應的sum+1
    }
    memset(sum,0,sizeof sum);
    j=0;
    for(int i=1;i<=n;i++)
    {
        while(j==m ||(j&& (tot1[j+1]!=query(a[i]-1) || tot2[j+1]!=query(a[i]))))//j==m說明成功匹配了一段，要清除sum部分值；直到匹配才結束while
        {
            for(int k=i-j;k<i-next[j];k++) sum[a[k]]--;
            j=next[j];
        }
        if(tot1[j+1]==query(a[i]-1) && tot2[j+1]==query(a[i])) j++;
        if(j==m) ans.push_back(i-m+1);
        sum[a[i]]++;
    }
    printf("%d\n",ans.size());
    int len=ans.size();
    for(int i=0;i<len;i++) printf("%d\n",ans[i]);
}