Uva-11732-strcmp() Anyone?

150 篇文章 0 订阅

Trie的一道练习题,但是好像不太像普通的实现方式,受教了。

题目要求求出两两相比较的最少值。

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=4100000;
struct Trie
{
    int head[maxn];
    int nxt[maxn];
    char ch[maxn];
    int tot[maxn];
}t;
long long ans,sz;
void insert(char *s)
{
    int u=0,v,len=strlen(s);
    t.tot[0]++;
    for(int i=0;i<=len;i++)
    {
	bool isFound=false;
	for(v=t.head[u];v!=0;v=t.nxt[v])
	   if(t.ch[v]==s[i])
	   {
	       isFound=true;
	       break;
	   }
	if(!isFound)
	{
	    v=sz++;
	    t.tot[v]=0;
	    t.ch[v]=s[i];
	    t.nxt[v]=t.head[u];
	    t.head[u]=v;
	    t.head[v]=0;
	}
	u=v;
	t.tot[u]++;
    }
}
void DFS(int u,int depth)
{
    if(t.head[u]==0)
	ans+=t.tot[u]*(t.tot[u]-1)*depth;
    else
    {
	int sum=0;
	for(int v=t.head[u];v!=0;v=t.nxt[v])
	    sum+=t.tot[v]*(t.tot[u]-t.tot[v]);
	ans+=sum/2*(2*depth+1);
	for(int v=t.head[u];v!=0;v=t.nxt[v])
	    DFS(v,depth+1);
    }
}
int main()
{
    int cas=1,n;
    while(scanf("%d",&n)&&n)
    {
	sz=1;
	ans=0;
	t.tot[0]=t.head[0]=t.nxt[0]=0;
	char str[1001];
	while(n--)
	{
	    scanf("%s",str);
	    insert(str);
	}
	DFS(0,0);
	printf("Case %d: %lld\n",cas++,ans);
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值