题目描述
Lweb 面对如山的英语单词,陷入了深深的沉思,”我怎么样才能快点学完,然后去玩三国杀呢?“。这时候睿智的凤老师从远处飘来,他送给了 Lweb 一本计划册和一大缸泡椒,他的计划册是长这样的:
—————序号 单词—————
1 2......n-2n-1 n—————
然后凤老师告诉 Lweb ,我知道你要学习的单词总共有 n 个,现在我们从上往下完成计划表,对于一个序号为 x 的单词(序号 1...x-1 都已经被填入):
- 如果存在一个单词是它的后缀,并且当前没有被填入表内,那他需要吃 n*n 颗泡椒才能学会;
- 当它的所有后缀都被填入表内的情况下,如果在 1...x-1 的位置上的单词都不是它的后缀,那么你吃 x 颗泡椒就能记住它;
- 当它的所有后缀都被填入表内的情况下,如果 1...x-1的位置上存在是它后缀的单词,所有是它后缀的单词中,序号最大为 y ,那么你只要吃 x-y 颗泡椒就能把它记住。
Lweb 是一个吃到辣辣的东西会暴走的奇怪小朋友,所以请你帮助 Lweb ,寻找一种最优的填写单词方案,使得他记住这 n 个单词的情况下,吃最少的泡椒。
输入输出格式
输入格式:输入一个整数 n ,表示 Lweb 要学习的单词数。
接下来 n 行,每行有一个单词(由小写字母构成,且保证任意单词两两互不相同)1<=n<=100000, 所有字符的长度总和 1<=|len|<=510000
输出格式:Lweb 吃的最少泡椒数
输入输出样例
2
a
ba
2
Solution:
写这题博客我是真的要无语了,昨晚写了一半结果保安拉闸断电,今早重写完了结果考试断网没发,关键是后面考完我常规操作关机了,这是第三遍写这题博客了。
先是吐槽,题意真的晦涩。
再简述下题意:本题就是给定n个字符串,然后需要确定它们的先后顺序使得总花费最少,对于第i个字符串,花费有3种情况:
1、字符串中有第i个字符串的后缀,且没有排在i之前,花费为i*i
2、字符串中没有第i个字符串的后缀,花费为i
3、字符串中有第i个字符串的后缀且全部排在i之前,花费为i-最近的是它后缀的字符串的排名k
思路:trie+贪心dfs。
首先对于判断一个串是另一个串的后缀,很容易想到fail,自然就能选用AC自动机了,当然本题不需要那么麻烦,我们可以把单词反转,题目就变成了判断前缀,于是就能加入trie树中去做。
贪心的想到,我们要尽可能避免第1种情况,若把空字符当作任意字符串的前缀且排名为0,那么第2种情况可以看作特殊的第3种情况,那么对于一个单词节点,要使花费最小,那么就要让它的最长前缀的排名尽可能接近,我们处理出每个单词节点的最长前缀位置并连边,形成的是一棵以0为根的树,题目转化为给这棵树节点标号且子节点标号要大于父节点标号,然后最小化子节点标号-父节点标号的差的和。
再贪心去想,很显然父子节点标号要尽量差值小,那么每次我们都往当前最小的子树走,并标号,可以保证下次回到初节点去标记其它子树节点时,使得初节点和子节点差值接近。
以这个贪心思想去求,最后只要统计答案就好了。
代码:
1 #include<bits/stdc++.h> 2 #include<ext/pb_ds/assoc_container.hpp> 3 #include<ext/pb_ds/priority_queue.hpp> 4 #define il inline 5 #define ll long long 6 #define For(i,a,b) for(int (i)=(a);(i)<=(b);(i)++) 7 #define Bor(i,a,b) for(int (i)=(b);(i)>=(a);(i)--) 8 using namespace std; 9 using namespace __gnu_pbds; 10 const int N=510005; 11 int n,ch[N][26],cnt,pre[N],num[N]; 12 int to[N],net[N],h[N],Cnt,siz[N]; 13 ll ans; 14 bool ed[N]; 15 char s[N]; 16 struct node{ 17 int u,d; 18 node(int a=0,int b=0){u=a,d=b;} 19 bool operator<(const node &a)const {return d>a.d;} 20 }; 21 22 il void insert(char *s,int id){ 23 int len=strlen(s),p=0,x; 24 Bor(i,0,len-1){ 25 x=s[i]-'a'; 26 if(!ch[p][x])ch[p][x]=++cnt,pre[cnt]=p; 27 p=ch[p][x]; 28 } 29 ed[p]=1,num[id]=p; 30 } 31 32 il void add(int u,int v){to[++Cnt]=v,net[Cnt]=h[u],h[u]=Cnt;} 33 34 il void dfs(int u){ 35 siz[u]=1; 36 for(int i=h[u];i;i=net[i]) 37 dfs(to[i]),siz[u]+=siz[to[i]]; 38 } 39 40 __gnu_pbds::priority_queue<node,less<node>,pairing_heap_tag>q,Q; 41 42 il void cal(int u){ 43 for(int i=h[u];i;i=net[i])q.push(node(to[i],siz[to[i]])); 44 while(!q.empty()){ 45 node x=q.top();q.pop(); 46 num[x.u]=++cnt; 47 cal(x.u); 48 } 49 } 50 51 il void query(int u){ 52 for(int i=h[u];i;i=net[i]){ 53 ans+=num[to[i]]-num[u]; 54 query(to[i]); 55 } 56 } 57 58 int main(){ 59 scanf("%d",&n); 60 For(i,1,n) scanf("%s",s),insert(s,i); 61 For(i,1,n) { 62 int p=pre[num[i]]; 63 while(p&&!ed[p])p=pre[p]; 64 add(p,num[i]); 65 } 66 int p=cnt; 67 memset(num,0,sizeof(num)); 68 dfs(0),cnt=0,cal(0),query(0); 69 cout<<ans; 70 return 0; 71 }