poj 3107 Godfather

来源：互联网发布：tensorflow与spark 编辑：程序博客网时间：2024/05/17 04:35

刚看到这道题啊

水水的树形dp啊

TLE

#include <cstdio>#include <cstring>#include <vector>#include <algorithm>using namespace std;#define maxx 50050int N;int T;vector <int>root[maxx];bool vis[maxx];int num[maxx];int dp[maxx];int dfs(int x){    int n=root[x].size();    num[x]=1;    vis[x]=1;    for(int i=0;i<n;i++)    {        if(!vis[root[x][i]])            num[x]+=dfs(root[x][i]);    }    return num[x];}void cal (int x){    vis[x]=1;    int n=root[x].size();    for(int i=0;i<n;i++)    {        int k=root[x][i];        if(vis[k])        {             dp[x]=max(dp[x],N-num[x]);        }        else        {            dp[x]=max(dp[x],num[k]);            cal(k);        }    }}int main (){    int a,b;    while(scanf("%d",&N)==1){    for(int i=0;i<N;i++)    root[i].clear();    for(int i=1;i<N;i++)    {        scanf("%d%d",&a,&b);        root[a].push_back(b);        root[b].push_back(a);    }   memset(num,0,sizeof(num));   memset(dp,0,sizeof(dp));   memset(vis,0,sizeof(vis));    dfs(1);    memset(vis,0,sizeof(vis));    cal(1);    int k;    int ans=100000000;    for(int i=1;i<=N;i++)    {        if(ans>=dp[i])        {            ans=dp[i];        }    }    int h=0;    int bb[maxx];     for(int i=1;i<=N;i++)     {         if(dp[i]==ans)         {             bb[h]=i;             h++;         }     }     for(int j=0;j<h;j++)     {         if(j==h-1)            printf("%d\n",bb[j]);         else            printf("%d ",bb[j]);     }    }}

想不通啊怎末班后来发现AC的代码全部都是邻接表写的

这怎末班啊啊啊啊啊啊

不会啊

学吧

AC代码如下

这道题的思路和poj 1655 poj 2378没差的改了改代码

只要你找到这个树的重心把它输出来就好了（可能存在多个）

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;struct node{    int xx,next;}g[100050];int cnt;int num[50050];int dp[50050];int head[50050];bool vis[50050];void add(int u,int v){    g[cnt].next=head[u];    g[cnt].xx=v;    head[u]=cnt++;}int N,ans;int dfs(int x){    vis[x]=1;    num[x]=1;    for ( int i = head [ x] ; i!=-1 ; i = g [ i ] . next )    {        int k=g[i].xx;        if(!vis[k])        {            num[x]+=dfs(k);            dp[x]=max(dp[x],num[k]);        }    }    dp[x]=max(dp[x],N-num[x]);    ans=min(ans,dp[x]);    return num[x];}int main (){    scanf("%d",&N);    int a,b;    memset(dp,0,sizeof(dp));    memset(num,0,sizeof(num));    memset(vis,0,sizeof(vis));    memset(head,-1,sizeof(head));    cnt=0;    for(int i=1;i<N;i++)    {        scanf("%d%d",&a,&b);        add(a,b);        add(b,a);    }     ans=0x7fffffff;    dfs(1);    for(int i=1;i<=N;i++)    {        if(dp[i]==ans)            printf("%d ",i);    }    puts("");}

0 0