洛谷——P1026 统计单词个数

原题链接

这道题的难度实际上并不是很大，只要将其拆分成两个子问题，很容易就找到问题的解法。

从题意来看，很容易知道这是一个动态规划，也很容易想到动态规划的状态是用dp[i][j]表示前i个字母分成j份所包含的最大单词数。那么很容易得到状态转移方程：dp[i][j]=max(dp[r][j-1]+word_dp[r+1][i]，dp[i][j])，word_dp[i][j]表示从i到j的单词数。翻译一下状态转移方程的意思就是：前i个字母分成j份的最大单词数=max(前i个字母分成j份的最大单词数,前r个单词分成j-1份的最大单词数+从第r+1到i个字母的最大单词数)。那么现在的问题就是如何求这个word_dp数组。

实际上，这也是一个动态规划。略有些不同的点在于这个是从后向前看。也就是说word_dp[i][j]=word_dp[i+1][j]+judge(i,j)。judge函数是判断以第i个字母为首字母，第j个字母之前是否存在单词。无论存在几个（至少一个），只要加一即可。因为首字母不能在作为另一个单词的开头。这样就解决出了这个题。

#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
using namespace std;
char w[6][20];
int word_dp[201][201];
// words_dp是上文的dp数组
int words_dp[201][201];
int s;
int p, k;
// 这里个words提前先放一个字符，方便之后动态规划从1开始
string words = "0", word;
bool judge(int j,int i) {
	for (int k = 0; k < s; k++) {
		int x1 = j, x2 = 0;
		while (w[k][x2] != '\0' && x1 <= i) {
			if (w[k][x2] != words[x1])
				break;
			else x2++, x1++;
		}
		if (w[k][x2] == '\0') {
			return true;
		}
	}
	return false;
}
int main()
{
	cin >> p >> k;
	for (int i = 0; i < p; i++) {
		cin >> word;
		words += word;
	}
	cin >> s;
	for (int i = 0; i < s; i++)
		cin >> w[i];

	for (int i = words.size() - 1; i >= 1; i--) {
		for (int j = i; j >= 1; j--) {
			word_dp[j][i] = word_dp[j + 1][i] + judge(j, i);
		}
	}

	
// 初始化
	for (int i = 1; i <= words.size(); i++)
		words_dp[i][1] = word_dp[1][i];
	for (int i = 1; i < words.size(); i++) {
		for (int j = 2; j <= k; j++) {
			for (int r = j ; r < i; r++) {
				words_dp[i][j] = max(word_dp[r + 1][i] + words_dp[r][j - 1], words_dp[i][j]);
			}
		}
	}

	
	cout << words_dp[words.size() - 1][k];
	return 0;

}