【PAT甲级 C++ BFS】1076 Forwards on Weibo (30 分)

给定一个起始点，从起始点开始求距离起始点一定深度内能访问的最大顶点个数

# include <iostream>
# include <queue>
# include <vector>
# include <algorithm>

/* 给定一个起始点，从起始点开始求距离起始点一定深度内能访问的最大顶点个数 
        使用DFS会遇到一个很大的困难：通过路线1达到遍历深度时，其后面的顶点会无法访问，
        但可能会有一条的路线2可以获得更大转发数，但是关键点却被标记过已访问，从而丢失了这条更优的路线
        
        而且DFS能导致同一个结点的转发次数被重复计算
*/ 

using namespace std;

struct Node{
    int id;
    int depth; // 记录顶点的深度
};

bool vis[1010] = {0};
vector<Node> V[1010];
int N, L;

void BFS(int startid, int &maxfwd){ // 起始顶点和最大转发数（也就是在一定深度内能访问的顶点个数）
	// 建立队列，将初始顶点入队并将其depth初始化为0、标记为已访问
    queue<Node> Q;
    Q.push({startid, 0});
    vis[startid] = true;
    
    // BFS
    while(!Q.empty()){
        // 访问（记录）当前顶点，并将其出队
        Node now = Q.front();
        Q.pop();
        
        // 遍历所有未访问过并且最大深度不超过L的邻接点
        for(int i = 0;i<V[now.id].size();++i){
            Node next = {V[now.id][i].id, now.depth + 1};
            if(vis[next.id] == false && next.depth <= L){
                Q.push(next);
                vis[next.id] = true;
                maxfwd++;
            }
        }
    }
}

int main(){
    cin >> N >> L;
    for(int i = 1;i<=N;++i){
        int t;
        cin >> t;
        for(int j = 0;j<t;++j){
            int e;
            cin >> e;
            // 关键的是图的建立，他给的时一个人的关注列表，但是一个人只有被关注才会被转发，
			// 所以不能V[i].push_back({e, 0});建立，要反过来
            V[e].push_back({i, 0});
        }
    }
    int K;
    cin >> K;
    for(int i = 0;i<K;++i){
        fill(vis, end(vis), false); // 记得每次vis数组要归零
        int id;
        cin >> id;
        int maxfwd = 0; // 记录每次的最大转发数
        BFS(id, maxfwd);
        cout << maxfwd <<endl;
    }

    return 0;
}