D - Denouncing Mafia Gym - 102346D（树形dp）

我还是不会写树形dp之类的题目，这个题还是不是自己写出来的看了题解，不过好在看了题解一下就会写了，也算是一种小小的提升吧。
这道题求k条路经最多可以覆盖多少个点，这个的话最长的那条边肯定是要选的，选了最长的那条边之后，这个数就变成了一个森林了，那么现在就是再到这个森林里找最长的那条边，这种找法很容易就想到了优先队列维护，所以这个题的关键还是树形dp我们可以在dfs中也加上一个优先队列，每一次找到他的从儿子节点回溯回来之后加入优先队列，儿子全部遍历完之后，再把最长的那条链赋值给这个点，剩余的优先队列里的值减去1之后就等于除去最长链后长度,因为把最长那条链一直没有放入记录答案的那个优先队列，所以最后还要把dis[1]放入。（我应该讲清楚了吧？）可以直接上代码理解了

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define ll long long
#define ms(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x & -x
#define fi first
#define ull unsigned long long
#define se second
#define lson (rt<<1)
#define rson (rt<<1|1)
#define endl "\n"
#define bug cout<<"----acac----"<
#define IOS ios::sync_with_stdio(false), cin.tie(0),cout.tie(0)
using namespace std;
const int maxn = 1e5 + 10;
const int maxm = 1.5e5 + 50;
const double eps = 1e-18;
const double inf = 0x3f3f3f3f;
const ll lnf = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9+7;
const  double pi = 3.141592653589;
vector<int>ve[maxn];
int n, k,dis[maxn];
priority_queue<int>q;
void dfs(int x)
{
	priority_queue<int>s;
    dis[x] = 1;
	s.push(dis[x]);
	for (int v : ve[x])//遍历ve[x][0],ve[x][1]......
	{
		dfs(v);
		s.push(dis[v] + 1);
	}
	dis[x] = s.top();
	s.pop();
	while (s.size())
	{
		q.push(s.top() - 1);
		s.pop();
	}
}
int main()
{
	scanf("%d%d", &n, &k);
	for (int i = 2; i <= n; i++)
	{
		int u;
		scanf("%d", &u);
		ve[u].push_back(i);
	}
	dfs(1);
	q.push(dis[1]);
	int ans = 0;
	int len = q.size();
	for (int i = 1; i <= min(len,k ); i++)
	{
		ans += q.top();
		q.pop();
	}
	printf("%d\n", ans);
	return 0;
}