結果

問題 No.399 動的な領主
ユーザー legendcn666legendcn666
提出日時 2024-10-13 00:28:23
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 119 ms / 2,000 ms
コード長 1,422 bytes
コンパイル時間 1,590 ms
コンパイル使用メモリ 169,656 KB
実行使用メモリ 28,660 KB
最終ジャッジ日時 2024-10-13 00:28:28
合計ジャッジ時間 3,955 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
6,984 KB
testcase_01 AC 3 ms
6,980 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 3 ms
7,112 KB
testcase_04 AC 3 ms
7,240 KB
testcase_05 AC 10 ms
9,780 KB
testcase_06 AC 113 ms
23,924 KB
testcase_07 AC 113 ms
23,996 KB
testcase_08 AC 108 ms
24,028 KB
testcase_09 AC 119 ms
24,016 KB
testcase_10 AC 4 ms
9,540 KB
testcase_11 AC 10 ms
9,968 KB
testcase_12 AC 87 ms
24,496 KB
testcase_13 AC 89 ms
24,560 KB
testcase_14 AC 72 ms
28,544 KB
testcase_15 AC 80 ms
28,660 KB
testcase_16 AC 87 ms
25,996 KB
testcase_17 AC 107 ms
24,064 KB
testcase_18 AC 112 ms
24,032 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# include <bits/stdc++.h>
using namespace std;
typedef long long ll; 
//# define int long long
# define lc u << 1
# define rc u << 1 | 1
# define fi first
# define se second
const int N = 100005;

int n, q;
vector <int> e[N];
int f[N][35], dep[N];
void dfs (int u, int fa)
{
	f[u][0] = fa, dep[u] = dep[fa] + 1;
	for (int v : e[u])
	{
		if (v == fa) continue;
		dfs (v, u);
	}
}
int getlca (int u, int v)
{
	if (dep[u] < dep[v]) swap (u, v);
	for (int i = 30; i >= 0; i -- )
	{
		if (dep[u] - dep[v] >= 1 << i)
			u = f[u][i];
	}
	if (u == v) return u;
	for (int i = 30; i >= 0; i -- )
	{
		if (f[u][i] != f[v][i])
			u = f[u][i], v = f[v][i];
	}
	return f[u][0];
}
ll cnt[N]; ll ans;
void dfs2 (int u, int fa)
{
	for (int v : e[u])
	{
		if (v == fa) continue;
		dfs2 (v, u);
	}
	if (fa >= 1) cnt[fa] += cnt[u];
	ans += cnt[u] * (cnt[u] + 1) / 2;
}
signed main ()
{
//	freopen ("tax.in", "r", stdin); freopen ("tax.out", "w", stdout);
	scanf ("%d", &n);
	for (int i = 1; i < n; i ++ )
	{
		int u, v; scanf ("%d%d", &u, &v); 
		e[u].push_back (v), e[v].push_back (u);
	}
	dfs (1, 0);
	for (int j = 1; j <= 30; j ++ )
	{
		for (int i = 1; i <= n; i ++ )
			f[i][j] = f[f[i][j - 1]][j - 1];
	}
	scanf ("%d", &q);
	while (q -- )
	{
		int u, v; scanf ("%d%d", &u, &v); 
		int lca = getlca (u, v);
		cnt[u] ++ , cnt[v] ++ , cnt[lca] -- ;
		if (lca != 1) cnt[f[lca][0]] -- ;
	}
	dfs2 (1, 0);
	printf ("%lld\n", ans);
	return 0;
}
0