首页 > 代码库 > poj3107(dfs,树形dp)
poj3107(dfs,树形dp)
和poj1655的方法完全一样,但是这道题的n的范围大了,用vector存图会TLE,所以改用前向星来存图就可以了。
这里解释一下前向星存图的方法:
其实就是用静态链表来实现邻接链表,这样可以避免使用指针。
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 intstruct 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("\n"); //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); } }}
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。