Bzoj4567---背单词

Problem 4567. – [Scoi2016]背单词

4567: [Scoi2016]背单词

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 970   Solved: 387
[ Submit][ Status][ Discuss]

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

Sample Output

2

吐槽

这位出题人的语文水平真的是有待提高，naive的我懵圈了好久

题解

我们可以发现最优解一定是不会出现第一种情况的，也就是说如果访问一个串之前一定要访问这个串的所有后缀。
那么我们把串反过来，转换成前缀，也就是沿着trie走的问题。然后就贪心一波，先走子树小的再走子树大的就行了。

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<vector>
using namespace std;
int n,tot,dfn,id[500005],fa[500005],ch[500005][35],bo[500005];
long long ans;
bool flag[500005];
char s[500005];
vector<pair<int,int> > a[500005];
inline void insert(char *s)
{
    int len=strlen(s+1),c,u=0;
    for (int i=len; i; i--){
        c=s[i]-'a';
        if (ch[u][c]) u=ch[u][c];
        else u=ch[u][c]=++tot;
    }
    bo[u]=1;
}
void build(int x,int f)
{
    if (bo[x]) fa[x]=f,f=x;
    for (int i=0; i<26; i++) if (ch[x][i]) build(ch[x][i],f);
}
void dfs(int x,int f)
{
    id[x]=dfn++; ans+=id[x]-id[f];
    for (int i=0; i<a[x].size(); i++) dfs(a[x][i].second,x);
}
int main()
{
    scanf("%d\n",&n);
    for (int i=1; i<=n; i++){
        scanf("%s",s+1); insert(s);
    }
    build(0,0);
    for (int x=tot; x; x--)
        if (bo[x]) bo[fa[x]]+=bo[x],a[fa[x]].push_back(make_pair(bo[x],x));
    for (int x=0; x<=tot; x++)
        if (bo[x]) sort(a[x].begin(),a[x].end());
    dfn=0; dfs(0,0);
    cout<<ans<<endl;
    return 0;
}