poj 4045

来源：互联网发布：android编程软件编辑：程序博客网时间：2024/05/22 12:58

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 50010vector<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");    }}