PAT【甲级】1004

题目链接：PAT【甲级】1004

#include<iostream>
#include<queue>
#include<string>
#include<unordered_map>
#include<vector>
using namespace std;

int mlayer = -1;
//其实这题一看就是先构建一颗数，但是树节点要有标记，所以用dfs会好一些
void dfs(unordered_map<string, vector<string>>& m, unordered_map<string, bool>& mt, vector<int>& ans, string id, int layer){
    mlayer = max(mlayer, layer);//更新层数
    if(m[id].size() == 0)//如果当前节点为叶子节点
        ans[layer]++;
    else{//否则当前节点为非叶子节点
        for (int i = 0; i < m[id].size();i++){
            if(mt[m[id].at(i)])
                ans[layer]++;
            else
                dfs(m, mt, ans, m[id].at(i), layer + 1);
        }
    }
    return;
}

int main(){
    int N, M, k;
    vector<string> v;
    string id;
    unordered_map<string, vector<string>> m;
    unordered_map<string, bool> mt(true);
    vector<int> ans(100, 0);
    cin >> N >> M;
    for (int i = 0; i < M;i++){
        cin >> id >> k;
        string temp;
        for (int j = 0; j < k;j++){
            cin >> temp;
            v.push_back(temp);
        }
        mt[id] = false;
        m[id] = v;
        v.clear();
    }
    dfs(m, mt, ans, "01", 0);
    cout << ans[0];
    for (int i = 1; i <= mlayer;i++){
        cout << " " << ans[i];
    }
    return 0;
}

这题目应该就是去建立一棵树，类似以链表的形式来存储的树，不过是这边用了标记来方便的判断是否为叶子节点。其实也可以通过当前节点是否能扩展来判断，这样会剩下这部分空间。处理好数据后，所需的就是dfs一下，同时记录它的高度，便于输出数字。对于树，而且是需要统计信息的这类问题，一般是用dfs来进行它的遍历即可。