P1026 统计单词个数——substr

#include<cstdio>
#include<string>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

string s,ch,a[10];
int sum_dic,p,k;

int sum[210][210];

bool check(int l,int r)
{
    string x;
    x=s.substr(l,r-l+1);
    for(int i=1;i<=sum_dic;i++) 
    {
        if(x.find(a[i])==0) return 1;
    }
    return 0;
}

int f[210][50];

int main()
{
    scanf("%d%d",&p,&k);
    s+="0";
    for(int i=1;i<=p;i++)
    {
        cin>>ch;
        s+=ch;
    }
    int len=s.length()-1;
    scanf("%d",&sum_dic);
    for(int i=1;i<=sum_dic;i++)
    {
        cin>>a[i];
    }
    
    for(int i=len;i>=1;i--)
    {
        for(int j=i;j>=1;j--)
        {
            sum[j][i]=sum[j+1][i];
            if(check(j,i)) sum[j][i]++;
        }
    }
    
    f[0][0]=0;
    for(int i=1;i<=k;i++) f[i][i]=f[i-1][i-1]+sum[i][i];
    
    for(int i=1;i<=len;i++) f[i][1]=sum[1][i];
    
    for(int i=1;i<=len;i++)
    {
        for(int j=1;j<=k&&j<i;j++)
        {
            for(int l=j;l<i;l++)
            {
                f[i][j]=max(f[i][j],f[l][j-1]+sum[l+1][i]);
            }
        }
    }
    
    printf("%d",f[len][k]);
    
    return 0;
}