2023GPLT模拟赛 L2-3 智能护理中心统计

智能护理中心系统将辖下的护理点分属若干个大区，例如华东区、华北区等；每个大区又分若干个省来进行管理；省又分市，等等。我们将所有这些有管理或护理功能的单位称为“管理结点”。现在已知每位老人由唯一的一个管理结点负责，每个管理结点属于唯一的上级管理结点管辖。你需要实现一个功能，来统计任何一个管理结点所负责照看的老人的数量。
注意这是一个动态问题，即随时可能有老人加入某个管理结点，并且老人是有可能从一个管理结点换到另一个管理结点去的。
 
输入格式：
输入在第一行中给出 2 个正整数：$N（\le 10^4）$ 是老人的总数量，即老人们从 1 到 $N$ 编号；$M（\le 10^5）$ 是归属关系的总数。
接下来是 $M$ 行，每行给出一对归属关系，格式为：

A B

表示 A 归属于 B。A 或 B 如果是某个管理结点，则用不超过 4 个大写英文字母表示其名称；如果是某位老人，则用老人的编号表示。这里每个 A 保证只有唯一的上级归属 B，且只有这个中心系统本身是没有上级归属的。此外，输入保证没有老人自己承担管理结点的角色，即 B 一定是一个管理结点，不可能是老人的编号。但一个管理结点既可以管辖下级结点，也可以直接护理一部分老人。
随后每行给出一个指令，格式为：

指令 内容

如果 指令 为 T，则表示有老人要入院或转院，内容 是某老人的编号和要去的管理结点的名称，以空格分隔；如果 指令 为 Q，则 内容 是一个管理结点的名称，意思是统计这个结点所负责照看的老人的数量；如果 指令 为 E，则表示输入结束。题目保证指令总数不会超过 100 个。
输出格式：
对每个 T 指令，将对应的老人转存到对应的管理结点名下；对每个 Q 指令，在一行中输出对应管理结点所负责照看的老人的数量。读到 E 指令就结束程序。

解题思路: 就是处理好数据间的关系，比如说要得知管理结点及其下属结点所管理的老人数，就得记录其所拥有的下属结点(其中可以存在有下属结点的下属结点)、记录老人所属管理结点和管理结点所管理的老人数。
将每个字符串都分配一个数字，能够很好的访问数组下标对应的管理结点(父节点)/下属管理结点/管理结点的老人数。
下属结点在后面的操作中是没有改变的，需要特别处理的就只有老人入院和转院的(入院:前面的输入的归属关系中未出现的老人名字；转院:就是前面确定过所属关系，更改老人所属管理结点)，不管是入院还是转院操作前都会获取老人对应的数字ID，就不用担心没有ID。未出现就会新分配一个，且他的对应的管理结点为 0 ，就表示未有所属关系，即要入院。转院的话就是先将原管理结点的老人数 -1 ，然后重新给老人所属管理结点更改成新的管理结点，最后新的管理结点对应老人数 +1 。

#include <bits/stdc++.h>
using namespace std;

const int N = 1e6+10; //要比1e4大，因为管理结点也需要ID
int fa[N]; //记录老人所属的管理结点
int peo[N]; //记录管理结点i的老人数
vector<int> adm[N]; //记录管理结点i的下属结点
map<string, int> id; //将字符串转化为数字记录

//给字符串分配数字ID
int idx = 1;
int getID(string s)
{
    if(!id[s]) return id[s] = idx++;
    return id[s];
}

//获取该管理结点及其下属结点的老人数
int getPeo(int ID)
{
    int sum = peo[ID];
    for(auto i:adm[ID])
    {
        sum += getPeo(i);
    }
    return sum;
}

int main()
{
    int n, m;
    cin>>n>>m;
    while(m--)
    {
        string a, b;
        cin>>a>>b;
        int ida = getID(a);
        int idb = getID(b);
        if(a[0]>='0' && a[0]<='9')
        {
            fa[ida] = idb;
            peo[idb]++;
        }
        else adm[idb].push_back(ida);
    }
    char op;
    while(cin>>op, op!='E')
    {
        if(op == 'T')
        {
            string a, b;
            cin>>a>>b;
            int ida = getID(a);
            int idb = getID(b);
            peo[fa[ida]]--; //原来没有出现老人对应父节点为0，直接减不影响(即0不属于任何管理结点)
            fa[ida] = idb;
            peo[idb]++;
        }
        else if(op == 'Q')
        {
            string a; cin>>a;
            cout<<getPeo(getID(a))<<'\n';
        }
    }
    return 0;
}