比赛 |
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;
}