UVA 1401 Remember the Word & LA 3942

最新推荐文章于 2019-07-25 17:51:29 发布

Wall_F

最新推荐文章于 2019-07-25 17:51:29 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

分类专栏： UVA（训练指南）

本文链接：https://blog.youkuaiyun.com/Wall_F/article/details/8598357

UVA（训练指南）专栏收录该内容

34 篇文章

订阅专栏

本文介绍了一种基于Trie树的数据结构来解决字符串分解问题的方法。通过构建Trie树存储字典中的单词，进而统计出给定字符串能被分解为字典中单词组合的不同方式的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

大意：给出一个有S个不同单词组成的字典和一个字符串。把这个字符串分解成若干个单词的连接，有多少种方法？

思路：令d[i]表示以字符i结束的字符串的分解方案数，即d[i+j] = sum(d[i]) (j为一个单词的长度)，初始化d[0] = 1;把所有单词组成Trie，然后试着在Trie中查找单词即可。

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

const int maxnode = 4010*100;
const int sigma_size = 26;
const int MAXN = 300010;
const int MOD = 20071027;

char str[MAXN];
int d[MAXN];

struct trie
{
	int ch[maxnode][sigma_size];
	int val[maxnode];
	int sz;
	int idx(char c) { return c - 'a';}
	void init()
	{
		sz = 1;
		memset(ch[0], 0, sizeof(ch[0]));
	}
	void insert(char *s, int v)
	{
		int u = 0, n = strlen(s);
		for(int i = 0; i < n; i++)
		{
			int c = idx(s[i]);
			if(!ch[u][c])
			{
				memset(ch[sz], 0, sizeof(ch[sz]));
				val[sz] = 0;
				ch[u][c] = sz++;
			}
			u = ch[u][c];
		}
		val[u] = v;
	}
	int find(char *s)
	{
		int u = 0;
		for(int i = 0; s[i]; i++)
		{
			int c = idx(s[i]);
			if(!ch[u][c]) return 0;
			u = ch[u][c];
		}
		return val[u];
	}
}Trie;

void read_case()
{
	Trie.init();
	int n;
	scanf("%d", &n);
	while(n--)
	{
		char temp[110];
		scanf("%s", temp);
		Trie.insert(temp, 1);
	}
}

void solve()
{
	read_case();
	memset(d, 0, sizeof(d));
	int len = strlen(str+1);
	d[0] = 1, str[0] = 'x';
	for(int i = 0; str[i]; i++)
	{
		int u = 0;
		for(int j = 1; j <= len; j++)
		{
			int c = Trie.idx(str[i+j]);
			u = Trie.ch[u][c];
			if(!u) break;
			else if(Trie.val[u] == 1)
			{
				d[i+j] += d[i];
				if(d[i+j] >= MOD) d[i+j] %= MOD;
			}
		}
	}
	printf("%d\n", d[len]);
}

int main()
{
	int times = 0;
	while(~scanf("%s", str+1))
	{
		printf("Case %d: ", ++times);
		solve();
	}
	return 0;
}