【学军NOIP开放题2-B】托萨妮娅（树）（dfs）

托萨妮娅

题目链接：学军NOIP开放题2-B

题目大意

给你 k 个 n 个点的树。
然后问你对于每一对点，有多少个点存在于每个树中这一对点的路径上。

思路

考虑用这么一个性质：
$dis_{x,y}\leqslant dis_{x,z}+dis_{z,k}$，等号成立当且仅当 $z$ 在 $x,y$ 的路径上。

那要每个图都满足，就是要 $\sum dis_{x,y}=\sum dis_{x,z}+\sum dis_{z,k}$。

那我们就可以求出每个图任意两点之间的距离，然后就可以暴力判断了。

然后求距离可以直接枚举 $x$ 然后 dfs。

代码

#include<cstdio>
#include<iostream>
#define ll long long

using namespace std;

int n, k, x, y, all_dis[501][501];
ll ans;
struct node {
	int to, nxt;
};
struct Graph {
	node e[1001];
	int le[501], KK, deg[501];
	
	void add(int x, int y) {
		e[++KK] = (node){y, le[x]}; le[x] = KK;
		e[++KK] = (node){x, le[y]}; le[y] = KK;
	}
	
	void dfs(int st, int now, int father) {
		all_dis[st][now] += deg[now];
		for (int i = le[now]; i; i = e[i].nxt)
			if (e[i].to != father) {
				deg[e[i].to] = deg[now] + 1;
				dfs(st, e[i].to, now);
			}
	}
}g[501];

int main() {
//	freopen("tosania.in", "r", stdin);
//	freopen("tosania.out", "w", stdout);
	
	scanf("%d %d", &n, &k);
	for (int i = 1; i <= k; i++) {
		for (int j = 1; j < n; j++) {
			scanf("%d %d", &x, &y);
			g[i].add(x, y);
		}
	}
	
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			g[i].deg[j] = 0;
			g[i].dfs(j, j, 0);
		}
	}
	
	for (int i = 1; i <= n; i++) {
		ans = 0;
		for (int j = 1; j <= n; j++) {
			for (int k = 1; k <= n; k++)
				if (all_dis[i][j] == all_dis[i][k] + all_dis[k][j]) {
					ans = ans + j;
				}
		}
		printf("%lld\n", ans);
	}
	
	return 0;
}

__EOF__

本文作者：あおいSakura
本文链接：https://www.cnblogs.com/Sakura-TJH/p/XJOI_3412_4.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！