poj 4045

http://poj.org/problem?id=4045

 

思路：

   1.先用一次dfs() 求出顶点 1 安放power station 的答案 和 各个顶点为根的树的顶点数

   2。dfs()每个边 u- > v , 如果已知ans[ u ]  那么  ans[ v ] = ans[u]  + n -cnt[ v ]  - cnt[ v ] 

代码转自 ：http://fayaa.com/code/view/26409/

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<math.h>
#include<string>
#include<map>
#include<set>
using namespace std;
typedef __int64 ll;
#define MAXN 50010

vector<int> e[MAXN],ans;
int cnt[MAXN],dsum[MAXN];
int dmin[MAXN],n;

void dfs(int pre,int u)
{
    cnt[u]=1;
    dsum[u]=0;
    for(int i=0;i<e[u].size();i++){
        if(e[u][i]==pre) continue;
        dfs(u,e[u][i]);
        cnt[u]+=cnt[e[u][i]];
        dsum[u]+=dsum[e[u][i]]+cnt[e[u][i]];
    }
}
void dfs2(int pre,int u)
{
    for(int i=0;i<e[u].size();i++){
        if(e[u][i]==pre) continue;
        int v=e[u][i];
        dmin[v]=dmin[u]+n-2*cnt[v];
        dfs2(u,e[u][i]);
    }
}
int main()
{
    int T, I, R ,x ,y;
    scanf("%d", &T);
    while (T--) {
        scanf("%d%d%d", &n,&I,&R);
        for(int i=0;i<=n;i++)
            e[i].clear();
        for(int i=0;i<n-1;i++){
            scanf("%d%d",&x,&y);
            e[x].push_back(y);
            e[y].push_back(x);
        }
        dfs(1,1);
        dmin[1]=dsum[1];
        dfs2(1,1);
        int Min=1<<30;
        for(int i=1;i<=n;i++)
            Min=min(Min,dmin[i]);

        printf("%I64d\n",(ll)Min*I*I*R);
        ans.clear();
        for(int i=1;i<=n;i++)
            if(dmin[i]==Min)
                ans.push_back(i);
        for(int i=0;i<ans.size();i++){
            if(i==ans.size()-1)
                printf("%d\n",ans[i]);
            else
                printf("%d ",ans[i]);
        }
        printf("\n");
    }
}