poj3107(dfs,树形dp)

poj1655的方法完全一样,但是这道题的n的范围大了,用vector存图会TLE(poj没有O2编译优化),所以改用前向星来存图就可以了。、

有关树的重心,看这里:poj1655

这里解释一下前向星存图的方法:

其实就是用静态链表来实现邻接链表,这样可以避免使用指针。

head[i]数组来记录每个节点的第一条边;每条边用结构体e[i]来存,e[i].v表示这条边指向的点,e[i].next表示这条边连向的下一条边。

它的巧妙之处在于每次插入到链表的首部而不是尾部,这样就避免了对链表的遍历。同一起点的各条边在邻接表中的顺序和读入顺序正好相反。

贴个模板:

struct node
{
    int v,next;
}e[m];//m是总边数
int head[n],cnt;//n是总节点数,cnt记录边数
void init()
{
    memset(head,-1,sizeof(head));
    cnt=0;
}
void add(int u,int v)
{
     e[cnt].v=v;
     e[cnt].next=head[u];
     head[u]=cnt++;
}    

本题AC代码:

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<vector>
#include<algorithm>
#include<stack>
#include<queue>
using namespace std;
#define INF 100000000
#define eps 1e-8
#define pii pair<int,int>
#define LL long long int
struct node
{
    int v,next;
}e[100010];
int n,a,b,head[50005],mi=1;
int num[50005],bal[50005],cnt=1;
void add(int aa,int bb);
int dfs1(int x,int fa);
void dfs2(int x,int fa);
int main()
{
    //freopen("in1.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    scanf("%d",&n);
    memset(head,-1,sizeof(int)*(n+1));
    for(int i=1;i<=n;i++) num[i]=1;
    for(int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    dfs1(1,-1);
    dfs2(1,-1);
    for(int i=1;i<=n;i++)
    {
        if(bal[i]<bal[mi])
        {
            mi=i;
        }
    }
    printf("%d",mi);
    for(int i=mi+1;i<=n;i++)
    {
        if(bal[i]==bal[mi])
        {
            printf(" %d",i);
        }
    }
    printf("
");
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}
void add(int aa,int bb)
{
    e[cnt].v=bb;
    e[cnt].next=head[aa];
    head[aa]=cnt++;
}
int dfs1(int x,int fa)
{
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        if(e[i].v==fa)
            continue;
        else
        {
            num[x]+=dfs1(e[i].v,x);
        }
    }
    return num[x];
}
void dfs2(int x,int fa)
{
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        if(e[i].v==fa)
        {
            bal[x]=max(bal[x],n-num[x]);
        }
        else
        {
            bal[x]=max(bal[x],num[e[i].v]);
            dfs2(e[i].v,x);
        }
    }
}
View Code
原文地址:https://www.cnblogs.com/zywscq/p/3898494.html