最后一个样例没过,但是A了……
查看原帖
最后一个样例没过,但是A了……
577955
Poitry楼主2022/10/19 22:13

代码:

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 100010;
int n, k, ans, maxn, rt, vis[MAXN], fa[MAXN], f[MAXN], flg;
vector<int>cd[MAXN], w[MAXN];
void dfs(int x, int s, int fath) {
//	cout << x << ' ' << s << endl;;
	if(s > maxn) {
		maxn = s;
		rt = x;
	}
	for(int i = 0; i < cd[x].size(); i ++) {
		int u = cd[x][i];
		if(u == fath) {
			continue;
		}
		if(flg) {
			fa[u] = x;
		}
		dfs(u, s + 1, x);
	}
}
void dp(int x, int fath) {
	for(int i = 0; i < cd[x].size(); i ++) {
		int u = cd[x][i];
		if(u == fath) {
			continue;
		}
		dp(u, x);
		ans = max(ans, f[x] + f[u] + w[x][i]);
		f[x] = max(f[x], f[u] + w[x][i]);
	}
}
signed main() {
//    freopen("xunluo7.in", "r", stdin);
//    freopen("xunluo.out", "w", stdout);
	cin >> n >> k;
	for(int i = 1; i < n; i ++) {
		int x, y;
		cin >> x >> y;
		cd[x].push_back(y);
		cd[y].push_back(x);
		w[x].push_back(1);
		w[y].push_back(1);
	}
	dfs(1, 0, 0);
	flg = 1;
	dfs(rt, 0, 0);
	if(k == 1) {
		cout << n * 2 - maxn - 1;
		return 0;
	}
	ans = n * 2 - maxn;
	int x = rt;
	while(fa[x]) {
		for(int i = 0; i < cd[fa[x]].size(); i ++) {
			int u = cd[fa[x]][i];
			if(u == x) {
				w[fa[x]][i] = -1;
				break;
			}
		}
		for(int i = 0; i < cd[x].size(); i ++) {
			int u = cd[x][i];
			if(u == fa[x]) {
				w[x][i] = -1;
				break;
			}
		}
		x = fa[x];
	}
	x = ans;
	ans = 0;
	dp(1, 0);
	cout << x - ans;
	return 0;
}

建议加强数据

2022/10/19 22:13
加载中...