P1481 魔族密码

题目背景

风之子刚走进他的考场，就……

花花：当当当当~~偶是魅力女皇——花花！！^^（华丽出场，礼炮，鲜花）

风之子：我呕……（杀死人的眼神）快说题目！否则……-_-###

题目描述

花花：……咦好冷我们现在要解决的是魔族的密码问题（自我陶醉：搞不好魔族里面还会有人用密码给我和菜虫写情书咧，哦活活，当然是给我的比较多拉*^_*）。

魔族现在使用一种新型的密码系统。每一个密码都是一个给定的仅包含小写字母的英文单词表，每个单词至少包含 $1$ 个字母，至多 $75$ 个字母。如果在一个由一个词或多个词组成的表中，除了最后一个以外，每个单词都被其后的一个单词所包含，即前一个单词是后一个单词的前缀，则称词表为一个词链。例如下面单词组成了一个词链：

• $\text{i}$；
• $\text{int}$；
• $\text{integer}$。

但下面的单词不组成词链：

• $\text{integer}$；
• $\text{intern}$。

现在你要做的就是在一个给定的单词表中取出一些词，组成最长的词链，就是包含单词数最多的词链。将它的单词数统计出来，就得到密码了。

风之子：密码就是最长词链所包括的单词数阿……

输入格式

这些文件的格式是，第一行为单词表中的单词数 $N$（$1\le N\le 2000$），下面每一行有一个单词，按字典顺序排列，中间也没有重复的单词。

输出格式

输出共一行，一个整数，表示密码。

样例 #1

样例输入 #1

5
i
int
integer
intern
internet

样例输出 #1

4

思路

思路1. Trie树

由题目中“词链”的概念想到用Trie树来存储所有的单词。

题中要求最长词链，因此可以枚举以每个单词结尾的词链。由于所有输入的单词都是按照字典顺序排列的，因此以单词str结尾的词链就等于：在str之前输入的，是str的前缀的，所有单词，再加上str本身。

代码

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

const int N = 150010;// 所有输入字符串的总长度，最多2000个单词，每个单词最长75

int n;
int son[N][26];// 26是因为只包含小写字母
// 存的是以当前这个节点结尾的单词有多少个，这道题每个单词都只出现一次
// 因此就代表以当前这个节点结尾的单词是否存在
int cnt[N];
int idx;// 标识Trie树中每个节点
int ans;

// 存储当前字符串的同时，在他之前的字符串中查找他的子串
int insert(char str[]) {
	int p = 0;
	int res = 1;// 词链最短只包含str本身，初始长度为1
	for (int i = 0; str[i]; i++) {
		int u = str[i] - 'a';
		if (!son[p][u]) {
			son[p][u] = ++idx;
		}
		p = son[p][u];
		// 在从根节点往下走的同时，每到一个节点，都判断一下当前是否有以当前这个节点结尾的单词
		// 如果有，就说明找到了一个前缀，词链长度+1
		if (cnt[p] != 0) {
			res++;
		}
	}
	cnt[p]++;
	return res;
}

int main() {
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {
		char str[N];
		scanf("%s", str);
		int res = insert(str);
		ans = max(ans, res);
	}
	printf("%d", ans);
	return 0;
}

时间复杂度

$O(N)$，$N$ 为所有字符串总长度

思路2. 动态规划

其实就是字符串版本的最长上升子序列。

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#define endl '\n'
using namespace std;

const int N = 2010;

int n;
string str[N];
int f[N];

int main() {
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> str[i];
	}
	for (int i = 0; i < n; i++) {

		f[i] = 1;// 直接在这里初始化，最短的上升子序列只包含自己

		for (int j = 0; j < i; j++) {

			// 判断单词 j 是否是单词 i 的前缀
			if (str[i].substr(0, str[j].size()) == str[j]) {

				f[i] = max(f[i], f[j] + 1);
			}
		}
	}
	int ans = 0;
	for (int i = 0; i < n; i++) {
		ans = max(ans, f[i]);
	}
	cout << ans << endl;
	return 0;
}

时间复杂度

$O(n^2)$，$n$ 为字符串总个数，即状态数量

补充

1. 判断前缀的方法

• 可以用strncmp函数，具体用法如下：

#include <iostream>
#include <cstring> // for strncmp

bool isPrefix(char prefix[], char str[]) {
    int prefixLength = strlen(prefix);
    int strLength = strlen(str);

    if (prefixLength > strLength) {
        return false;
    }

    return std::strncmp(prefix, str, prefixLength) == 0;
}

• 也可以使用字符串哈希