CodeForces D. Choosing Capital for Treeland (转化为树形DP)

恭喜一下自己……这题没看题解做出来的(虽然思路不小心瞄到了一眼)

题意:给一个n节点的有向无环图,要找一个这样的点:该点到其它n-1要逆转的道路最少,(边,如果v要到u去,则要逆转该边方向)如果有多个这样的点,则升序输出所有


思路:把边的方向化为权值,正向为1,逆向为0。

问题转化为找哪些点的在遍历全图后总权值最大。

这就是树形DP了,考虑每个节点,它可以从子树收获价值,也可以从父亲收获。所以dfs两遍,一边把子树的价值存到dps[i]里,再一遍把父亲的价值存到dpf[i]里。ans[i] = dps[i] + dpf[i]。这都是老套路了!

按照题意再想个办法记录一下哪些点是最大。

总的来说这题还是简单

【代码】

/* ***********************************************
Author        :angon

************************************************ */
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define lld %I64d
#define REP(i,k,n) for(int i=k;i'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}

struct Edge
{
    int v,w,next;
}edge[N*2];
int head[N],tot;
void addedge(int u,int v,int w)
{
    edge[tot] = (Edge){v,w,head[u]};
    head[u] = tot++;
}

int dps[N],dpf[N];

void dfs(int u,int fa)
{
    for(int i=head[u]; ~i;i=edge[i].next)
    {
        int v=edge[i].v;
        int w=edge[i].w;
        if(v==fa) continue;
        dfs(v,u);
        dps[u] += w;
        dps[u] += dps[v];
    }
}

void dfs2(int u,int fa)
{
    for(int i=head[u]; ~i;i=edge[i].next)
    {
        int v=edge[i].v;
        int w=edge[i].w;
        if(v==fa) continue;
        dpf[v] = dpf[u] + (w?0:1) + dps[u] - dps[v] - w;
        dfs2(v,u);
    }
}

struct node
{
    int v,id;
}ans[N];
bool cmp(node a,node b)
{
    return a.v>b.v || a.v==b.v&&a.id


你可能感兴趣的:(CodeForces D. Choosing Capital for Treeland (转化为树形DP))