[树形dp] Jzoj P5248 花花的聚会

Description

 

Input

Output

 

Sample Input

7 7
3 1
2 1
7 6
6 3
5 3
4 3
7 2 3
7 1 1
2 3 5
3 6 2
4 2 4
5 3 10
6 1 20
3
5
6
7

Sample Output

10
22
5

 

Data Constraint

 

Hint

 

题解

• 题目大意：给定一棵有n个节点的树，在某一些节点，可以用wi的费用向上走ki步，现在查询q个询问，询问一些节点向根节点的最小代价
• 设f[u]为u点走到根节点的最小代价
• 显然，方程就是：f[u]=min(f[u],f[v]+wi)(而且i要满足dep[u]-dep[v]<=ki)
• 然后，可以暴力的向上跳到祖先，在ki的范围内取min值

代码

#include <cstdio>
#include <iostream>
using namespace std;
const int inf=0x7fffffff;
struct edge { int k,w,to; }e[100010];
int n,m,q,cnt,fa[100010],f[100010],head[100010];
void insert(int v,int k,int w) { e[++cnt].k=k; e[cnt].w=w; e[cnt].to=head[v]; head[v]=cnt; }
int dp(int x)
{
    if (x==1) return 0;
    if (f[x]!=-1) return f[x];
    f[x]=inf;
    for (int i=head[x];i;i=e[i].to)
    {
        int v=fa[x];
        for (int j=1;j<=e[i].k&&v>=1;j++)
        {
            f[x]=min(f[x],dp(v)+e[i].w);
            v=fa[v];
        }
    }
    return f[x];
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n-1;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        fa[u]=v;
        f[i]=head[i]=-1;
    }
    f[n]=head[n]=-1;
    for (int i=1;i<=m;i++)
    {
        int v,k,w;
        scanf("%d%d%d",&v,&k,&w);
        insert(v,k,w);
    }
    scanf("%d",&q);
    for (int i=1;i<=q;i++)
    {
        int x;
        scanf("%d",&x);
        printf("%d\n",dp(x));
    }
    return 0;
}