题目描述
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 吃的最少泡椒数
输入输出样例
输入样例#1: 复制
2
a
ba
输出样例#1: 复制
2
题解
选择1.告诉你选了就一定不是最优的。
选择2.告诉你如果当前的单词没有后缀,那么你加入它的代价为x。
选择3.告诉你如果当前的单词有后缀,那么你加入它的代价为x-y,y为最接近x的当前单词的后缀的序号。
本题意就是告诉你。
其实3个选择相当于一个选择,1选不了,2和3等价。因为2操作的y=0。
题意就转化为了接水问题。
假如n个人,分别需要\(a_i\)的时间。我们肯定优先从小到大接水。
这样所有人等待的时间最少。
现在问题转化到了一棵树上,就优先跑子树最小的即可。
Ps:我们可以把后缀转化为前缀建字典树,然后转化成一颗正常的树。
Code
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<queue>
using namespace std;
const int N=1e6+5;
int tot,num,idx,n,head[N],size[N];
long long ans=0;
struct node{
int to,nex;
}e[N<<1];
struct tree{
int ch[26],end;
}t[N<<1];
char s[N];
void add(int from,int to){
num++;
e[num].to=to;
e[num].nex=head[from];
head[from]=num;
}
void build(){
int l=strlen(s+1),now=0;
for(int i=l;i>=1;i--){
int j=s[i]-'a';
if(!t[now].ch[j])
t[now].ch[j]=++tot;
now=t[now].ch[j];
}t[now].end=1;
}
void init(int x,int ff){
for(int i=0;i<26;i++){
if(t[x].ch[i]){
if(t[t[x].ch[i]].end){
add(ff,t[x].ch[i]);
init(t[x].ch[i],t[x].ch[i]);
}
else init(t[x].ch[i],ff);
}
}
}
void prep(int x,int ff){
size[x]=1;
for(int i=head[x];i;i=e[i].nex){
int v=e[i].to;
if(v==ff)continue;
prep(v,x);size[x]+=size[v];
}
}
void dfs(int x,int y){
priority_queue<pair<int,int> >q;
idx++; ans+=1ll*(idx-y);int tmp=idx;
for(int i=head[x];i;i=e[i].nex){
int v=e[i].to;
q.push(make_pair(-size[v],v));
}
while(q.size()){dfs(q.top().second,tmp);q.pop();}
}
int main(){
cin>>n;
for(int i=1;i<=n;i++)
scanf("%s",s+1),build();
init(0,0);
prep(0,0);
dfs(0,0);
cout<<ans-1<<endl;
return 0;
}
原文地址:https://www.cnblogs.com/hhh1109/p/10667123.html