8.10-Day2T3 镇守府

emm

IOI原题（洛谷，bzoj都有）

其实并没有怎么搞懂dp方程转移那部分

就...粘个（抄来的）代码过来吧

#include<bits/stdc++.h>
using namespace std;

inline int read()
{
    int sum = ,p = ;
    char ch = getchar();
    while(ch < '' || ch > '')
    {
        if(ch == '-')
            p = -;
        ch = getchar();
    }
    while(ch >= '' && ch <= '')
    {
        (sum *= ) += ch - '';
        ch = getchar();
    }
    return sum * p;
}

const int N = ,M = ;
const int inf = 0x3f3f3f3f;
int n,m,head[N],w[N],f[N],cnt;
int dis[N][N],dp[N][M][N];
struct edge
{
    int nxt,to;
}e[N];

void add(int a,int b)
{
    e[++cnt].nxt = head[a];
    e[cnt].to = b;
    head[a] = cnt;
}

void dfs(int u)
{
    for(int last = u,i = f[u]; i != -; i = f[i])
    {
        dis[u][i] = dis[u][last] + dis[last][i];
        last = i;
    }
    for(int i = head[u]; i; i = e[i].nxt)
        dfs(e[i].to);
}

void tree_dp(int u)
{
    bool flag = true;
    for(int i = head[u]; i; i = e[i].nxt)
    {
        flag = false;
        tree_dp(e[i].to);
    }
    if(flag)
    {
        for(int i = f[u]; i != -; i = f[i])
            dp[u][][i] = dis[u][i] * w[u];
        return;
    }
    if(u != )
        dp[u][][u] = inf;
    for(int i = head[u]; i; i = e[i].nxt)
    {
        int v = e[i].to;
        for(int p = f[u]; p != -; p = f[p])
        {
            for(int j = m; j >= ; j--)
            {
                int tmp = inf;
                for(int k = ; k <= j; k++)
                    tmp = min(tmp,dp[u][k][p] + dp[v][j - k][p]);
                dp[u][j][p] = tmp;
            }
        }
        for(int j = m; j >= ; j--)
        {
            int tmp = inf;
            for(int k = (u != ); k <= j; k++)
                tmp = min(tmp,dp[u][k][u] + dp[v][j - k][u]);
            dp[u][j][u] = tmp;
        }
    }
    for(int p = f[u]; p != -; p = f[p])
        for(int i = ; i <= m; i++)
        {
            dp[u][i][p] += dis[u][p]*w[u];
            dp[u][i][p] = min(dp[u][i][p],dp[u][i][u]);
        }
}

int main()
{
    freopen("riv.in","r",stdin);
    freopen("riv.out","w",stdout);
    n = read(),m = read();
    for(int i = ; i <= n; i++)
    {
        w[i] = read();
        f[i] = read();
        dis[i][f[i]] = read();
        add(f[i],i);
    }
    f[] = -;
    dfs();
    tree_dp();
    printf("%d\n",dp[][m][]);
    return ;
}