首页 > 代码库 > 【bzoj4567】SCOI2016背单词
【bzoj4567】SCOI2016背单词
题号莫名喜感。
倒序建Trie,dfs这棵Trie,贪心一下,每次按照size排序计算贡献就好。
#include<bits/stdc++.h> #define N 100010 #define M 500010 using namespace std; typedef long long ll; ll ans=0; int ch[M][26],val[M],tpos[M],cnt,rt,tot,n,size[M]; char s[M]; vector<int> a[N]; inline bool cmp(int x,int y){return size[x]<size[y];} inline void ins(char *s){ int len=strlen(s+1);int now=rt,v; for(int i=len;i;i--){ v=s[i]-‘a‘; if(!ch[now][v])ch[now][v]=++tot; now=ch[now][v]; } val[now]=1; } void dfs(int u){ if(val[u]){a[tpos[u]].push_back(++cnt);tpos[u]=cnt;} for(int i=0;i<26;i++){ if(ch[u][i]){ tpos[ch[u][i]]=tpos[u]; dfs(ch[u][i]); } } } void dfs2(int u){ size[u]=1;int v; for(int i=0;i<a[u].size();i++){v=a[u][i];dfs2(v);size[u]+=size[v];} } void dfs3(int u){ tpos[u]=++cnt; sort(a[u].begin(),a[u].end(),cmp); for(int i=0;i<a[u].size();i++){ ans+=cnt+1-tpos[u]; dfs3(a[u][i]); } } int main(){ rt=tot=1;scanf("%d",&n); for(int i=1;i<=n;i++){ scanf("%s",s+1);ins(s); } tpos[rt]=cnt=1; dfs(rt);dfs2(1);cnt=0;dfs3(1); cout<<ans<<endl; }
【bzoj4567】SCOI2016背单词
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。