打卡信奥刷题（2058）用C++实现信奥 P11378 [GESP202412 七级] 燃烧

P11378 [GESP202412 七级] 燃烧

题目描述

小杨有一棵包含 $n$ 个节点的树，其中节点的编号从 $1$ 到 $n$。节点 $i$ 的权值为 $a_i$。

小杨可以选择一个初始节点引燃，每个燃烧的节点会将其相邻节点中权值严格小于自身权值的在节点间扩散直到不会有新的节点被引燃。

小杨想知道在合理选择初始节点的情况下，最多可以燃烧多少个节点。

输入格式

第一行包含一个正整数 $n$，表示节点数量。

第二行包含 $n$ 个正整数 $a_1,a_2,\dots ,a_n$，代表节点权值。

之后 $n-1$ 行，每行包含两个正整数 $u_i,v_i$，代表存在一条连接节点 $u_i$ 和 $v_i$ 的边。

输出格式

输出一个正整数，代表最多燃烧的节点个数。

输入输出样例 #1

输入 #1

5
6 2 3 4 5
1 2
2 3
2 5
1 4

输出 #1

3

说明/提示

对于全部数据，保证有 $1\le n\le 10^5$，$1\le a_i\le 10^6$。

C++实现

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
int n,a[1000006],vis[1000006];
vector<int> edge[1000006];
int dfs(int x){
	if(vis[x] != 0) return vis[x];
	int cnt = 0;
	for(auto &v : edge[x])
		if(a[x] > a[v]) //如果连接的点比自己的小 
			cnt += dfs(v); //先统计 
	return vis[x] = cnt + 1;
}
main(){
	cin >> n;
	for(int i = 1;i <= n; i++) cin >> a[i];
	for(int i = 1,x,y;i < n; i++){
		cin >> x >> y;
		edge[x].push_back(y);
		edge[y].push_back(x);
	}
	int ans = 0;
	for(int i = 1;i <= n; i++)
		ans = max(ans,dfs(i));
	cout << ans;
} 

后续

接下来我会不断用C++来实现信奥比赛中的算法题、GESP考级编程题实现、白名单赛事考题实现，记录日常的编程生活、比赛心得，感兴趣的请关注，我后续将继续分享相关内容