Hat’s Words--字典树 / map

最新推荐文章于 2021-10-19 16:23:47 发布

原创最新推荐文章于 2021-10-19 16:23:47 发布 · 272 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

145 篇文章

订阅专栏

本文深入探讨了两种字符串匹配算法——字典树和map在解决特定问题中的应用。通过实例讲解了如何使用这两种方法来判断一个字符串是否能由给定字符串集合中的两个字符串组合而成，同时提供了详细的代码实现。

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1247

题目大意

给你一些串，其中一些串可能会由给出的串中的两个串组成，让你求出这些串，并按字典序输出。

分析

字典序做法，先把每个串放入树中，然后把这些串排序，遍历每个串，把每个串分成两部分（所有可能都考虑），如果这两个部分都能在字典树中查询到，那么就满足条件。

map做法：直接用map标记，查询直接在map中查。

代码

// 字典树做法 
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int N = 1e5 + 5;
int k, trie[N][30];
bool vis[N];
string s[N];
bool cmp(const string a, const string b) {
	return a < b;
}
void insert(string s) {
	int len = s.length();
	int p = 0;
	for(int i = 0; i < len; i++) {
		int x = s[i] - 'a';
		if(!trie[p][x])
			trie[p][x] = k++;
		p = trie[p][x];
	}
	vis[p] = 1;
}
bool search(string s) {
	int len = s.length();
	int p = 0;
	for(int i = 0; i < len; i++) {
		int x = s[i] - 'a';
		if(!trie[p][x]) return 0;
		p = trie[p][x];
	}
	return vis[p] == 1;
}
int main() {
	k = 1;
	memset(trie, 0, sizeof trie);
	memset(vis, 0, sizeof vis);
	int cnt = 0;
	while(cin >> s[cnt]) {
		insert(s[cnt]);
		cnt++;
	}
	sort(s, s + cnt, cmp);
	for(int i = 0; i < cnt; i++) {
		int len = s[i].length();
		string s1, s2;
		for(int j = 0; j < len - 1; j++) {
			s1 = s[i].substr(0, j + 1);
			s2 = s[i].substr(j + 1, len - j);
			if(search(s1) && search(s2)) {
				cout << s[i] << endl;
				break;
			}
		}
	}
	return 0;
}

// map做法 
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <tr1/unordered_map>
using namespace std;
const int N = 5e4 + 5;
string s[N];
tr1::unordered_map<string, int> mp;
bool cmp(const string a, const string b) {
	return a < b;
}
int main() {
	int cnt = 0;
	while(cin >> s[cnt]) {
		mp[s[cnt]] = 1;
		cnt++;
	}
	sort(s, s + cnt, cmp);
	for(int i = 0; i < cnt; i++) {
		int len = s[i].length();
		string s1, s2;
		for(int j = 0; j < len - 1; j++) {
			s1 = s[i].substr(0, j + 1);
			s2 = s[i].substr(j + 1, len - j);
			if(mp[s1] && mp[s2]) {
				cout << s[i] << endl;
				break;
			}
		}
	}
	return 0;
}