Trie树字典树简单运用——T1259:最短前缀

Trie树字典树简单运用——最短前缀

---

例题：计蒜客T1259：最短前缀

---

链接:https://vjudge.net/problem/%E8%AE%A1%E8%92%9C%E5%AE%A2-T1259

题目

一个字符串的前缀是从该字符串的第一个字符起始的一个子串。

例如 “carbon” 的字串是: “c”, “ca”, “car”, “carb”, “carbo”, 和 “carbon”。注意到这里我们不认为空串是子串, 但是每个非空串是它自身的子串. 我们现在希望能用前缀来缩略的表示单词。例如, “carbohydrate” 通常用 “carb” 来缩略表示. 现在给你一组单词, 要求你找到唯一标识每个单词的最短前缀。

在下面的例子中，“carbohydrate” 能被缩略成"carboh", 但是不能被缩略成"carbo" (或其余更短的前缀) 因为已经有一个单词用 “carbo” 开始。

一个精确匹配会覆盖一个前缀匹配，例如，前缀 “car” 精确匹配单词 “car”. 因此 “car” 是 “car” 的缩略语是没有二义性的 , “car” 不会被当成 “carriage” 或者任何在列表中以 “car” 开始的单词。

输入格式

输入包括至少 22 行，至多 10001000 行. 每行包括一个以小写字母组成的单词，单词长度至少是 11，至多是 2020。

输出格式

输出的行数与输入的行数相同。

每行输出由相应行输入的单词开始，后面跟着一个空格接下来是相应单词的没有二义性的最短前缀标识符。

Sample Input

carbohydrate
cart
carburetor
caramel
caribou
carbonic
cartilage
carbon
carriage
carton
car
carbonate

Sample Output

carbohydrate carboh
cart cart
carburetor carbu
caramel cara
caribou cari
carbonic carboni
cartilage carti
carbon carbon
carriage carr
carton carto
car car
carbonate carbona

---

Solution

​ 通过trie字典树，插入每个字符串，在插入过程中用vis[]累加值（意思是：截止某字母，前缀相同的种数）。

​ 查询过程中，依次遍历字符串输出直至vis[]=1为止，即前缀相同的字符串只有1个，符合题意。

Code

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
inline ll read() {
	ll x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9') {
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9') {
		x=(x<<1)+(x<<3)+ch-'0';
		ch=getchar();
	}
	return x*f;
}
const int maxn=1e5+5;
string str[1005];
int trie[maxn][30],tot=1,t=0;
int vis[maxn];
void insert(string s){
	int len=s.size(),root=1;
	for(int k=0;k<len;k++){
		int ch=s[k]-'a';
		if(trie[root][ch]==0) trie[root][ch]=++tot;
		vis[trie[root][ch]]++;
		root=trie[root][ch];
	}
}
void search(string s){
    int len=s.size(),root=1;
    for(int i=0;i<len;i++){
    	cout<<s[i];
    	int ch=s[i]-'a';
    	if(vis[trie[root][ch]]==1) break;
    	root=trie[root][ch];
	}
}
int main(){
	while(cin>>str[++t]){
		insert(str[t]);
	}
	for(int i=1;i<=t;i++){
		cout<<str[i]<<" ";
	    search(str[i]);
	    cout<<endl;
	}
}