PAT A1076

题目大意

给出每个用户关注的人的 id，和转发最多的层数，求一个 id 发了条微博最多会有多少个人转发。

分析

带层数的广度优先，因为一个用户只能转发一次，所以用 vis 数组判断当前结点是否入队过了，如果入队过了就不能重复入队（即重复转发消息）。queue的数据类型必须为node，同时保存它的 id 和 layer 层数，控制不超过层。

AC代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
int n, m, l, k;
vector<vector<int> > v;
struct Node{
	int id, layer;
};

int bfs(Node node)
{
	int vis[1005] = { 0 };
	queue<Node> q;
	q.push(node);
	vis[node.id] = 1;
	int cnt = 0;
	while (!q.empty())
	{
		Node now = q.front();
		q.pop();
		for (int i = 0; i < v[now.id].size(); i++)
		{
			if (!vis[v[now.id][i]] && now.layer < l)
			{
				Node next = {v[now.id][i], now.layer + 1};
				q.push(next);
				vis[next.id] = 1;
				cnt++;
			}
		}
	}
	return cnt;
}

int main()
{
	scanf("%d%d", &n, &l);
	v.resize(n + 1);
	for (int i = 1; i <= n; i++)
	{
		scanf("%d", &m);
		for (int j = 0; j < m; j++)
		{
			int t;
			scanf("%d", &t);
			v[t].push_back(i);
		}
	}
	scanf("%d", &k);
	for (int i = 0; i < k; i++)
	{
		int id;
		scanf("%d", &id);
		Node node = {id, 0};
		printf("%d\n", bfs(node));
	}
	return 0;
}