洛谷 P1026 [NOIP2001 T3] 统计单词个数

题目描述

给出一个长度不超过200的由小写英文字母组成的字母串(约定;该字串以每行20个字母的方式输入，且保证每行一定为20个)。要求将此字母串分成k份(1<k<=40)，且每份中包含的单词个数加起来总数最大(每份中包含的单词可以部分重叠。当选用一个单词之后，其第一个字母不能再用。例如字符串this中可包含this和is，选用this之后就不能包含th)。

单词在给出的一个不超过6个单词的字典中。

要求输出最大的个数。

输入输出格式

输入格式：

每组的第一行有二个正整数(p，k)

p表示字串的行数;

k表示分为k个部分。

接下来的p行，每行均有20个字符。

再接下来有一个正整数s，表示字典中单词个数。(1<=s<=6)

接下来的s行，每行均有一个单词。

输出格式：

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

输入输出样例

输入样例#1：

1 3
thisisabookyouareaoh
4
is
a
ok
sab

输出样例#1：

7

说明

this/isabookyoua/reaoh

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

DP~

预处理出以每个位置为开始的最短结束位置，然后DP。

f[i][j] 表示长度为i的原串切成j分所有的最大单词数。

时间复杂度很惊人啊，不过数据比较小，也还是很快的~

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

int p,k,m,len[7],minn[201],f[201][201];
char tot[201],s[7][201];

bool pd(int u,int v)
{
	if(u+len[v]-1>p) return 0;
	for(int i=0;i<len[v];i++)
	  if(s[v][i]!=tot[u+i]) return 0;
	return 1;
}

int cal(int u,int v)
{
	int ans=0;
	for(int i=u;i<=v;i++)
	  if(minn[i]<=v) ans++;
	return ans;
}

int main()
{
	memset(minn,127,sizeof(minn));
	memset(f,-1,sizeof(f));
	scanf("%d%d",&p,&k);
	for(int i=1;i<=p;i++) scanf("%s",tot+(i-1)*20+1);
	scanf("%d",&m);
	for(int i=1;i<=m;i++) scanf("%s",s[i]),len[i]=strlen(s[i]);
	p*=20;f[0][0]=0;
	for(int i=1;i<=p;i++)
	  for(int j=1;j<=m;j++)
	    if(pd(i,j)) minn[i]=min(minn[i],i+len[j]-1);
	for(int i=1;i<=p;i++)
	  for(int j=0;j<i;j++)
	  {
	  	int kkz=cal(j+1,i);
	  	for(int z=1;z<=k;z++)
	  	  if(f[j][z-1]!=-1) f[i][z]=max(f[i][z],f[j][z-1]+kkz);
	  }
	printf("%d\n",f[p][k]);
	return 0;
}