南京网预赛02 BY bly

最新推荐文章于 2021-07-05 09:59:05 发布

diligent-team

最新推荐文章于 2021-07-05 09:59:05 发布

阅读量646

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/u012222742/article/details/11970715

本文分享了一道涉及KMP算法变形与动态规划相结合的问题解决经验。通过调整KMP算法来寻找字符串匹配的位置，并利用动态规划求解最优解。文章记录了从遇到问题到解决问题的心路历程。

刚开始的时候把题目想简单了，交了好多炮，始终是WA，浪费了很多时间。

其实这题就是按规则将kmp做个变形，找出原串中所有可以匹配的起始位置，然后dp一遍即可得到结果。

可能是kmp打的太少，比赛的时候因为在找到匹配时，没有沿失配边往回走一步，结果没有过……

赛后加上一句话就过了，sigh～

#pragma comment(linker, "/STACK:10240000000,10240000000")
#include <cstring>
#include <iostream>
#include <cstdio>
#include <set>
#include <algorithm>
using namespace std;
const int maxn=100010;
int n,m,k;
int f[maxn],vis[maxn];
set<int> s[30];
int p[maxn];
void getfail(int *p)
{
    f[0]=0;
    f[1]=0;
    for(int i=1;i<m;i++)
    {
        int j=f[i];
        while(j && p[i]!=p[j]) j=f[j];
        f[i+1]=p[i]==p[j]?j+1:0;
    }
}
int P[maxn];
int x[maxn],a[maxn];
int judge(int j,int i)
{
    if(j==0)
    {
        P[j]=x[i];
        return 1;
    }
    if(p[j]==-1)
    {
        P[j]=x[i];
        if(P[j]>P[j-1]) return a[j]>a[j-1];
        else if(P[j]==P[j-1]) return a[j]==a[j-1];
        else return a[j]<a[j-1];
    }
    else
    {
        if(P[p[j]]==x[i])
        {
            P[j]=x[i];
            return 1;
        }
        else return 0;
    }
}
void find(int* T)
{
    int j=0;
    for(int i=0;i<n;i++)
    {
        while(j && !judge(j,i)) j=f[j];
        if(judge(j,i)) j++;
        if(j==m)
        {
            vis[i-m+1]=1;
            j=f[j];
        }
        //cout<<j<<endl;
    }
}
int dp[maxn],v[maxn];
int dfs(int u)
{
    if(u==n) return 0;
    if(v[u]) return dp[u];
    v[u]=1;
    dp[u]=dfs(u+1);
    if(vis[u] && u+m<=n) dp[u]=max(dp[u],dfs(u+m)+1);
    return dp[u];
}
int main()
{
    //freopen("data","r",stdin);
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        for(int i=1;i<=k;i++) s[i].clear();
        for(int i=0;i<n;i++) scanf("%d",&x[i]);
        for(int i=0;i<m;i++)
        {
            scanf("%d",&a[i]);
            s[a[i]].insert(i);
        }
        for(int i=1;i<=k;i++)
        {
            if(s[i].empty()) continue;
            int tmp=*s[i].begin();
            for(set<int>::iterator it=s[i].begin();it!=s[i].end();it++)
            {
                if(it==s[i].begin()) p[*it]=-1;
                else p[*it]=tmp;
            }
        }
        //for(int i=0;i<m;i++) cout<<i<<" "<<p[i]<<endl;
        getfail(a);
        memset(vis,0,sizeof(vis));
        find(x);
        //for(int i=0;i<n;i++) cout<<vis[i]<<endl;
        memset(v,0,sizeof(v));
        printf("%d\n",dfs(0));
    }
    return 0;
}