[UVA11732] strcmp() Anyone? && 字符串

本文介绍了一种使用Trie树解决字符串计数问题的方法,通过左儿子右兄弟表示法来节省空间。代码实现了字符串插入操作,并计算了特定计数值。

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

一个计数问题构造trie解决

此处val域统计的是出现次数 

由于空间紧张需要使用左儿子右兄弟表示方法

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define SF scanf
#define PF printf
#define max(a, b) ((a) < (b) ? (b) : (a))
using namespace std;
typedef long long LL;
const int MAXN = 1000;
const int MAXL = 4000000;
const int SZ = 26;
char s[MAXN+10];
LL ans, Ncnt;
struct Trie_Tree
{
	int son[MAXL+10], bro[MAXL+10], val[MAXL+10], cnt[MAXL+10];
	char ch[MAXL+10];
	void init() { val[0] = son[0] = bro[0] = 0; }
	void insert(char *s)
	{
		ans += val[0];
		val[0]++;
		int u = 0, v, n = strlen(s);
		for(int i = 0; i < n; i++) 
		{
			bool ok = false;
			for(v = son[u]; v; v = bro[v])  if(ch[v] == s[i]) { ok = true; break; }
			if(!ok) {
				v = ++Ncnt;
				val[v] = cnt[v] = son[v] = 0;
				ch[v] = s[i]; bro[v] = son[u]; son[u] = v;
			}
			u = v;
			ans += val[u] * 2;
			val[u]++;
		}
		if(cnt[v]) ans += cnt[v];
		cnt[v]++;
	}
}trie;
int main()
{
	int n;
	int kase = 0;
	char t[110];
	while(SF("%d", &n) && n)
	{
		ans = Ncnt = 0;
		trie.init();
		for(int i = 1; i <= n; i++) SF("%s", s), trie.insert(s);
		PF("Case %d: %lld\n", ++kase, ans);
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值