垃圾ACMer的暑假训练220711

垃圾ACMer的暑假训练220711

DFS判奇环、dsu

14.1.2 Split Into Two Sets ($2\mathrm{s}$)

题意

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据输入一个偶数$n(2\le n\le 2\mathrm{e}5)$,表示牌数.每张牌上有$1\sim n$中的两个数字.现需将牌分为两个集合,使得每个集合中牌上的数字不重复,若能做到,输出"YES",否则输出"NO".数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$.

思路I:图匹配

显然一张牌的两数字相同,或$1\sim n$中有数字出现超过$2$次时无解.

若存在牌 { a , b } \{a,b\} {a,b},则节点$a$与$b$间连双向边,并更新每个节点的度数.遍历一遍节点$1\sim n$,检查每个节点的度数是否为$2$,若不为$2$,则不满足.同时记录连通的节点数$cnt$,若遍历完后$cnt$为奇数,则无解.

代码I

const int MAXN = 2e5 + 5;
int n;

int main() {
	CaseT{
		cin >> n;

		bool ok = true;
		multiset<int> ss;
		vi edges[n + 1], deg(n + 1);
		for (int i = 0; i < n; i++) {
			int a, b; cin >> a >> b;
		
			if (ok && a == b) ok = false;
			
			if (ok) {
				ss.insert(a), ss.insert(b);
				edges[a].push_back(b), edges[b].push_back(a);  // 建双向边
				deg[a]++, deg[b]++;  // 更新节点度数
			}
		}
		
		for (int i = 1; i <= n; i++) {
			if (ss.count(i) != 2) {
				ok = false;
				break;
			}
		}

		if (!ok) {
			cout << "NO" << endl;
			continue;
		}

		vb vis(n + 1);
		int cnt = 0;  // 遍历到的节点数
		bool flag = true;

		function<void(int)> dfs = [&](int u)->void {  // 遍历一遍连通的节点,检查度数
			if (vis[u]) return;

			vis[u] = true;
			if (deg[u] != 2) flag = false;
			cnt++;
			for (int v : edges[u]) dfs(v);
		};

		for (int i = 1; i <= n; i++) {
			if (!vis[i]) {
				cnt = 0;
				flag = true;
				dfs(i);

				if (flag && (cnt & 1)) {
					ok = false;
					break;
				}
			}
		}
		
		cout << (ok ? "YES" : "NO") << endl;
	}
}

---

思路II:扩展域并查集

用扩展域并查集维护每个数属于哪个集合.

代码II

const int MAXN = 2e5 + 5;
int n;
int fa[MAXN << 1];  // 扩展域并查集

int find(int x) { return x == fa[x] ? x : fa[x] = find(fa[x]); }

void merge(int x, int y) {
	x = find(x), y = find(y);
	if (x != y) fa[x] = y;
}

int main() {
	CaseT{		
		cin >> n;
		for (int i = 1; i <= n << 1; i++) fa[i] = i;

		vi deg(n + 1);
		for (int i = 0; i < n; i++) {
			int a, b; cin >> a >> b;
			deg[a]++, deg[b]++;
			merge(a, b + n), merge(a + n, b);
		}

		bool ok = true;
		for (int i = 1; i <= n; i++) {
			if (deg[i] > 2) ok = false;
			if (find(i) == find(i + n)) ok = false;
		}

		cout << (ok ? "YES" : "NO") << endl;
	}
}

---

---

Trie树

13.2.5 Equate Multisets ($4\mathrm{s}$)

题意

定义两有重集合相等为两集合的元素的出现次数相等.

有$t(1\le t\le 1\mathrm{e}4)$组测试数据.每组测试数据第一行输入一个$n(1\le n\le 2\mathrm{e}5)$,表示有重集合$a$和$b$的元素.第二行输入$n$个数$a_1,\cdots ,a_n(1\le a_i\le 1\mathrm{e}9)$,表示集合$a$的元素.第三行输入$n$个数$b_1,\cdots ,b_n(1\le b_i\le 1\mathrm{e}9)$,表示集合$b$的元素.数据保证所有测试数据的$n$之和不超过$2\mathrm{e}5$.先只对集合$b$进行如下两种操作,使其与$a$相等:对$\forall x\in b$,①将$x$替换为$2x$;②将$x$替换为$\lfloor \frac{x}{2}\rfloor$.若能通过若干次操作(可能是$0$次)使得集合$b$与$a$相等,输出"YES";否则输出"NO".

思路

将每个数转为二进制的字符串,建Trie树.将两个操作转化为对Trie树节点的操作.

代码

const int MAXN = 2e5 + 5;
int n;

void solve() {
	cin >> n;
	vi a(n), b(n);
	for (int i = 0; i < n; i++) cin >> a[i];
	for (int i = 0; i < n; i++) cin >> b[i];

	vector<vi> trie(1, vi(2, -1));
	vi val(1, 0);
	for (int i = 0; i < 2 * n; i++) {
		int x = (i < n ? b[i] : a[i - n]);
		int sgn = (i < n ? 1 : -1);
		while (x % 2 == 0) x >>= 1;

		int bit = 30;
		while (!(x & (1 << bit))) bit--;  // bit移动到最高位1
		int t = 0;
		for (int j = bit; j >= 0; j--) {  // 扫一遍数的二进制表示,建trie树
			int d = (x >> j) & 1;
			if (trie[t][d] == -1) {  // 未用过的节点
				trie[t][d] = (int)trie.size();
				trie.emplace_back(2, -1);
				val.push_back(0);
			}
			t = trie[t][d];
			val[t] += sgn;
			if (val[t] < 0) {
				cout << "NO" << endl;
				return;
			}
		}
	}
	cout << "YES" << endl;
}

int main() {
	CaseT{
		solve();
	}
}

---

---