hdu 3290 (簡單dfs)

題意:沒有兒子的節點所結蘋果數是節點的編號,有兒子的所結蘋果是兒子數量(k+1)/2個,求跟節點的蘋果數

直接遞歸一下,先求出所有兒子的蘋果樹,在排序,,剛開始以爲1就是根節點,根節點不確定,,




#include<stdio.h>
#include<string.h>
#include<stdlib.h>
const int N=20010;
int w[N],vis[N],head[N],num,p[N];
struct edge
{
	int st,ed,next;
}e[N*4];
int cmp(void const *a,void const *b)
{
	return *(int *)a-*(int *)b;
}
void addedge(int x,int y)
{
	e[num].st=x;e[num].ed=y;e[num].next=head[x];head[x]=num++;
}
int dfs(int u)
{
	w[u]=u;
	int i,j=0;
	for(i=head[u];i!=-1;i=e[i].next)
		dfs(e[i].ed);
	for(i=head[u];i!=-1;i=e[i].next)
		p[j++]=w[e[i].ed];
	if(j>0)
	{
	  qsort(p,j,sizeof(p[0]),cmp);
	  w[u]=p[(j+1)/2-1];
	}
	return w[u];
}
int main()
{
	int i,x,n,m;
	while(scanf("%d",&n)!=-1)
	{
		memset(head,-1,sizeof(head));
		num=0;
		memset(w,0,sizeof(w));
		memset(vis,0,sizeof(vis));
		for(i=1;i<=n;i++)
		{
			scanf("%d",&m);
			while(m--)
			{
				scanf("%d",&x);
				addedge(i,x);
				vis[x]=1;
			}
		}
		for(i=1;i<=n;i++)
		{
			if(vis[i]==0)
		 printf("%d\n",dfs(i));
		}
	}
	return 0;
}






發佈了236 篇原創文章 · 獲贊 3 · 訪問量 22萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章