結果

問題 No.922 東北きりきざむたん
ユーザー pockynypockyny
提出日時 2019-11-08 23:44:42
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 362 ms / 2,000 ms
コード長 2,547 bytes
コンパイル時間 1,081 ms
コンパイル使用メモリ 90,352 KB
実行使用メモリ 34,128 KB
最終ジャッジ日時 2023-10-13 05:27:16
合計ジャッジ時間 7,004 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
24,068 KB
testcase_01 AC 8 ms
24,072 KB
testcase_02 AC 8 ms
24,012 KB
testcase_03 AC 8 ms
24,272 KB
testcase_04 AC 9 ms
24,140 KB
testcase_05 AC 8 ms
24,144 KB
testcase_06 AC 9 ms
24,244 KB
testcase_07 AC 8 ms
24,064 KB
testcase_08 AC 9 ms
24,096 KB
testcase_09 AC 90 ms
29,288 KB
testcase_10 AC 74 ms
25,864 KB
testcase_11 AC 86 ms
28,652 KB
testcase_12 AC 35 ms
28,340 KB
testcase_13 AC 29 ms
26,048 KB
testcase_14 AC 142 ms
32,864 KB
testcase_15 AC 21 ms
27,376 KB
testcase_16 AC 297 ms
33,648 KB
testcase_17 AC 289 ms
33,808 KB
testcase_18 AC 300 ms
33,716 KB
testcase_19 AC 294 ms
33,768 KB
testcase_20 AC 297 ms
33,544 KB
testcase_21 AC 362 ms
34,128 KB
testcase_22 AC 356 ms
34,036 KB
testcase_23 AC 243 ms
28,636 KB
testcase_24 AC 219 ms
27,908 KB
testcase_25 AC 189 ms
28,632 KB
testcase_26 AC 182 ms
28,484 KB
testcase_27 AC 183 ms
28,560 KB
testcase_28 AC 113 ms
33,028 KB
testcase_29 AC 166 ms
32,060 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <map>
#include <cassert>
using namespace std;
const int LOG = 22;
vector<int> G[100010],ver[100010];
int parent[25][100010];
long long d[100010];
void dfs(int v, int p){
	for(int x:G[v]){
		if(x==p) continue;
		parent[0][x] = v;
		d[x] = d[v] + 1;
		dfs(x,v);
	}
}

void init(int v,int root){
	d[root] = 0; parent[0][root] = -1;
	dfs(root,-1);
	for(int k=0;k+1<LOG;k++){
		for(int i:ver[v]){
			if(parent[k][i]<0){
				parent[k+1][i] = -1;
			}else{
				parent[k+1][i] = parent[k][parent[k][i]];
			}
		}
	}
}

int lca(int u, int v){
	if(d[u]>d[v]) swap(u,v);
	for(int k = 0;k<LOG;k++){
		if((d[v] - d[u]) & (1<<k)){
			v = parent[k][v];
		}
	}
	if(u==v) return u;
	for(int k = LOG-1;k>=0;k--){
		if(parent[k][u]!=parent[k][v]){
			u = parent[k][u];
			v = parent[k][v];
		}
	}
	return parent[0][u];
}

long long dist(int u, int v){
	int z = lca(u,v);
	return d[u] + d[v] - 2*d[z];
}

int par[100010],sz[100010];
void unite_init(int n){
	for(int i=0;i<n;i++){
		par[i] = i; sz[i] = 1;
	}
}

int find(int x){
	if(par[x]==x) return x;
	return par[x] = find(par[x]);
}

void unite(int x, int y){
	x = find(x); y = find(y);
	if(x==y) return;
	if(sz[x]<sz[y]) swap(x,y);
	sz[x] += sz[y]; par[y] = x;
}

bool same(int x, int y){
	return find(x)==find(y);
}

map<int,long long> ma[100010];
int dp[100010];
int dfs2(int v, int p){
	int ret = 0,x = find(v);
	if(ma[x].find(v)!=ma[x].end()){
		ret += ma[x][v];
	}
	for(int u:G[v]){
		if(u==p) continue;
		ret += dfs2(u,v);
	}
	dp[v] = ret;
	return ret;
}
int sum[100010] = {};
int search(int v,int p){
	int mx = -1,j = -1;
	for(int u:G[v]){
		if(u==p) continue;
		if(mx<dp[u]){
			mx = dp[u];
			j = u;
		}
	}
	if(mx>sum[find(v)] - dp[j]) return search(j,v);
	return v;
}

int main(){
	int i,n,m,q;
	cin >> n >> m >> q;
	unite_init(n);
	for(i=0;i<m;i++){
		int u,v;
		cin >> u >> v; u--; v--;
		G[u].push_back(v); G[v].push_back(u);
		unite(u,v);
	}
	for(i=0;i<n;i++){
		ver[find(i)].push_back(i);
	}
	for(i=0;i<n;i++){
		if(par[i]==i){
			init(find(i),find(i));
		}
	}
	long long ans = 0;
	for(i=0;i<q;i++){
		int a,b; cin >> a >> b;
		a--; b--;
		if(same(a,b)){
			ans += dist(a,b);
		}else{
			ma[find(a)][a]++; ma[find(b)][b]++;
		}
	}
	for(i=0;i<n;i++){
		if(par[i]==i){
			for(auto x:ma[find(i)]){
				sum[find(i)] += x.second;
			}
		}
	}
	for(i=0;i<n;i++){
		if(par[i]!=i) continue;
		if(ma[find(i)].size()<=1) continue;
		dfs2(i,-1);
		int z = search(i,-1);
		for(auto v:ma[find(i)]){
			ans += v.second*dist(v.first,z);
		}
	}
	cout << ans << endl;
}
0