传送门
题意: 给你一棵n个节点的树,试标记其中k个节点为工业区,剩下的为旅游区,使得所有工业区走到1号根节点所经过的旅游区数量max。
思路:
- 将深度最大的k个节点标记为工业区即可。
- s[u]表示u节点子树上节点的数目。
- d[u]表示u节点的深度。
- 排序找到d[u]-s[u]最大的k歌节点即可。
代码实现:
#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {
{
1, 0}, {
-1, 0}, {
0, 1}, {
0, -1}};
using namespace std;
const int inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll mod = 1e9 + 7;
const int N = 2e5 + 5;int n, k, ans;
vector<int> g[N];
int s[N], d[N];void dfs(int u, int f){
s[u] = 1;for(auto v : g[u]){
if(v == f) continue;d[v] = d[u]+1;dfs(v, u);s[u] += s[v];}
}signed main(){
IOS;cin >> n >> k;for(int i = 1; i < n; i ++){
int u, v; cin >> u >> v;g[u].push_back(v);g[v].push_back(u);}dfs(1, -1);for(int i = 1; i <= n; i ++) s[i] = d[i]-(s[i]-1);sort(s+1, s+n+1);for(int i = n; i > n-k; i --) ans += s[i];cout << ans << endl;return 0;
}