4567: [Scoi2016]背单词

题目链接

题目大意：n个单词，你需要确定一个填的顺序
设现在填第x个单词
1.如果存在一个单词是它的后缀，且还没被填入，代价为n*n；
2.如果不存在一个单词是它的后缀，代价为x；
3.如果存在一个单词是它的后缀，且已填入的是它后缀的单词中序号最大的为y，代价为x-y

题解：操作1的代价过高，所以在一个字符串插入之前必须将他的所有的后缀全部插入，所以对反串建Trie

把root和单词点看成一棵树，问题转化为给树上每个点标号，每个点的标号大于其父亲的编号，每个点代价为他的标号减去其父亲标号，最小化代价

贪心，dfs先遍历size小的儿子

我的收获：orz

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int N=600005;

int n,cntt=1;
int size[N];
long long sum,ans;
char s[N];

vector<int> E[N];

bool cmp(int x,int y){return size[x]<size[y];}

struct Trie{

#define idx s[i]-'a'

int tot;
int dang[N];
int c[N][27];

void insert(int v=1){
    int x=0,len=strlen(s);
    for(int i=len-1;i>=0;i--){
        if(!c[x][idx]) c[x][idx]=++tot;
        x=c[x][idx];
    }
    dang[x]=1;
}

void build(int k,int x)
{
    if(dang[x]) E[k].push_back(++cntt),size[k=cntt]=1;
    for(int i=0;i<26;i++) if(c[x][i]) build(k,c[x][i]);
}

}T;

void dfssize(int x)
{
    for(int i=0;i<E[x].size();i++)
        dfssize(E[x][i]),size[x]+=size[E[x][i]];
}

void dfs(int x,int wfa)
{
    sum++;ans+=sum-wfa;wfa=sum;
    sort(E[x].begin(),E[x].end(),cmp);
    for(int i=0;i<E[x].size();i++) dfs(E[x][i],wfa);
}

void init()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%s",s),T.insert();
    T.build(1,0);dfssize(1);dfs(1,1);cout<<ans<<endl;
}

int main()
{
    init();
    return 0;
}