답안 #241511

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
241511 2020-06-24T10:42:12 Z NONAME Dostavljač (COCI18_dostavljac) C++14
0 / 140
338 ms 3304 KB
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
using pll = pair <ll, ll>;

const ll MX = 2e9;
const ll N = 1e5 + 500;

int n, m;
ll a[510], f[510][510][2];
vector <int> g[510];

void upd(ll &a, ll b) { a = max(a, b); }

void dfs(int v, int pr) {
    f[v][1][0] = a[v];

    for (int u : g[v]) {
        if (u == pr)
            continue;

        dfs(u, v);

        for (int tv = m; tv > 0; --tv)
        for (int tu = m; tu > 0; --tu) {
            if (tv + tu + 2 <= m) {
                upd(f[v][tv + tu + 2][0], f[v][tv][0] + f[u][tu][0]);
                upd(f[v][tv + tu + 2][1], f[v][tv][1] + f[u][tu][0]);
            }

            if (tv + tu + 1 <= m)
                upd(f[v][tv + tu + 1][1], f[v][tv][0] + max(f[u][tu][0], f[u][tu][1]));
        }
    }
}

int main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

//    #ifdef _LOCAL
//        freopen("in", "r", stdin);
//        freopen("out", "w", stdout);
//    #endif // _LOCAL

    cin >> n >> m;
    for (int i = 0; i < n; ++i)
        cin >> a[i];

    for (int i = 0; i < n - 1; ++i) {
        int x, y;
        cin >> x >> y;
        --x, --y;

        g[x].push_back(y);
        g[y].push_back(x);
    }

    dfs(0, 0);

    ll ans = 0;

    for (int i = 0; i <= m; ++i)
    for (int t = 0; t < 2; ++t)
        upd(ans, f[0][i][t]);

    cout << ans << "\n";
}
# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 384 KB Output isn't correct
# 결과 실행 시간 메모리 Grader output
1 Incorrect 5 ms 512 KB Output isn't correct
# 결과 실행 시간 메모리 Grader output
1 Incorrect 6 ms 640 KB Output isn't correct
# 결과 실행 시간 메모리 Grader output
1 Incorrect 5 ms 640 KB Output isn't correct
# 결과 실행 시간 메모리 Grader output
1 Incorrect 14 ms 896 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 8 ms 1152 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 93 ms 1400 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 35 ms 2092 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 177 ms 2808 KB Output isn't correct
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Incorrect 338 ms 3304 KB Output isn't correct
2 Halted 0 ms 0 KB -