#396607 (C++17(1z)) No.922 東北きりきざむたん

• 問題: No.922 東北きりきざむたん
• ユーザー: pockyny
• 提出日時: 2019-11-08 22:13:37
• 言語: C++17(1z) (gcc 10.1.0 + boost 1.73.0)
• 結果: WA .
• 実行時間: -
• コード長: 2,154 Byte
• コンパイル時間: 747 ms
• 使用メモリ: 32,944 KB
• 最終ジャッジ日時: 2020-07-02 18:42:15

入力	結果	実行時間; 使用メモリ
testcase_00	AC	10 ms; 13,408 KB
testcase_01	AC	10 ms; 13,400 KB
testcase_02	AC	11 ms; 13,392 KB
testcase_03	AC	11 ms; 13,408 KB
testcase_04	AC	11 ms; 13,508 KB
testcase_05	AC	11 ms; 13,460 KB
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	AC	28 ms; 24,240 KB
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	AC	251 ms; 27,172 KB
testcase_26	AC	259 ms; 27,180 KB
testcase_27	AC	248 ms; 27,180 KB
testcase_28	AC	151 ms; 32,944 KB
testcase_29	AC	236 ms; 31,280 KB

ソースコード

#include <iostream>
#include <vector>
#include <set>
using namespace std;
const int LOG = 22;
vector<int> G[100010],ver[100010];
int parent[25][100010];
int 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];
}

int 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);
}

vector<int> vec[100010];
set<int> s[100010];
void dfs2(int v, int p){
	if(s[find(v)].count(v)){
		vec[find(v)].push_back(v);
	}
	for(int u:G[v]){
		if(u==p) continue;
		dfs2(u,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),par[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{
			s[find(a)].insert(a);
			s[find(b)].insert(b);
		}
	}
	for(i=0;i<n;i++){
		if(par[i]!=i) continue;
		dfs2(i,-1);
		if(vec[i].size()<=1) continue;
		int x = vec[i][0],y = vec[i].back();
		int z = lca(x,y);
		for(int v:s[i]){
			ans += dist(v,z);
		}
	}
	cout << ans << endl;
}