luogu P1026 统计单词个数 序列DP

dp[i][k]表示，在i及i左侧，分成k块，的最大单词数目。

w[i][j]表示，在[i,j]，内部多少个单词在此区间开始和结束。

转移方程为dp[i][k] = max(dp[j][k - 1] + w[j + 1][i])

通过考虑每个单词的结束而非开始，而避开后后效性问题。

如果某个单词是另一个单词的前缀，则另一个单词无需考虑。

字符串处理的有问题，调了挺久的....

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n,k,len,tot,root,p;
int dp[210][50],w[210][210],ch[1000][30];
char s[210],str[210];
bool end[1000];
void add(char *voc)
{
    int l = strlen(voc + 1),u = root;
    for (int i = 1;i <= l;i++)
    {
        if (end[u] == true)
            break;
        if (ch[u][voc[i] - 'a'] != 0)
            u = ch[u][voc[i] - 'a'];
        else
            u = ch[u][voc[i] - 'a'] = ++tot;
    }
    end[u] = true; 
}
int main()
{
    root = tot = 1;
    scanf("%d%d",&n,&k);
    for (int i = 1;i <= n;i++)
        scanf("%s",s + (i - 1) * 20 + 1);
    len = strlen(s + 1);
    scanf("%d",&p);
    for (int i = 1;i <= p;i++)
    { 
        scanf("%s",str + 1);
        add(str);
    } 
    for (int i = 1;i <= len;i++)
    {
        for (int j = i;j <= len;j++)
        {
            w[i][j] += w[i][j - 1];
            int u = root,t = j;
            while (u != 0)
            {
                u = ch[u][s[t] - 'a'];
                if (end[u] == true)
                {
                    w[i][t]++;
                    break;
                }
                t++;
            }
        }
    }
    for (int i = 1;i <= len;i++)
        dp[i][1] = w[1][i];
    for (int o = 2;o <= k;o++)
        for (int i = o;i <= len;i++)
            for (int j = o - 1;j <= i - 1;j++)
                dp[i][o] = max(dp[j][o - 1] + w[j + 1][i],dp[i][o]);
    printf("%d",dp[len][k]);
}