結果

問題 No.922 東北きりきざむたん
ユーザー QCFiumQCFium
提出日時 2019-11-08 22:34:14
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
RE  
実行時間 -
コード長 2,501 bytes
コンパイル時間 2,280 ms
コンパイル使用メモリ 181,356 KB
実行使用メモリ 14,448 KB
最終ジャッジ日時 2023-10-13 04:09:40
合計ジャッジ時間 8,845 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,356 KB
testcase_01 AC 1 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 1 ms
4,348 KB
testcase_04 AC 2 ms
4,352 KB
testcase_05 RE -
testcase_06 RE -
testcase_07 AC 2 ms
4,352 KB
testcase_08 RE -
testcase_09 AC 42 ms
10,104 KB
testcase_10 RE -
testcase_11 RE -
testcase_12 AC 18 ms
10,704 KB
testcase_13 RE -
testcase_14 RE -
testcase_15 AC 14 ms
10,716 KB
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int ri() {
	int n;
	scanf("%d", &n);
	return n;
}

struct UnionFind {
	std::vector<int> data;
	UnionFind(int n) : data(n, -1) {}
	int root(int i) { return data[i] < 0 ? i : (data[i] = root(data[i])); }
	void unite(int i, int j) {
		i = root(i);
		j = root(j);
		if (i == j) return;
		if (data[i] > data[j]) std::swap(i, j);
		data[i] += data[j];
		data[j] = i;
	}
	bool same(int i, int j) { return root(i) == root(j); }
};

int n, m;
std::vector<std::vector<int> > hen, parent;
std::vector<bool> used;
std::vector<int> cnt, sum, depth;

void dfs1(int i, int prev, int d) {
	parent[i].push_back(prev);
	{
		int cur = prev;
		for (int k = 0; k < (int) parent[cur].size(); k++) {
			parent[i].push_back(cur = parent[cur][k]);
		}
	}
	depth[i] = d;
	used[i] = true;
	int cur = 0;
	for (auto j : hen[i]) {
		if (j == prev) continue;
		dfs1(j, i, d + 1);
		cur += sum[j];
	}
	sum[i] = cur + cnt[i];
}
int dfs2(int i, int prev, int over) {
	for (auto j : hen[i]) {
		if (j == prev) continue;
		if (sum[j] >= over) return dfs2(j, i, over);
	}
	return i;
}
int64_t dfs3(int i, int prev, int dist) {
	int64_t res = 0;
	for (auto j : hen[i]) if (j != prev) res += dfs3(j, i, dist + 1);
	return res + (int64_t) dist * cnt[i];
}
int lca(int i, int j) {
	if (i == j) return i;
	if (depth[i] > depth[j]) std::swap(i, j);
	int dif = depth[j] - depth[i];
	for (int i = 0; i < 20; i++) if (dif >> i & 1) j = parent[j][i];
	if (i == j) return i;
	for (int k = 20; k >= 0; k--) {
		if (k < (int) parent[i].size() and parent[i][k] != parent[j][k]) {
			i = parent[i][k];
			j = parent[j][k];
		}
	}
	return parent[i][0];
}


int main() {
	n = ri(), m = ri();
	int q = ri();
	hen.resize(n);
	used.resize(n);
	cnt.resize(n);
	sum.resize(n);
	depth.resize(n);
	parent.resize(n);
	UnionFind uni(n);
	for (int i = 0; i < m; i++) {
		int a = ri() - 1;
		int b = ri() - 1;
		hen[a].push_back(b);
		hen[b].push_back(a);
		uni.unite(a, b);
	}
	std::vector<std::pair<int, int> > sames;
	for (int i = 0; i < q; i++) {
		int a = ri() - 1, b = ri() - 1;
		if (uni.same(a, b)) {
			sames.push_back({a, b});
		} else {
			cnt[a]++;
			cnt[b]++;
		}
	}
	int64_t res = 0;
	for (int i = 0; i < n; i++) {
		if (!used[i]) {
			used[i] = true;
			dfs1(i, -1, 0);
			int root = dfs2(i, -1, (sum[i] + 1) / 2);
			res += dfs3(root, -1, 0);
		}
	}
	for (auto &i : sames) {
		int lc = lca(i.first, i.second);
		res += depth[i.first] + depth[i.second] - 2 * depth[lc];
	}
	std::cout << res << std::endl;
	
	
	
	return 0;
}
0