思考了一会儿,发现这题目要一层层的考虑。
即,在插入一个字符串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; }}
新闻热点
疑难解答