UVA 11732 strcmp() Anyone? Trie的左儿子右兄弟表示法

        思考了一会儿,发现这题目要一层层的考虑。

        即,在插入一个字符串str到Trie的时候,同时更新答案

        两个字符串,在对比到第i个位置不同时,通过的比较次数是(i<<1)+1,注意,此处i是从0开始的。

        假若到最后都相同的话,比较的次数是(length+1)<<1,即length*2+2,要考虑末尾的\0

        抱着这样的想法,我发现普通的Trie并不好维护这个玩意,看了题解才明白。

        没想到在这里遇见了左儿子右兄弟表示法,也是,这个表示法是比较符合这个场景的

        边插入边计算答案,每到一层就+上那些当前位和自己不同的字符串的比较次数,到最后+上和自己相同的比较次数

        用val数组来维护当前通过该点的字符串个数

        总之感觉用法很好,在Trie中插入节点的时候有点数组模拟邻接表的感觉了

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
typedef long long ll;

int n,kcase=1,sz,bro[4000010],son[4000010],val[4000010];
char str[1005],ch[4000010];
void init(),ins(char* str);
ll ans;

int main(){
    ios_base::sync_with_stdio(false);
    while(cin>>n&&n){
        init();
        ans=0;
        for(int i=0;i<n;++i){
            cin>>str;
            ins(str);
        }
        cout<<"Case "<<kcase++<<": "<<ans<<endl;
    }
    return 0;
}

void init(){
    sz=1;
    ch[0]=bro[0]=son[0]=val[0]=0;
}

void ins(char* str){
    int len=strlen(str),u=0,p;
    for(int i=0;i<=len;++i){
        for(p=son[u];p;p=bro[p])
        if(ch[p]==str[i])
            break;

        if(!p){
            p=sz++;
            ch[p]=str[i];
            bro[p]=son[u];
            son[p]=val[p]=0;
            son[u]=p;
        }

        ans+=(val[u]-val[p])*((i<<1)+1);
        if(len==i){
            ans+=val[p]*((i+1)<<1);
            ++val[p];
        }
        val[u]++;
        u=p;
    }
}

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值