codevs 1040 统计单词个数

和上次写的山区建小学有点像，直接放题：

时间限制: 1 s
空间限制: 128000 KB

题目描述 Description

给出一个长度不超过200的由小写英文字母组成的字母串(约定;该字串以每行20个字母的方式输入，且保证每行一定为20个)。
要求将此字母串分成k份(1 < k<=40)， 且每份中包含的单词个数加起来总数最大(每份中包含的单词可以部分重叠。当选用一个单词之后，其第一个字母不能再用。例如字符串this中可包含this和is，选用this之后就不能包含th)（管理员注：这里的不能再用指的是位置，不是字母本身。比如thisis可以算做包含2个is）。
单词在给出的一个不超过6个单词的字典中。
要求输出最大的个数。

输入描述 Input Description

第一行为一个正整数(0 < n<=5)表示有n组测试数据
每组的第一行有二个正整数(p，k)
p表示字串的行数;
k表示分为k个部分。
接下来的p行，每行均有20个字符。
再接下来有一个正整数s，表示字典中单词个数。(1<=s<=6)
接下来的s行，每行均有一个单词。

输出描述 Output Description

每行一个整数，分别对应每组测试数据的相应结果。

样例输入 Sample Input

1
1 3
thisisabookyouareaoh
4
is
a
ok
sab

样例输出 Sample Output

7

数据范围及提示 Data Size & Hint

this/isabookyoua/reaoh

同样是dp，刷完山区建小学，其实就差不多有思路了。
这两题有点像，但是这题复杂很多很多（或许是我写复杂了？）

放重点：
1、预处理：我先把字符串每一位作为开头时，存在的单词记录下来。init[i][0]表示以第i位开头有多少个单词，init[i][j]记录这之中第j个单词具体是哪一个（1≤j≤init[i][[0]）。
2、同样是预处理：set[i][j]表示从第i位到第j位存在多少单词，这里可以用dp，具体见代码
3、dp方程：ans[i][j]表示从第1位到第i位放j个分割线，这里我定义最后一条就在第i位后，所以也可以说分成j-1份，则有ans[i][j]=max(ans[i][j],ans[l][j-1]+set[l+1][i]),且j-1≤l≤i。

代码来啦

代码

#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int p,k,s,init[202][7],set[202][202],ans[202][42];string str,word[7];
int findd(int start,int len){      //计算第start位开头能不能增加单词数，由于一个开头只能用一次，所以一旦有就直接返回1
    for (int i=1;i<=init[start][0];i++){
        int k=init[start][i];
        if (word[k].length()<=len) return 1;
    }
    return 0;
}
int main(){

    int n;
    cin>>n;
    while (n>0){
        n--;str="";
        memset(init,0,sizeof(init));
        memset(set,0,sizeof(set));
        memset(ans,0,sizeof(ans));

        cin>>p>>k;
        for (int i=1;i<=p;i++){
            string s1;
            cin>>s1;
            str+=s1;  //先把字符串合起来
        }

        cin>>s;
        for (int i=1;i<=s;i++) cin>>word[i];

        for (int i=0;i<str.length();i++)
        for (int j=1;j<=s;j++)
        if (str[i]==word[j][0] && str.length()-i>=word[j].length()){ //如果开头字母一样，且空间足够放得下
            string s1=str.substr(i,word[j].length());
            if (s1==word[j]) {
                init[i][0]++;
                init[i][init[i][0]]=j;  //这里具体指向某个单词（即word[j]）
            }
        }
//----------------------------------------用来测试的     
//      for (int i=0;i<str.length();i++)
//      for (int j=1;j<=init[i][0];j++) cout<<"Num "<<i<<" : "<<init[i][j]<<endl;
//----------------------------------------  
        for (int i=str.length();i>=1;i--)
        for (int j=i;j>=1;j--) set[j][i]=set[j+1][i]+findd(j-1,i-j+1); //一个小dp，开头位置每次前移一位，并判断这么移动是否增加了单词数
//----------------------------------------用来测试的
//      for (int i=1;i<=str.length();i++)
//      for (int j=i;j<=str.length();j++) cout<<"from "<<i<<" to "<<j<<" : "<<set[i][j]<<endl;
//----------------------------------------
        for (int i=1;i<str.length();i++) ans[i][1]=set[1][i];

        for (int i=2;i<k;i++)
        for (int j=i;j<str.length();j++)
        for (int l=i-1;l<j;l++) ans[j][i]=max(ans[j][i],ans[l][i-1]+set[l+1][j]); //核心dp

        int out=0;
        for (int i=k;i<str.length();i++) out=max(out,ans[i][k-1]+set[i+1][str.length()]);
        //别忘了算最后那个区间

        cout<<out;
    }
    return 0;
}