제출 #1358225

#제출 시각아이디문제언어결과실행 시간메모리
1358225goulthenChase (CEOI17_chase)C++20
30 / 100
158 ms167360 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long
#define rep(i,a,b) for (int i = a; i <= b; i++)
#define per(i,a,b) for (int i = a; i >= b; i--)
#define pii pair<int,int>
#define fi first
#define se second
#define pb push_back

const int MAXN = 1e5+10;
const int INF = 1e9+67;
const int MOD = 1e9+7;
int a[MAXN], dp[MAXN][200], val[MAXN],k;
vector<int> g[MAXN];

void calc(int u, int p = -1) {
    for(int &v : g[u]) if(v!=p) {
        val[u] += a[v];
        calc(v,u);
    }
}

void dfs(int u,int p = -1) {
    for (int &v : g[u]) if (v!=p) {
        dfs(v,u);
    }

    rep(ki,1,k) {
        for(int &v : g[u]) dp[u][ki] = max({dp[u][ki], dp[v][ki], dp[v][ki-1] + val[u]});
    }
}

void solve() {
    int n; cin >> n >> k;
    rep(i,1,n) cin >> a[i];
    rep(i,1,n-1) {
        int u,v; cin >> u >> v;
        g[u].pb(v);
        g[v].pb(u);
    }
    calc(1);
    dfs(1);

    cout << dp[1][k] << endl;
}

int32_t main() {
    ios_base::sync_with_stdio(0);cin.tie(nullptr);
    int tt=1; 
    //cin >> tt;

    while (tt--) solve();
}
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…
#결과 실행 시간메모리채점기 출력
결과를 불러오는 중입니다…