uva11732 strcmp 字典树存储

最新推荐文章于 2021-11-21 21:42:55 发布

原创最新推荐文章于 2021-11-21 21:42:55 发布 · 240 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#uva

字典树专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用字典树优化字符串比较的方法，通过构建字典树来减少字符串两两比较所需的次数，并采用左儿子右兄弟表示法提高效率。

字典树，左儿子右孩子

//uva11732
//题目来源：UVa 11732 strcmp() Anyone?
//题意：求若干个字符串两两比较需要的次数 than 和 that 需要比7次 there和the
//需要7次 就是LCP的长度*2+1 特判2个字符串一样的情况abc和abc需要8次
//思路：利用字典树 每个点出现分叉的时候说明有些字符串可以比较出来了 
//总和加上当前深度*和当前节点不一样的兄弟数 链表超时了 学习了左儿子又兄弟表示法
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

#define maxn 4000010

struct node
{
	char y;
    int son; //指向儿子
	int right; //指向右边的兄弟节点
    int sum ; //表示有多少个儿子
}trie[maxn];

int ch[123];
char str[1010];
int n , trie_s ;
long long  sum ;

void init()
{
	trie[0].son = trie[0].right = -1;
	trie[0].sum = 0;
	sum = 0;
	trie_s = 1;
}

void insert1()
{
    int i , j , k = strlen(str);
    //int x , y = 0;
    int u = 0;
	bool flag;
    for(i = 0; i <= k ; i++)
    {
		flag = false;
		for(j = trie[u].son ; j != -1 ; j = trie[j].right) //遍历u的儿子结点 
		{
			if(trie[j].y == str[i])
			{
				flag = true;
				break;
			}
		}
		if(!flag)
		{
			j = trie_s++;  //新开一个结点，作为u的第一个孩子 
			trie[j].right = trie[u].son; //插入到儿子右链表的表头 
			trie[j].y = str[i];
			trie[j].sum = 0;
			trie[j].son = -1;

			trie[u].son = j;
		}

        if(i == k) 
			{ sum += trie[u].sum+trie[j].sum; trie[j].sum++;} //特判结尾 Trie中不会进入结尾字符,需特判统计
        else
            sum += (trie[u].sum+trie[j].sum);
            
        trie[u].sum++;  //u的儿子数量加1 
        u = j;
    }
}

int main()
{
    int cas = 1;
    int i ;
    
    freopen("a.txt","r",stdin) ;
    while(scanf("%d" , &n) && n)
    {
		init();
        for(i = 1; i <= n; i++)
        {
            scanf("%s" , str);
            insert1();
        }
        cout<<"Case "<<cas++<<": ";
        cout<<sum<<endl;
    }
    return 0;
}
/*
2
a
b
4
cat
hat
mat
sir
0
*/