欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【uva11732】"strcmp()" Anyone?

程序员文章站 2022-04-17 14:44:23
...

题目链接https://vjudge.net/problem/UVA-11732
题解
建一棵trie树,每次经过一个点,该点的计数器+1
插入的时候顺便统计,分字符相同(*2)和不同(*1)讨论一下就好
注意最后一位的比较,可以都赋值为47(’\’)
考虑到串只有4000个但串比较长,需要使用左儿子右兄弟表示法

下面的代码会T,我也没办法啊(一脸无辜),路过的大佬们帮忙看看
时限2s,极限数据3.5s,难道这就是所谓的卡常数?

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;

const int N=4000010;
int son[N],bro[N],val[N];
int T,n;
LL ans;
char str[4010];
char ch[N];
int size;

void insert(char *s)
{
    int len=strlen(s+1),t=1,p;
    val[1]++;
    for(int i=1;i<=len+1;i++)
    {
        for(p=son[t];p;p=bro[p])
            if(ch[p]==s[i]) break;
        if(p==0) 
        {
            p=++size;
            ch[p]=s[i];
            bro[p]=son[t];
            son[p]=0;
            val[p]=1;
            son[t]=p;
            ans+=val[t]-1;
            t=p;
        }
        else ans+=val[t]-1+val[t],val[p++],t=p;
    }
}

int main()
{
//  freopen("in.in","r",stdin);
//  freopen("out1.out","w",stdout);
    while(~scanf("%d",&n)&&n)
    {
        size=1;
        ch[1]=val[1]=bro[1]=son[1]=0;
        ans=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",str+1);
            insert(str);
        }
        printf("Case %d: %lld\n", ++T, ans);
    }
    return 0;
}