Zzuli OJ-----1907小火山的宝藏收益

最新推荐文章于 2023-03-20 23:37:28 发布

pupil_blue

最新推荐文章于 2023-03-20 23:37:28 发布

阅读量449

点赞数

CC 4.0 BY-SA版权

分类专栏： DFS/BFS

本文链接：https://blog.youkuaiyun.com/blue_hpu/article/details/52183224

DFS/BFS 专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一道经典的算法题目——小火山寻宝，玩家需要在一个由多个房间组成的地图中选择最优路径获取最大宝藏价值。文章提供了完整的C++代码实现，并详细解释了如何通过深度优先搜索(DFS)算法来解决这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1907: 小火山的宝藏收益

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 169 Solved: 47

Submit Status Web Board

Description

进去宝藏后，小火山发现宝藏有N个房间，且这n个房间通过N-1道门联通。

每一个房间都有一个价值为Ai的宝藏，但是每一个房间也都存在一个机关。如果小火山取走了这个房间的宝藏，那么这个房间通往其他房间的门就永远打不开了，也就是说后面的宝藏小火山是得不到了（进入这个房间的门是不会关闭的，小火山还是可以回去的）；如果小火山不取这个宝藏，而是去打开通往另一房间的门，那么这个房间的宝藏就会消失，小火山就得不到这个房间的宝藏。

不过，小火山已经有了藏宝图，知道每一个房间的宝藏的价值，现在想请你帮小火山算一下，他最多能获得多少钱去买股票？

Input

输入第一行是一个整数T（T <= 50), 表示一共有T组数据。

对于每一组数据，第一行是两个数N, S（1 <= N <= 10000, 1 <= S <= N), N代表有N个房间， S代表小火山进去宝藏后的

起始房间（小火山怎么进入起始房间不重要），第二行是N个数，代表每个房间宝藏的价值，随后N-1行，每行两个数A, B，代表

A, B这两个房间联通。

Output

对于每一组数据输出一个整数，代表小火山能获得的最大钱数。

Sample Input

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

Sample Output

20
6

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 10010
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
int val[maxn], head[maxn];
bool vis[maxn];
int num, ans, S, N;
struct node{
    int to, next;
}edge[maxn*2];
void add(int u, int v){
    edge[num].to = v;
    edge[num].next = head[u];
    head[u] = num++;
}
int dfs(int S){
    int cnt = 0;
    for(int i = head[S]; i != -1; i = edge[i].next){
        int u = edge[i].to;
        if(!vis[u]){
            vis[u] = true;
            cnt += dfs(u);
        }
    }
    return max(cnt, val[S]);
}
int main(){
	int t, m, n;
	scanf("%d", &t);
	while(t--){
        memset(val, 0, sizeof(val));
        memset(head, -1, sizeof(head));
        memset(vis, false, sizeof(vis));
        num = 0;
        scanf("%d%d", &N, &S);
        for(int i = 1; i <= N; i++){
            scanf("%d", &val[i]);
        }
        for(int i = 0; i < N-1; i++){
            scanf("%d%d", &m, &n);
            add(m, n);
            add(n, m);
        }
        vis[S] = true;
        ans = dfs(S);
        printf("%d\n", ans);
	}
	return 0;
}