【PAT甲级】1076. Forwards on Weibo (30)

最新推荐文章于 2020-07-23 15:38:43 发布

lambsnow

最新推荐文章于 2020-07-23 15:38:43 发布

阅读量209

点赞数

分类专栏： PAT 文章标签： PAT

本文链接：https://blog.csdn.net/lambsnow/article/details/74457598

版权

PAT 专栏收录该内容

78 篇文章 0 订阅

订阅专栏

#include <iostream>
#include <vector>
#include <stack>
#include <stdlib.h>
#include <string.h>
using namespace std;

int main() {
    int n, l;
    cin >> n >> l;
    vector<int> *v = new vector<int>[n + 1];
    int m, fol;
    for (int i = 1; i <= n; i++) {
        cin >> m;
        for (int j = 0; j < m; j++) {
            cin >> fol;
            v[fol].push_back(i);
        }
    }
    int k, q;
    cin >> k;
    int *set = new int[n + 1];
    int count = 0;
    for (int i = 0; i < k; i++) {
        count = 0;
        cin >> q;
        memset(set, 0, (n + 1) * sizeof(int));
        set[q] = 1;
        stack<pair<int, int> > s;
        s.push(make_pair(q, l));
        while (!s.empty()) {
            int t = s.top().first;
            int dep = s.top().second;
            s.pop();
            if (dep >= 1) {
                for (int j = 0; j < v[t].size(); j++) {
                    if (set[v[t][j]] == 0) {
                        set[v[t][j]] = 1;
                        count++;
                        s.push(make_pair(v[t][j], dep - 1));
                    }
                }
            }
        }
        cout << count << endl;
    }
    delete[] set;

    return 0;
}