智能护理中心统计 DFS+懒标记删除

智能护理中心统计

题目

智能护理中心系统将辖下的护理点分属若干个大区，例如华东区、华北区等；每个大区又分若干个省来进行管理；省又分市，等等。我们将所有这些有管理或护理功能的单位称为“管理结点”。现在已知每位老人由唯一的一个管理结点负责，每个管理结点属于唯一的上级管理结点管辖。你需要实现一个功能，来统计任何一个管理结点所负责照看的老人的数量。

注意这是一个动态问题，即随时可能有老人加入某个管理结点，并且老人是有可能从一个管理结点换到另一个管理结点去的。

输入格式：

输入在第一行中给出 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

代码长度限制

16 KB

Java (javac)

时间限制

2500 ms

内存限制

512 MB

其他编译器

时间限制

2500 ms

内存限制

64 MB

思路 DFS+懒删除

​ 比较简单的题，赛时光想着并查集了，后来发现数据爆搜完全够了，下次做PTA题得注意了，，，还是先看数据量在选择做题方法比较通用，具体思路上级向下级建立一条单向边，然后对字符串做一下hash，是数字记作权值为\(1\)的点，那么对于\(Q\)询问，直接搜子树权值就行了，比较简单，主要是将老人移交到另一个树中， 我们直接开个数组懒删除一下，然后dfs的时候查询是否被删除，被删除就不会搜下去了。

Code

#include <bits/stdc++.h>
#define ff first
#define ss second
#define INF 0x3f3f3f3f
typedef std::pair<int, int> PII;
using i64 = long long;

const int N = 1e6 + 10, M = 2e6 + 10;

int n, m, cnt;

int p[M], count[M], type[N];

std::unordered_map<std::string, int> idx;

std::vector<int> g[N];

bool st[N];

int tot, lazytag[N];

int getid(std::string t) {
	if (!idx.count(t)) {
		return idx[t] = ++ cnt;	
	}
	return idx[t];
}

bool isnumber(std::string t) {
	for (auto x: t) {
		if (!isdigit(x)) return false;
	}
	return true;
}

void dfs(int u) {
	if (lazytag[u]) return;
	
	tot += type[u];

	// std::cout << u << "\n";
	
	for (auto x: g[u]) {
		if (!st[x]) {
			st[x] = true;
			dfs(x);
		}
	}
}

int main() {
	std::cin >> n >> m;

	while (m --) {
		std::string t1, t2;

		std::cin >> t1 >> t2;

		int k1 = getid(t1), k2 = getid(t2);

		g[k2].push_back(k1);

		if (isnumber(t1)) {
			type[k1] = 1;
		}
	}

	std::string act;

	while (true) {
		std::cin >> act;

		std::string t1, t2;

		if (act == "E") break;

		if (act == "T") {
			std::cin >> t1 >> t2;

			// 懒删除并获取新的id
			lazytag[getid(t1)] = 1;
			idx[t1] = ++ cnt;

			type[cnt] = 1;
			g[getid(t2)].push_back(cnt);
			
		} else {
			memset(st, 0, sizeof st);
			tot = 0;

			std::cin >> t1;
			
			st[getid(t1)] = true;	
			dfs(getid(t1));

			std::cout << tot << "\n";
		}
	}	
}