3294 [SCOI2016]背单词

题目描述

Lweb 面对如山的英语单词,陷入了深深的沉思,”我怎么样才能快点学完,然后去玩三国杀呢?“。这时候睿智的凤老师从远处飘来,他送给了 Lweb 一本计划册和一大缸泡椒,他的计划册是长这样的:

—————序号 单词—————

1 2......n-2n-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 个单词的情况下,吃最少的泡椒。

输入输出格式

输入格式:

输入一个整数 n ,表示 Lweb 要学习的单词数。

接下来 n 行,每行有一个单词(由小写字母构成,且保证任意单词两两互不相同)1<=n<=100000, 所有字符的长度总和 1<=|len|<=510000

输出格式:

Lweb 吃的最少泡椒数

输入输出样例

输入样例#1: 
2
a
ba
输出样例#1: 
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 }

猜你喜欢

转载自www.cnblogs.com/five20/p/9486778.html