洛谷 P1026 统计单词个数 字典树+动态规划

洛谷 P1026 统计单词个数 字典树+动态规划

题目大意：给你一串字符串，求将这个字符串分成k块的，每块单独计算包含单词个数，他们的总和的最大值，每算一个单词之后，该单词的起点不能再被计算另一个单词。

首先建立对已有单词的字典树，为之后的计算区间内的单词树做准备

计算num[ i ] [ j ]，代表i到j区间内，有多少个位置能作为单词的起点

然后初始化dp[ i ] [ i ]和dp[ i ] [ 1 ]

#include <bits/stdc++.h>
using namespace std;
int p,k,s,len;
char str[201],ss[201];
int trie[201][26],tot=0,pos[201];
int dp[201][201],num[201][201];
void build(int t){
    int l=strlen(ss),root=0;
    for(int i=0;i<l;i++){
        int id=ss[i]-'a';
        if(!trie[root][id])
            trie[root][id]=++tot;
        root=trie[root][id];
    }
pos[root]=t;
}//建立字典树
bool sear(int st,int en){
    for(int tt=0;st<=en;st++){
        int id=str[st]-'a';
        tt=trie[tt][id];
        if(!tt)
            return 0;
        else if(pos[tt])
            return 1;
    }
return 0;
}//计算st位置以后的单词是否为已有单词
int main(){
    scanf("%d %d",&p,&k);
    for(int i=0;i<p;i++)
        scanf("%s",str+1+i*20);
    len=strlen(str+1);//从第一个位置开始，方便之后的处理
    scanf("%d",&s);
    for(int i=1;i<=s;i++){
        scanf("%s",ss);
        build(i);
    }
    for(int i=len;i>=1;i--)
        for(int j=i;j>=1;j--){//该循环从后往前，避免了后效性
            if(sear(j,i))
                num[j][i]=num[j+1][i]+1;
            else num[j][i]=num[j+1][i];
        }
    //dp[i][j]代表到第i个位置分成j块的最大单词数
    for(int i=1;i<=k;i++)
        dp[i][i]=dp[i-1][i-1]+num[i][i];//初始化到第i个位置，且分成i块的情况
    for(int i=1;i<=len;i++)  
        dp[i][1]=num[1][i];//初始化到第i个位置，只有一块的情况
    for(int i=1;i<=len;i++)
        for(int j=1;j<=k && j<i ;j++)
            for(int t=j;t<i;t++)
                dp[i][j]=max(dp[i][j],dp[t][j-1]+num[t+1][i]);
    printf("%d",dp[len][k]);
return 0;
}