luogu1026_统计单词个数_区间dp

题意

https://www.luogu.org/problemnew/show/P1026

solution

1. 设f[i][j]表示1~i分成j分，最多的单词数
2. 预处理一个sum[i][j]数组，表示i~j包含的单词数
3. 预处理的时候需要注意一个顺序j从大到小枚举，i从j到1枚举
4. 那么sum[i][j]=sum[i+1][j],再看看从i开始有没有新的单词，有就++
5. dp的时候就是区间dp，dp[i][j]表示考虑到第i个字母，分为j段的最大字母数量
6. 转移就是枚举在哪里分段dp[i][j]=dp[l][j-1]+sum[l+1][i]

code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<ctime>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
inline int read(){
	char ch=' ';int f=1;int x=0;
	while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x*f;
}
string s;
string a[10];
int sum[210][210];
int f[210][210];
int q;
bool check(int l,int r)
{
	string x=s.substr(l,r-l+1);
	for(int i=1;i<=q;i++)
	{
		if(x.find(a[i])==0) return true;
	}
	return false;
}
int main()
{
	int n,m;int i,j,k;
	s+='0';
	n=read();m=read();
	for(i=1;i<=n;i++)
	{
		string tmp;
		cin>>tmp;
		s+=tmp;
	}
	int len=s.length()-1;
	q=read();
	for(i=1;i<=q;i++) cin>>a[i];
	for(j=len;j>=1;j--)
	{
		for(i=j;i>=1;i--)
		{
			sum[i][j]=sum[i+1][j];
			if(check(i,j)) sum[i][j]++;
		}
	}
	f[0][0]=0;
	for(i=1;i<=m;i++)
	{
		f[i][i]=f[i-1][i-1]+sum[i][i];
	}
	for(i=1;i<=len;i++)
	{
		f[i][1]=sum[1][i];
	}
	for(i=1;i<=len;i++)
	{
		for(j=1;j<=m&&j<i;j++)
		{
			for(k=j;k<i;k++)
			{
				f[i][j]=max(f[i][j],f[k][j-1]+sum[k+1][i]);
			}
		}
	}
	cout<<f[len][m]<<endl;
	return 0;
}