pat_1004

源地址:http://www.patest.cn/contests/pat-a-practise/1004

题目大意:给你一棵有n个节点的树,它的树根编号为1,有m个非叶子节点,它们分别含有k个子节点,然后跟着k个节点的编号,最后输出s个数,代表这s层每层的叶子节点的个数。

刚开始把输入想的太理想化了,,以为都是一层一层来输入的,其实题目中并没有任何这样的信息。WA了一次。

后来想想用bfs的话就不存在这样的问题了,感觉就像对二叉树的遍历一样,顺便记录下每层的叶子节点数。

#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<time.h>
#include<queue>
#include<stack>
#include<iterator>
#include<math.h>
#include<stdlib.h>
#include<limits.h>
#include<set>
#include<map>
//#define ONLINE_JUDGE
#define eps 1e-3
#define INF 0x7fffffff
#define FOR(i,a) for((i)=0;i<(a);(i)++)
#define MEM(a) (memset((a),0,sizeof(a)))
#define sfs(a) scanf("%s",a)
#define sf(a) scanf("%d",&a)
#define sfI(a) scanf("%I64d",&a)
#define pf(a) printf("%d\n",a)
#define pfI(a) printf("%I64d\n",a)
#define pfs(a) printf("%s\n",a)
#define sfd(a,b) scanf("%d%d",&a,&b)
#define sft(a,b,c)scanf("%d%d%d",&a,&b,&c)
#define for1(i,a,b) for(int i=(a);i<b;i++)
#define for2(i,a,b) for(int i=(a);i<=b;i++)
#define for3(i,a,b)for(int i=(b);i>=a;i--)
#define MEM1(a) memset(a,0,sizeof(a))
#define MEM2(a) memset(a,-1,sizeof(a))
const double PI=acos(-1.0);
template<class T> T gcd(T a,T b){return b?gcd(b,a%b):a;}
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
template<class T> inline T Min(T a,T b){return a<b?a:b;}
template<class T> inline T Max(T a,T b){return a>b?a:b;}
using namespace std;
#define ll __int64
int n,m;
#define Mod 1000000007
#define N 510
#define M 1000100
const int size = 10010;
const int mod = 9901;
int vis[N];
struct Node{
	int id;	//代表某一个节点所在的层数的编号
	vector<int> v;
}a[N];
void bfs(){
	Node cur,next;
	queue<Node> q;
	memset(vis,0,sizeof vis);
	a[1].id=0;
	q.push(a[1]);
	vis[a[1].id] = 1;	//vis[i]代表第i层有vis[i]个叶子节点
	int mmax = 0;
	while(!q.empty()){
		cur = q.front();
		q.pop();
		int size = (int)cur.v.size();
		vis[cur.id+1] += size;	//下一层的节点数增加 size个
		if(size)	//如果size不为0,那么cur.id所在的层的叶子节点就要少一个(即cur这个节点)
			vis[cur.id]--;
		for(int i=0;i<size;i++){
			int v = cur.v[i];
			if(a[v].id == -1){
				a[v].id = cur.id+1;
				mmax = Max(mmax,a[v].id);	//记录最大的层数
				q.push(a[v]);
			}
		}
	}
	printf("%d",vis[0]);
	for(int i=1;i<=mmax;i++)	//输出所有层的叶子节点数
		printf(" %d",vis[i]);
	printf("\n");
}
int main(){
#ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
//  freopen("out.txt","w",stdout);
#endif
    while(scanf("%d%d",&n,&m)!=EOF){
    	int u,k,x;
    	for(int i=0;i<=n;i++){
    		a[i].v.clear();
    		a[i].id = -1;
    	}
    	for(int i=0;i<m;i++){
    		scanf("%d%d",&u,&k);
    		for(int j=0;j<k;j++){
    			scanf("%d",&x);
    			a[u].v.push_back(x);
    		}
    	}
    	bfs();
    }
return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值