cf 135div2D Choosing Capital for Treeland 【樹形dp】

鏈接:http://codeforces.com/problemset/problem/219/D

題意:n個點,m個有向邊,現在要你求出每個點作爲根使得圖爲一棵樹需要改變多少邊的方向,輸出最少改變多少次,和這些點。

分析:建圖時發現,一個點爲根時他的出度邊不用改變,所以將所有的入邊的權值爲1,然後用dfs統計每個點的子樹有多少這樣的邊。

我們還可以發現,一個點(u)和他的父親(fa)的關係,就是u成爲整個樹的root時,答案的改變只和(u,fa)這條邊有關,在 第二次dfs是可以更新。

代碼:

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 200005
#define Mm 400010
#define mod 1000000007
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
struct edge {
    int v,w,next;
}e[Mm];
int tot=0,head[Mn];
void addedge(int u,int v,int w) {
    e[tot].v=v;
    e[tot].w=w;
    e[tot].next=head[u];
    head[u]=tot++;
}
int dp[Mn];
void dfs(int u,int fa) {
    for(int i=head[u];~i;i=e[i].next) {
        int v=e[i].v;
        int w=e[i].w;
        if(v==fa) continue;
        dfs(v,u);
       // cout<<u<<" "<<v<<" "<<dp[v]<<" "<<w<<endl;
        dp[u]+=dp[v]+w;
    }
}
int minn;
void dfs2(int u,int fa) {
    minn=min(dp[u],minn);
    for(int i=head[u];~i;i=e[i].next) {
        int v=e[i].v;
        int w=e[i].w;
        if(v==fa) continue;
        if(w) dp[v]=dp[u]-1;
        else dp[v]=dp[u]+1;

        dfs2(v,u);
    }
}
void init() {
    CLR(head,-1);
    tot=0;
}
int main() {
    int n;
    init();
    scanf("%d",&n);
    minn=n;
    for(int i=1;i<n;i++) {
        int u,v;
        scanf("%d%d",&u,&v);
        addedge(u,v,0);
        addedge(v,u,1);
    }
    dfs(1,0);
    dfs2(1,0);
    printf("%d\n",minn);
    for(int i=1;i<=n;i++) {
        if(dp[i]==minn)
        printf("%d ",i);
    }
    return 0;
}



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章