제출 #1026618

#제출 시각아이디문제언어결과실행 시간메모리
102661812345678Chase (CEOI17_chase)C++17
70 / 100
231 ms94488 KiB
#include <bits/stdc++.h> using namespace std; const int nx=1e5+5, kx=105; #define ll long long ll n, k, b[nx], u, v, sm[nx], dp[nx][kx], res; vector<ll> d[nx]; void dfs(int u, int p) { sm[u]=0; for (auto v:d[u]) if (v!=p) sm[u]+=b[v]; for (int i=1; i<=k; i++) dp[u][i]=max(dp[p][i], dp[p][i-1]+sm[u]), res=max(res, dp[u][i]); //cout<<"debug "<<u<<' '<<i<<' '<<dp[u][i]<<'\n'; for (auto v:d[u]) if (v!=p) dfs(v, u); } int main() { cin.tie(NULL)->sync_with_stdio(false); cin>>n>>k; for (int i=1; i<=n; i++) cin>>b[i]; for (int i=1; i<n; i++) cin>>u>>v, d[u].push_back(v), d[v].push_back(u); for (int rt=1; rt<=((n<=1000)?n:1); rt++) { for (int i=0; i<=k; i++) for (int j=1; j<=n; j++) dp[j][i]=0; //cout<<"root "<<rt<<'\n'; dfs(rt, 0); } cout<<res; } /* 12 2 2 3 3 8 1 5 6 7 8 3 5 4 2 1 2 7 3 4 4 7 7 6 5 6 6 8 6 9 7 10 10 11 10 12 */
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...