比赛 202504月赛 评测结果 WWWWW
题目名称 搜城探宝 最终得分 0
用户昵称 NahidaOI 运行时间 0.015 s
代码语言 C++ 内存使用 3.42 MiB
提交时间 2025-04-22 15:11:32
显示代码纯文本
#include<bits/stdc++.h>
#define int long long
#define Sangonomiya signed
#define Kokomi main()
#define Love return
#define Nahida 0
#define Forever ;
#define IOS cin.tie(nullptr)->sync_with_stdio(false)
#define cin std::cin
#define cout std::cout
const int N=1e6;
struct val{
    int v,cnt;
};
val w[N];
struct node{
    int v,next;
};
node G[N];
int cnt,head[N];
int ans,nowans;
void add(int u,int v){
    G[++cnt].next=head[u];
    G[cnt].v=v;
    head[u]=cnt;
}
int n,k;
void dfs(int x,int f,int step){
    for(int i=head[x];i;i=G[i].next){
        int v=G[i].v;
        if(v==f) continue;
        nowans+=w[i].v;
        w[i].cnt++;
        dfs(v,x,step--);
        if(step==0) break;
    }
}
Sangonomiya Kokomi{
	freopen("hzoi_key.in","r",stdin);
    freopen("hzoi_key.out","w",stdout);
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        int u,v;
        add(u,v);
    }
    for(int i=1;i<=n;i++){
        cin>>w[i].v;
        w[i].cnt=0;
    } 
    for(int i=1;i<=n;i++){
        nowans=0;
        dfs(1,n,k);
        ans=std::max(ans,nowans);
    }
    int maxs=-1;
    for(int i=1;i<=n;i++){
        if(w[i].cnt==0 && w[i].v>maxs) maxs=w[i].v;
    }
    cout<<ans+maxs;
	Love Nahida Forever;
}