oj.uz

Submission #1095002

# Submission time Handle Problem Language Result Execution time Memory
1095002 2024-10-01T07:07:14 Z blackslex Museum (CEOI17_museum) C++17
20 / 100
 346 ms 1048576 KB 
#include<bits/stdc++.h>

using namespace std;
using ll = long long;
using pii = pair<int, int>;

const int N = 1e4 + 5;
int n, k, c, x, y, z, sz[N];
ll dp[N][2][N];
vector<pii> v[N];

int main() {
    scanf("%d %d %d", &n, &k, &c);
    for (int i = 1; i < n; i++) {
        scanf("%d %d %d", &x, &y, &z);
        v[x].emplace_back(y, z); v[y].emplace_back(x, z);
    }
    for (int i = 1; i <= n; i++) {
        for (auto j: {0, 1}) {
            for (int k = 2; k <= n; k++) dp[i][j][k] = 1e18;
        }
    }
    function<void(int, int)> dfs = [&] (int cur, int par) {
        sz[cur] = 1;
        for (auto &[x, y]: v[cur]) {
            if (par == x) continue;
            dfs(x, cur);
            for (int i = sz[cur]; ~i; i--) {
                for (int j = sz[x]; ~j; j--) {
                    dp[cur][0][i + j] = min(dp[cur][0][i + j], dp[cur][1][i] + dp[x][0][j] + y);
                    dp[cur][0][i + j] = min(dp[cur][0][i + j], dp[cur][0][i] + dp[x][1][j] + y * 2);
                    dp[cur][1][i + j] = min(dp[cur][1][i + j], dp[cur][1][i] + dp[x][1][j] + y * 2);
                }
            }
            sz[cur] += sz[x];
        }
    };
    dfs(c, 0);
    printf("%lld", min(dp[c][0][k], dp[c][1][k]));
}

Compilation message

museum.cpp: In function 'int main()':
museum.cpp:13:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   13 |     scanf("%d %d %d", &n, &k, &c);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
museum.cpp:15:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   15 |         scanf("%d %d %d", &x, &y, &z);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 604 KB Output is correct
2 Correct 0 ms 860 KB Output is correct
3 Correct 0 ms 860 KB Output is correct
4 Correct 1 ms 860 KB Output is correct
5 Correct 0 ms 632 KB Output is correct

Subtask #2
0 / 25.0

# Verdict Execution time Memory Grader output
1 Runtime error 346 ms 1048576 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #3
0 / 35.0

# Verdict Execution time Memory Grader output
1 Runtime error 346 ms 1048576 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #4
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 604 KB Output is correct
2 Correct 0 ms 860 KB Output is correct
3 Correct 0 ms 860 KB Output is correct
4 Correct 1 ms 860 KB Output is correct
5 Correct 0 ms 632 KB Output is correct
6 Runtime error 346 ms 1048576 KB Execution killed with signal 9
7 Halted 0 ms 0 KB -