結果

問題 No.2337 Equidistant
ユーザー 👑 ygussanyygussany
提出日時 2023-06-02 22:45:10
言語 C
(gcc 12.3.0)
結果
AC  
実行時間 632 ms / 4,000 ms
コード長 2,864 bytes
コンパイル時間 501 ms
コンパイル使用メモリ 32,896 KB
実行使用メモリ 29,184 KB
最終ジャッジ日時 2024-06-09 00:29:54
合計ジャッジ時間 7,336 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 3 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 206 ms
29,184 KB
testcase_12 AC 205 ms
29,056 KB
testcase_13 AC 204 ms
29,184 KB
testcase_14 AC 196 ms
29,184 KB
testcase_15 AC 191 ms
29,184 KB
testcase_16 AC 203 ms
29,184 KB
testcase_17 AC 202 ms
29,184 KB
testcase_18 AC 208 ms
29,184 KB
testcase_19 AC 204 ms
29,056 KB
testcase_20 AC 204 ms
29,184 KB
testcase_21 AC 322 ms
29,184 KB
testcase_22 AC 168 ms
29,056 KB
testcase_23 AC 193 ms
29,184 KB
testcase_24 AC 431 ms
29,184 KB
testcase_25 AC 198 ms
29,184 KB
testcase_26 AC 632 ms
29,184 KB
testcase_27 AC 210 ms
29,184 KB
testcase_28 AC 226 ms
29,184 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>

typedef struct Edge {
	struct Edge *next;
	int v;
} edge;

void init_LCA(int N, edge* adj[], int par[][20], int depth[])
{
	static int i, u, w, x, q[200001], head, tail;
	static edge *p;
	for (u = 1, depth[0] = -1; u <= N; u++) {
		for (i = 0; i < 20; i++) par[u][i] = 0;
		depth[u] = -1;
	}
	par[1][0] = 0;
	depth[1] = 0;
	q[0] = 1;
	for (head = 0, tail = 1; head < tail; head++) {
		u = q[head];
		for (p = adj[u]; p != NULL; p = p->next) {
			w = p->v;
			if (depth[w] < 0) {
				par[w][0] = u;
				depth[w] = depth[u] + 1;
				for (i = 0, x = u; x != 0; x = par[x][i++]) par[w][i+1] = par[x][i];
				q[tail++] = w;
			}
		}
	}
}

int get_LCA(int par[][20], int depth[], int u, int w)
{
	int i;
	while (depth[u] > depth[w]) {
		for (i = 1; depth[par[u][i]] >= depth[w]; i++); 
		u = par[u][i-1];
	}
	while (depth[u] < depth[w]) {
		for (i = 1; depth[par[w][i]] >= depth[u]; i++); 
		w = par[w][i-1];
	}
	while (u != w) {
		for (i = 1; par[u][i] != par[w][i]; i++); 
		u = par[u][i-1];
		w = par[w][i-1];
	}
	return u;
}

int main()
{
	int i, N, Q, u, w;
	edge *adj[200001] = {}, e[400001], *p;
	scanf("%d %d", &N, &Q);
	for (i = 0; i < N - 1; i++) {
		scanf("%d %d", &u, &w);
		e[i*2].v = w;
		e[i*2+1].v = u;
		e[i*2].next = adj[u];
		e[i*2+1].next = adj[w];
		adj[u] = &(e[i*2]);
		adj[w] = &(e[i*2+1]);
	}
	
	static int par[200001] = {}, size[200001] = {}, q[200001], head, tail;
	par[1] = 1;
	q[0] = 1;
	for (head = 0, tail = 1; head < tail; head++) {
		u = q[head];
		for (p = adj[u]; p != NULL; p = p->next) {
			w = p->v;
			if (par[w] == 0) {
				par[w] = u;
				q[tail++] = w;
			}
		}
	}
	for (head--; head >= 0; head--) {
		u = q[head];
		size[u]++;
		if (head > 0) size[par[u]] += size[u];
	}
	
	int s, t, r;
	static int depth[200001], LCA_par[200001][20];
	init_LCA(N, adj, LCA_par, depth);
	while (Q--) {
		scanf("%d %d", &s, &t);
		if ((depth[s] + depth[t]) % 2 != 0) {
			printf("0\n");
			continue;
		}
		r = get_LCA(LCA_par, depth, s, t);
		if (depth[s] < depth[t]) {
			w = t;
			while (depth[w] > (depth[t] - depth[s]) / 2 + depth[r] + 1) {
				for (i = 1; depth[LCA_par[w][i]] > (depth[t] - depth[s]) / 2 + depth[r]; i++); 
				w = LCA_par[w][i-1];
			}
			printf("%d\n", size[par[w]] - size[w]);
		} else if (depth[s] > depth[t]) {
			u = s;
			while (depth[u] > (depth[s] - depth[t]) / 2 + depth[r] + 1) {
				for (i = 1; depth[LCA_par[u][i]] > (depth[s] - depth[t]) / 2 + depth[r]; i++); 
				u = LCA_par[u][i-1];
			}
			printf("%d\n", size[par[u]] - size[u]);
		} else {
			u = s;
			w = t;
			while (depth[u] > depth[r] + 1) {
				for (i = 1; depth[LCA_par[u][i]] > depth[r]; i++); 
				u = LCA_par[u][i-1];
			}
			while (depth[w] > depth[r] + 1) {
				for (i = 1; depth[LCA_par[w][i]] > depth[r]; i++); 
				w = LCA_par[w][i-1];
			}
			printf("%d\n", N - size[u] - size[w]);
		}
	}
	fflush(stdout);
	return 0;
}
0