oj.uz

Submission #554709

# Submission time Handle Problem Language Result Execution time Memory
554709 2022-04-29T08:49:45 Z Yazan_Alattar Dostavljač (COCI18_dostavljac) C++14
28 / 140
 166 ms 4376 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define F first
#define S second
#define pb push_back
#define endl "\n"
#define all(x) x.begin(), x.end()
const int M = 2007;
const ll inf = 2e9;
const ll mod = 1e9 + 7;
const double pi = acos(-1);
const double eps = 1e-6;
const int dx[] = {0, -1, 0, 1}, dy[] = {1, 0, -1, 0};
const int block = 320;

vector <int> adj[M];
ll n, m, dp[M][M][2], ans;

void dfs(int node, int p){
    for(auto i : adj[node]) if(i != p){
        dfs(i, node);

        for(int j = m; j; --j)
            for(int k = 1; k + 1 < j; ++k)
                dp[node][j][0] = max(dp[node][j][0], dp[node][j - k - 2][0] + dp[i][k][0]),
                dp[node][j][1] = max(dp[node][j][1], dp[node][j - k - 1][0] + dp[i][k][1]),
                dp[node][j][1] = max(dp[node][j][1], dp[node][j - k - 2][1] + dp[i][k][0]);
    }
    return;
}

int main(){
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> n >> m;
    for(int i = 1; i <= n; ++i) cin >> dp[i][1][0], dp[i][1][1] = dp[i][1][0];
    for(int i = 1; i < n; ++i){
        int a, b;
        cin >> a >> b;
        adj[a].pb(b);
        adj[b].pb(a);
    }

    dfs(1, 0);

    for(int i = 1; i <= m; ++i) for(int j = 0; j < 2; ++j) ans = max(ans, dp[1][i][j]);
    cout << ans << endl;

    return 0;
}

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 468 KB Output is correct

Subtask #2
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 512 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 596 KB Output is correct

Subtask #4
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 596 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 896 KB Output is correct
2 Incorrect 2 ms 852 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #6
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 1020 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 38 ms 1780 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 15 ms 1980 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 76 ms 3148 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 14.0

# Verdict Execution time Memory Grader output
1 Incorrect 166 ms 4376 KB Output isn't correct
2 Halted 0 ms 0 KB -