【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;
}

猜你喜欢

转载自blog.csdn.net/MYMarcoreus/article/details/113794783