智能护理中心系统将辖下的护理点分属若干个大区,例如华东区、华北区等;每个大区又分若干个省来进行管理;省又分市,等等。我们将所有这些有管理或护理功能的单位称为“管理结点”。现在已知每位老人由唯一的一个管理结点负责,每个管理结点属于唯一的上级管理结点管辖。你需要实现一个功能,来统计任何一个管理结点所负责照看的老人的数量。
注意这是一个动态问题,即随时可能有老人加入某个管理结点,并且老人是有可能从一个管理结点换到另一个管理结点去的。
输入格式:
输入在第一行中给出 2 个正整数:N(≤104)是老人的总数量,即老人们从 1 到 N 编号;M(≤105)是归属关系的总数。
接下来是 M 行,每行给出一对归属关系,格式为:
A B
表示
A
归属于B
。A
或B
如果是某个管理结点,则用不超过 4 个大写英文字母表示其名称;如果是某位老人,则用老人的编号表示。这里每个A
保证只有唯一的上级归属B
,且只有这个中心系统本身是没有上级归属的。此外,输入保证没有老人自己承担管理结点的角色,即B
一定是一个管理结点,不可能是老人的编号。但一个管理结点既可以管辖下级结点,也可以直接护理一部分老人。随后每行给出一个指令,格式为:
指令 内容
如果
指令
为T
,则表示有老人要入院或转院,内容
是某老人的编号和要去的管理结点的名称,以空格分隔;如果指令
为Q
,则内容
是一个管理结点的名称,意思是统计这个结点所负责照看的老人的数量;如果指令
为E
,则表示输入结束。题目保证指令总数不会超过 100 个。输出格式:
对每个
T
指令,将对应的老人转存到对应的管理结点名下;对每个Q
指令,在一行中输出对应管理结点所负责照看的老人的数量。读到E
指令就结束程序。输入样例:
10 23 EAST CNTR ZJ EAST SD EAST WEST CNTR SX WEST HZ ZJ JN SD 2 JN 8 JTH 6 XAHP 4 ZDYH 5 ZDYH ZDYH HZ HUZ ZJ JX ZJ 1 JX 3 JN WZ ZJ XAHP XIAN XIAN SX YL SX JTH XIAN 7 XAHP Q EAST T 1 YL Q EAST Q SX T 8 ZDYH Q HZ Q HUZ T 10 XAHP Q CNTR E
输出样例:
5 4 4 3 0 9
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,m;
cin>>n>>m;
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
unordered_map<string,map<string,int>>super;
unordered_map<string,map<string,int>>user;
unordered_map<string,string>link;
unordered_map<string,int>nums;
for(int i=0;i<m;i++){
string A,B;
cin>>A>>B;
if(A[0] >='A' && A[0]<='Z'){
//super link super
super[B][A]=1;
}else{
//user link super
user[B][A]=1;
link[A] = B;
nums[B]++;
}
}
char ch;
while(cin>>ch){
if(ch=='E')break;
if(ch=='T'){
string id;
string to;
cin>>id>>to;
if(link[id].length() > 0){
user[link[id]][id]=0;
nums[link[id]]--;
user[to][id]=1;
link[id]=to;
nums[to]++;
}else{
user[to][id]=1;
link[id]=to;
nums[to]++;
}
}else if(ch=='Q'){
queue<string>q;
string tmp;
cin>>tmp;
q.push(tmp);
int ans =0;
while(!q.empty()){
string now = q.front();
q.pop();
ans += nums[now];
for(auto it:super[now]){
if(it.second==1){
q.push(it.first);
}
}
}
cout<<ans<<"\n";
}
}
return 0;
}