oj.uz

Submission #347732

# Submission time Handle Problem Language Result Execution time Memory
347732 2021-01-13T10:48:44 Z parsabahrami Beads and wires (APIO14_beads) C++17
0 / 100
 4 ms 4972 KB 
// Call my Name and Save me from The Dark
#include <bits/stdc++.h>
 
using namespace std;

typedef long long int ll;
typedef pair<int, int> pii;
 
#define SZ(x)                       (int) x.size()
#define F                           first
#define S                           second

const int N = 2e5 + 10, MOD = 1e9 + 7;
int dp[2][N], n;
vector<pii> adj[N];

void DFS(int v, int p = -1) {
    dp[0][v] = 0, dp[1][v] = -MOD;
    for (pii u : adj[v]) {
        if (u.F != p) DFS(u.F, v);
    }
    int sum = 0;
    for (pii u : adj[v])
        if (u.F != p) {
            dp[0][v] = max(dp[0][v] + max(dp[0][u.F], dp[1][u.F] + u.S), dp[1][v] + dp[0][u.F] + u.S);
            dp[1][v] = max(dp[1][v] + max(dp[0][u.F], dp[1][u.F] + u.S), sum + dp[0][u.F] + u.S);
            sum += max(dp[0][u.F], dp[1][u.F] + u.S);
        }
    dp[0][v] = max(dp[0][v], sum);
    //printf("%d %d %d\n", v, dp[0][v], dp[1][v]);
}

int main() {
    scanf("%d", &n);
    for (int i = 1; i < n; i++) {
        int u, v, w; scanf("%d%d%d", &u, &v, &w);
        adj[u].push_back({v, w});
        adj[v].push_back({u, w});
    }
    DFS(1);
    printf("%d\n", dp[0][1]);
    return 0;
}

Compilation message

beads.cpp: In function 'int main()':
beads.cpp:34:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   34 |     scanf("%d", &n);
      |     ~~~~~^~~~~~~~~~
beads.cpp:36:27: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   36 |         int u, v, w; scanf("%d%d%d", &u, &v, &w);
      |                      ~~~~~^~~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 13.0

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

Subtask #2
0 / 15.0

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

Subtask #3
0 / 29.0

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

Subtask #4
0 / 43.0

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