[ZJOI2008]骑士 题解

[ZJOI2008]骑士

https://ac.nowcoder.com/acm/problem/20481

本题考察基环外向树上的树形dp
首先需要找到环上的任意一条边,然后删去这条边,再对这条边的两个点进行树形dp,最后选择根节点都不选的最大值,就是最后的答案。
根节点都不选的理由是:如果选上根结点,由于环上找出来的这条边切断了,所以在进行树形dp的时候,没有考虑切断的这条边上的另一个点,有可能把那个点选上了,答案会比实际答案大,造成错误。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1000100;
struct node
{
    int v,next;
}e[N<<1];
int head[N];
int val[N];
bool vis[N];
int cnt=0,n;
int pa,pb,pc;
ll ans,f[N][2];
void add(int u,int v)
{
    e[++cnt].v=v;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void dfs(int u,int pre)
{
    vis[u]=true;
    for(int i=head[u];i;i=e[i].next)
    {
        int to=e[i].v;
        if(!vis[to]) dfs(to,u);
        else if(to!=pre) pa=u,pb=to,pc=(i+1)>>1;
    }
}
void dp(int u,int pre)
{
    f[u][0]=0,f[u][1]=val[u];
    for(int i=head[u];i;i=e[i].next)
    {
        int to=e[i].v;
        if(to!=pre&&((i+1)>>1)!=pc)
        {
            dp(to,u);
            f[u][0]+=max(f[to][1],f[to][0]);
            f[u][1]+=f[to][0];
        }
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int ce,v;
        scanf("%d%d",&ce,&v);
        val[i]=ce;
        add(i,v),add(v,i);
    }
    for(int i=1;i<=n;i++)
    {
        if(!vis[i]) {
            dfs(i,0);
            dp(pa,0);
            ll va=f[pa][0];
            dp(pb,0);
            ll vb=f[pb][0];
            ans+=max(va,vb);
        }
    }
    printf("%lld\n",ans);
}
全部评论

相关推荐

02-14 12:40
门头沟学院 Java
程序员花海:1.面试要求必须Java笔试不一定 2.难度对等秋招 远超于日常实习是因为同一批次且转正很多 竞争压力大 3.第一个加点指标,上线了就把接口性能加上去 使用本地缓存这个不算亮点 只是技术选型,要把为什么采用这个和背后的思考写出来而不是单纯堆叠技术没意义 4.八股要一直看 很容易忘记 5.拼团交易这个老问题 堆积技术 另外建议你把奖项合并到教育背景 没必要拆出来放最后
我的简历长这样
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务