bzoj5195: [Usaco2018 Feb]Directory Traversal（树形dp）

Solution

对于每个点，我们给一个权值 $v_i$ (文件夹为长度，文件为长度-1)
然后dp就行了..
转移要向上和向下分别考虑

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 100010
#define inf (ll)1e17
#define ll long long
int n,m,num=0,tot=0,flag[N],fa[N],sz[N],h[N];
ll mn=inf,v[N],f[N][2],sum[N];
char str[11];
struct node1{int to,next;}mp[N];
inline void insert(int x,int y){
    mp[++num].to=y;mp[num].next=h[x];h[x]=num;
}
void dfs1(int x){
    f[x][1]=0;
    for(int i=h[x];i;i=mp[i].next){
        int y=mp[i].to;
        fa[y]=x;dfs1(y);
        sz[x]+=sz[y];sum[x]+=sum[y];
        f[x][1]+=f[y][1]+v[y]*sz[y];
    }
    if(flag[x]) sz[x]=1;
}
void dfs2(int x){
    if(x!=1) f[x][0]=f[fa[x]][0]-v[x]*sz[x]+(ll)3*(tot-sz[x]);
    else f[x][0]=f[x][1];
    for(int i=h[x];i;i=mp[i].next){
        int y=mp[i].to;
        dfs2(y);
    }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int x;scanf("%s%d",str,&x);
        v[i]=strlen(str)+1;
        if(!x) flag[i]=1,v[i]--,tot++;
        for(int j=1;j<=x;j++){
            int y;scanf("%d",&y);
            insert(i,y);
        }
    }
    dfs1(1);dfs2(1);
    for(int i=1;i<=n;i++){
        if(!flag[i]){
            mn=min(mn,f[i][0]);
        }
    }
    printf("%lld\n",mn);
    return 0;
}