题目链接
题意:
给一个点数量为 n 的树,让其中 k 个结点变为工业城市,其余为旅游城市。而每个工业城市到根节点 1 的路径上存在的旅游城市数量之和求最大。
思路:
思路到是简单,就是我没有写出来。。
既然求最大,肯定是求最远的距离。到是问题就在,我怎么知道每个工业城市的前面有多少个旅游城市(因为他的前面也可能有旅游城市)。所以这里我们用dfs来搜到最底层,求出每个工业城市的前面有多少个旅游城市,再排序求前k个最大的和。
#include <bits/stdc++.h>
using namespace std;
#define NewNode (ListNode *)malloc(sizeof(ListNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 3e5 + 5;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const unsigned long long mod = 998244353;
const int II = 3.1415926535;
typedef long long ll;
typedef unsigned long long ull;
struct node
{
int z;
int Next;
}Node[N*2];
int Head[N],dis[N],m,s[N];
void Add(int a,int b)
{
m++;
Node[m].z = b;
Node[m].Next = Head[a];
Head[a] = m;
}//链式前向星
void dfs(int u,int v,int h)//u到v路径,此节点高度为h
{
dis[u] = h;
s[u] = 1;//此时有一个工业城市
for(int i = Head[u];i;i = Node[i].Next)
{
if(Node[i].z == v)
continue;
dfs(Node[i].z,u,h+1);
// cout << endl;
// cout << u << " " << v << endl;
// cout << s[u] << " " << s[Node[i].z] << endl;
s[u] += s[Node[i].z];//前面一共有多少个工业城市
dis[u] -= s[Node[i].z];//高度减去前面的工业城市得到前面一共有多少个旅游城市
}
}
int main()
{
std::ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
int n,k;
cin >> n >> k;
for(int i = 0;i < n-1;i++)
{
int a,b;
cin >> a >> b;
Add(a,b);
Add(b,a);//无向图,存俩次
}
dfs(1,0,0);
sort(dis+1,dis+n+1, greater<int>() );
ll sum = 0;
for(int i = 1;i <= k;i++)
sum += dis[i];
cout << sum << endl;
}