背单词(bzoj 4567)
Description
Lweb 面对如山的英语单词,陷入了深深的沉思,“我怎么样才能快点学完,然后去玩三国杀呢?”。这时候睿智
的凤老师从远处飘来,他送给了 Lweb 一本计划册和一大缸泡椒,他的计划册是长这样的:
—————
序号 单词
—————
1
2
……
n-2
n-1
n
—————
然后凤老师告诉 Lweb ,我知道你要学习的单词总共有 n 个,现在我们从上往下完成计划表,对于一个序号为 x
的单词(序号 1...x-1 都已经被填入):
1) 如果存在一个单词是它的后缀,并且当前没有被填入表内,那他需要吃 n×n 颗泡椒才能学会;
2) 当它的所有后缀都被填入表内的情况下,如果在 1...x-1 的位置上的单词都不是它的后缀,那么你吃 x 颗泡
椒就能记住它;
3) 当它的所有后缀都被填入表内的情况下,如果 1...x-1的位置上存在是它后缀的单词,所有是它后缀的单词中
,序号最大为 y ,那么你只要吃 x-y 颗泡椒就能把它记住。
Lweb 是一个吃到辣辣的东西会暴走的奇怪小朋友,所以请你帮助 Lweb ,寻找一种最优的填写单词方案,使得他
记住这 n 个单词的情况下,吃最少的泡椒。
Input
输入一个整数 n ,表示 Lweb 要学习的单词数。接下来 n 行,每行有一个单词(由小写字母构成,且保证任意单
词两两互不相同)1≤n≤100000, 所有字符的长度总和 1≤|len|≤510000
Output
Lweb 吃的最少泡椒数
Sample Input
2
a
ba
a
ba
Sample Output
2
/* 贪心+字典树 首先第一个条件肯定不是最优的,所以我们得让s的后缀都在s前面,我们可以把字符串翻转建trie,删除所有不是字符串尾部的节点,然后就是找一种编号方式。 比较显然的是我们可以通过按照子树大小排序然后贪心dfs序。 */ #include<cstdio> #include<iostream> #include<vector> #include<algorithm> #include<cstring> #define N 100010 #define M 500010 using namespace std; int n,tot=1,a[M][27],id[M],fa[N],head[N],sz[N],f[N],cnt; char s[M]; struct node{int v,pre;}e[N]; vector<pair<int,int> > v[N]; void add(int x,int y){ e[++cnt].v=y;e[cnt].pre=head[x];head[x]=cnt; fa[y]=x; } void dfs(int x,int f){ if(id[x])add(f,id[x]),f=id[x]; for(int i=0;i<=25;i++) if(a[x][i])dfs(a[x][i],f); } void getsz(int x){ sz[x]=1; for(int i=head[x];i;i=e[i].pre){ getsz(e[i].v); sz[x]+=sz[e[i].v]; v[x].push_back(make_pair(sz[e[i].v],e[i].v)); } sort(v[x].begin(),v[x].end()); } void getf(int x){ if(x)f[x]=++tot; for(int i=0;i<v[x].size();i++)getf(v[x][i].second); } int main(){ scanf("%d",&n); for(int i=1;i<=n;i++){ scanf("%s",s+1);int len=strlen(s+1),now=1; for(int j=len;j;j--){ int x=s[j]-'a'; if(!a[now][x])a[now][x]=++tot; now=a[now][x]; } id[now]=i; } dfs(1,0); getsz(0); tot=0; getf(0); long long ans=0; for(int i=1;i<=tot;i++)ans+=(long long)f[i]-f[fa[i]]; cout<<ans; return 0; }