Codeforces Round #635 (Div. 2) C. Linova and Kingdom

题目链接



题意:
给一个点数量为 n 的树，让其中 k 个结点变为工业城市，其余为旅游城市。而每个工业城市到根节点 1 的路径上存在的旅游城市数量之和求最大。

思路:
思路到是简单，就是我没有写出来。。
既然求最大，肯定是求最远的距离。到是问题就在，我怎么知道每个工业城市的前面有多少个旅游城市(因为他的前面也可能有旅游城市)。所以这里我们用dfs来搜到最底层，求出每个工业城市的前面有多少个旅游城市，再排序求前k个最大的和。

#include 
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;
}