PAT A1076 Forwards on Weibo (30分)【图的BFS遍历】

今天也是为了cc，努力奋斗的一天ヾ(≧▽≦*)o

代码

//注意数组下标是从1开始的 
#include<stdio.h>
#include<vector>
#include<queue>

using namespace std;

//基本单位，里面包含节点值和层数 
struct Node{
	int v;
	int layer;
};

int n,l; 
const int MAXN  = 1010;
vector<int> Adj[MAXN];


//用于实现BFS的函数。 
int BFS(int v){
	bool inq[MAXN] = {false};
	
	//定义一个变量保存最大的层数，初始值为0 
	int maxLayer = 0;
	
	//定义一个变量保存转发的最大人数
	int maxNum = 0; 
	queue<Node> q;
	Node start;
	start.v = v;
	start.layer = 0;
	q.push(start);
	inq[v] = true;	//设置为已经入队
	
	
	while(!q.empty()){	//只要队列不为空 
		Node now = q.front();	//取出队列首 
		q.pop();	//弹出队列首 
		
		for(int i=0;i<Adj[now.v].size();i++){
			
			if( inq[Adj[now.v][i]] == false){
				Node next;
				next.v = Adj[now.v][i];
				next.layer = now.layer + 1;
				inq[next.v] = true;
				q.push(next);
				
				if(next.layer > maxLayer){
					maxLayer = next.layer;
				}
				
				if(maxLayer > l){
					maxLayer = next.layer;
					return maxNum;
				}else{
					maxNum++;
				}
			}			
		}
	}
	
	return maxNum;	
}

int main(){
	scanf("%d %d",&n,&l);
	
	for(int i=1;i<=n;i++){
		int m;
		scanf("%d",&m);
		for(int j=0;j<m;j++){
			int k;
			scanf("%d",&k);
			Adj[k].push_back(i);
		}	
	} 
	
	int cnt;
	scanf("%d",&cnt);
	
	for(int i=0;i<cnt;i++){
		int search;
		scanf("%d",&search);
		int num = BFS(search);
		printf("%d\n",num);	
	} 
	
	
	return 0;
}

反思

1. 刚开始题目的意思没看懂。。。看了《算法笔记》上的中文才知道要干嘛。。。是我英文太差了吗，哭了。
2. 题目的限制是层数，然后在层数的限制下求最多的转发的人数。
3. 其他还好，直接套用模板就ok啦。
4. ps：我的代码写的好辣鸡，不优雅呀，要继续加油。

参考文档

算法笔记

二刷代码

不知道为啥，代码写得越多，就越容易粗心了，比如我二刷的时候只考虑了刚好小于等于L的情况，没有考虑小于L的情况。。。

//比较trivial的层序遍历题
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int maxn = 1010;

int n,l;
vector<int> adj[maxn];
bool vis[maxn];

void bfs(int u,int l){
	fill(vis,vis+maxn,false); 
	queue<int> q;
	q.push(u);
	vis[u]=true;
	int cnt=0;
	int curLayer=0;
	
	while(!q.empty()){
		int size = q.size();
		if(curLayer<=l){
			//printf("%d\n",cnt-1);//减去自身~ 
			cnt+=size;
		}
	
		for(int j=0;j<size;j++){//访问每一层 
			int cur = q.front();
			q.pop();
			
			for(int i=0;i<adj[cur].size();i++){
				int v = adj[cur][i];
				if(vis[v]==false){
					q.push(v);
					vis[v]=true;
				} 
			}				
		}		
		curLayer++;
	}
	printf("%d\n",cnt-1);
}

int main(){
//	freopen("input.txt","r",stdin); 
	scanf("%d %d",&n,&l);
	for(int i=1;i<=n;i++){
		int m;
		scanf("%d",&m);
		for(int j=0;j<m;j++){
			int cur;
			scanf("%d",&cur);
			adj[cur].push_back(i);
		}	
	}
	
	//查询
	int k;
	scanf("%d",&k);
	for(int i=0;i<k;i++){
		int u;
		scanf("%d",&u);
		bfs(u,l);
	}
	 
	return 0;
}