bzoj 4567: [Scoi2016]背单词 字典树+贪心

题意

意思就是挨个放单词，放每个单词之前一定把这个单词的后缀都先放上去，每个单词的代价等于这个单词的位置减去上一个出现的这个单词的后缀的位置
n<=100000,sigma(len)<=510000

分析

一开始的想法是用后缀数组来建图，要真打起来的话估计会T。而且也没想到要怎么贪心。。。还是太弱。

其实只要一颗字典树就可以处理好建图问题啦。
然后谈心的话。。。每次找size较小的子树去贪即可。证明嘛。。。这不显然嘛。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;

int cnt,n,tim,size[N*10],last[N*20],ch[N*10][30],top,sta[N],val[N*10],tot,dfn[N*10];
long long ans;
struct edge{int to,next;}e[N];
char s[N*10];

void ins()
{
    int len=strlen(s),now=1;
    for (int i=len-1;i>=0;i--)
        if (ch[now][s[i]-'a']) now=ch[now][s[i]-'a'];
        else now=ch[now][s[i]-'a']=++tot;
    val[now]=1;
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void build(int x,int y)
{
    if (val[x]) addedge(y,x),y=x;
    for (int i=0;i<26;i++)
        if (ch[x][i]) build(ch[x][i],y);
}

void get_size(int x)
{
    size[x]=1;
    for (int i=last[x];i;i=e[i].next) get_size(e[i].to),size[x]+=size[e[i].to];
}

bool cmp(int a,int b)
{
    return size[a]<size[b];
}

void dfs(int x)
{
    dfn[x]=++tim;int l=top+1;
    for (int i=last[x];i;i=e[i].next) sta[++top]=e[i].to;
    if (top<l) return;
    sort(sta+l,sta+top+1,cmp);
    for (int i=l;i<=top;i++) dfs(sta[i]),ans+=dfn[sta[i]]-dfn[x];
    top=l-1;
}

int main()
{
    scanf("%d",&n);
    tot=1;
    for (int i=1;i<=n;i++)
    {
        scanf("%s",s);
        ins();
    }
    build(1,1);
    get_size(1);
    dfs(1);
    printf("%lld",ans);
    return 0;
}