結果

問題 No.922 東北きりきざむたん
ユーザー QCFiumQCFium
提出日時 2019-11-08 22:42:05
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 189 ms / 2,000 ms
コード長 2,518 bytes
コンパイル時間 4,786 ms
コンパイル使用メモリ 181,240 KB
実行使用メモリ 31,608 KB
最終ジャッジ日時 2023-10-13 04:20:54
合計ジャッジ時間 8,285 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 1 ms
4,356 KB
testcase_02 AC 1 ms
4,348 KB
testcase_03 AC 1 ms
4,356 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 1 ms
4,348 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 2 ms
4,348 KB
testcase_08 AC 2 ms
4,352 KB
testcase_09 AC 37 ms
9,288 KB
testcase_10 AC 25 ms
5,136 KB
testcase_11 AC 37 ms
8,068 KB
testcase_12 AC 12 ms
8,060 KB
testcase_13 AC 9 ms
4,372 KB
testcase_14 AC 56 ms
12,208 KB
testcase_15 AC 8 ms
8,424 KB
testcase_16 AC 117 ms
15,812 KB
testcase_17 AC 115 ms
16,016 KB
testcase_18 AC 117 ms
15,968 KB
testcase_19 AC 115 ms
16,088 KB
testcase_20 AC 117 ms
15,788 KB
testcase_21 AC 150 ms
19,612 KB
testcase_22 AC 143 ms
19,596 KB
testcase_23 AC 146 ms
17,752 KB
testcase_24 AC 152 ms
17,864 KB
testcase_25 AC 125 ms
17,160 KB
testcase_26 AC 123 ms
17,288 KB
testcase_27 AC 124 ms
17,092 KB
testcase_28 AC 29 ms
9,300 KB
testcase_29 AC 189 ms
31,608 KB
権限があれば一括ダウンロードができます

ソースコード

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) {
	if (prev != -1) {
		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