HDU 4616

多校被虐暴了

五个小时一道题没出是什么个节奏

这道题还不算很难 重点在树上任意一点开始的最长路径问题

dp[i][j][k]表示第i个节点之前有j个炸弹的最大价值 k=1是代表最优路径 k=0代表次优 同时保存是由哪个子节点转移过来的

这么在第二次搜索的时候如果最优路径正好是当前子节点转移过来的话就选择次优路径

详情就是见代码了

代码写的这叫一个丑陋啊 太尼玛伤心了

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 100020
struct data
{
    int val;
    int son;
} dp[N][6][2];
int ans;
int n,C,gift[N],trk[N];
int head[N],next[N],to[N],size;
void add(int s,int t)
{
    next[size]=head[s];
    to[size]=t;
    head[s]=size++;
}
int init()
{
    memset(dp,0,sizeof(dp));
    memset(head,-1,sizeof(head));
    size=0,ans=0;
    int s,t;
    scanf ("%d%d",&n,&C);
    for(int i=0; i<n; i++)
        scanf("%d%d",&gift[i],&trk[i]);
    for(int i=0; i<n-1;i++)
    {
        scanf ("%d%d",&s,&t);
        add(s,t);
        add(t,s);
    }
    return 0;
}
void dfs(int pre,int now)
{
    for(int i=head[now]; i!=-1; i=next[i])
    {
        int v=to[i];
        if(v!=pre)
        {
            dfs(now,v);
            for(int i=0; i<C; i++)
                if(dp[now][i+trk[v]][1].val<dp[v][i][1].val+gift[v])
                {
                    dp[now][i+trk[v]][0].val=dp[now][i+trk[v]][1].val;
                    dp[now][i+trk[v]][1].val=dp[v][i][1].val+gift[v];
                    dp[now][i+trk[v]][1].son=v;
                }
                else
                if(dp[now][i+trk[v]][0].val<dp[v][i][1].val+gift[v])
                {
                    dp[now][i+trk[v]][0].val=dp[v][i][1].val+gift[v];
                }
        }
    }
    return;
}
void getans(int pre,int now)
{
    if(pre!=-1)
    {
        for(int i=0; i<C; i++)
            if(dp[pre][i][1].son!=now)
            {
                if(dp[now][i+trk[pre]][1].val<dp[pre][i][1].val+gift[pre])
                {
                    dp[now][i+trk[pre]][1].val=dp[pre][i][1].val+gift[pre];
                    dp[now][i+trk[pre]][1].son=pre;
                }
                else
                if(dp[now][i+trk[pre]][0].val<dp[pre][i][1].val+gift[pre])
                {
                    dp[now][i+trk[pre]][0].val=dp[pre][i][1].val+gift[pre];
                }
            }
            else
            {if(dp[now][i+trk[pre]][1].val<dp[pre][i][0].val+gift[pre])
            {
                dp[now][i+trk[pre]][1].val=dp[pre][i][0].val+gift[pre];
                dp[now][i+trk[pre]][1].son=pre;
            }
            else
            if(dp[now][i+trk[pre]][0].val<dp[pre][i][0].val+gift[pre])
                {
                    dp[now][i+trk[pre]][0].val=dp[pre][i][0].val+gift[pre];
                }
            }
    }
    for(int i=head[now]; i!=-1; i=next[i])
    {
        if(to[i]!=pre)
            getans(now,to[i]);
    }
    return;
}
int main ()
{
    int t;
    scanf ("%d",&t);
    while(t--)
    {
        init();
        dfs(-1,0);
        getans(-1,0);
        for(int i=0; i<n; i++)
            for(int j=0; j<C; j++)
                ans=max(ans,dp[i][j][1].val+gift[i]);
        printf("%d\n",ans);
    }
}