Submission #650686

# Submission time Handle Problem Language Result Execution time Memory
650686 2022-10-14T16:32:35 Z Spade1 Islands (IOI08_islands) C++14
64 / 100
359 ms 131072 KB
#include<bits/stdc++.h>
#define pii pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define ld long double
#define st first
#define nd second
#define pb push_back
#define INF INT_MAX
using namespace std;

const int N = 1e6 + 10;

vector<pair<int, pll>> adj[N];
bool vis[N];

int cycle;

vector<pll> v;
ll dp[N], qs[N], ans, cur;

pll dfs2(int i, int prt1, int prt2) {
    ll mx_w = 0, mx = i;
    for (auto [j, w] : adj[i]) {
        if (j == prt1 || j == prt2) continue;
        pll ret = dfs2(j, i, 0);
        if (ret.nd + w.nd > mx_w) {
            mx_w = ret.nd + w.nd;
            mx = ret.st;
        }
    }
    return {mx, mx_w};
}

ll dfs3(int i, int prt1, int prt2, int prt3) {
    ll mx = 0;
    for (auto [j, w] : adj[i]) {
        if (j == prt1 || j == prt2 || j == prt3) continue;
        ll mx = max(mx, dfs3(j, i, prt2, prt3));
    }
    return mx;
}

int dfs(int i, int ed, int prt) {
    if (vis[i]) {
        cycle = prt;
        return i;
    }
    vis[i] = 1;
    int ret = 0, K, W;
    for (auto [j, w] : adj[i]) {
        if (j == prt && ed == w.st) continue;
        if (ret != 0 && vis[j]) continue;
        int k = dfs(j, w.st, i);
        if (k) {
            ret = k;
            K = j;
            W = w.nd;
        }
    }
    if (ret == i) {
        auto [val, w] = dfs2(i, K, cycle);
        cur = max(cur, dfs3(val, 0, K, cycle));
        v.pb({w, W});
        return 0;
    }
    else if (ret) {
        auto [val, w] = dfs2(i, prt, K);
        cur = max(cur, dfs3(val, 0, prt, K));
        v.pb({w, W});
    }
    return ret;
}

void solve() {
    int n; cin >> n;
    for (int i = 1; i <= n; ++i) {
        int b, w; cin >> b >> w;
        adj[i].pb({b ,{i, w}});
        adj[b].pb({i, {i, w}});
    }

    for (int i = 1; i <= n; ++i) {
        if (vis[i]) continue;
        dfs(i, 0, 0);
        int m = v.size();
        for (int j = 0; j <= m+1; ++j) qs[j] = 0, dp[j] = 0;
        for (int j = 1; j <= m; ++j) {
            qs[j] = qs[j-1] + v[j-1].nd;
            dp[j] = qs[j] + v[j-1].st;
        }
        for (int j = m-1; j > 0; --j) dp[j] = max(dp[j], dp[j+1]);
        for (int j = 1; j < m; ++j) {
            cur = max(cur, v[j-1].st - qs[j] + dp[j+1]);
        }
        for (int j = 0; j <= m+1; ++j) dp[j] = 0;
        for (int j = m; j >= 1; --j) {
            dp[j] = qs[j] + v[j-1].st;
        }
        for (int j = 2; j <= m; ++j) dp[j] = max(dp[j], dp[j-1]);
        for (int j = 2; j <= m; ++j) {
            cur = max(cur, qs[m] + v[j-1].st - qs[j] + dp[j-1]);
        }
        ans += cur;
        v.clear();
        cur = 0, cycle = 0;
    }
    cout << ans << '\n';
}

int main() {
    ios_base::sync_with_stdio(0); cin.tie(NULL);
    int t = 1;
//    cin >> t;
    while (t--) {
        solve();
    }
}

Compilation message

islands.cpp: In function 'std::pair<long long int, long long int> dfs2(int, int, int)':
islands.cpp:24:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   24 |     for (auto [j, w] : adj[i]) {
      |               ^
islands.cpp: In function 'long long int dfs3(int, int, int, int)':
islands.cpp:37:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   37 |     for (auto [j, w] : adj[i]) {
      |               ^
islands.cpp: In function 'int dfs(int, int, int)':
islands.cpp:51:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   51 |     for (auto [j, w] : adj[i]) {
      |               ^
islands.cpp:62:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   62 |         auto [val, w] = dfs2(i, K, cycle);
      |              ^
islands.cpp:68:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   68 |         auto [val, w] = dfs2(i, prt, K);
      |              ^
islands.cpp:63:28: warning: 'K' may be used uninitialized in this function [-Wmaybe-uninitialized]
   63 |         cur = max(cur, dfs3(val, 0, K, cycle));
      |                        ~~~~^~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 13 ms 23764 KB Output is correct
2 Incorrect 13 ms 23764 KB Output isn't correct
3 Correct 13 ms 23892 KB Output is correct
4 Correct 13 ms 23764 KB Output is correct
5 Correct 13 ms 23820 KB Output is correct
6 Correct 13 ms 23764 KB Output is correct
7 Correct 13 ms 23816 KB Output is correct
8 Correct 15 ms 23824 KB Output is correct
9 Correct 13 ms 23820 KB Output is correct
10 Incorrect 13 ms 23780 KB Output isn't correct
11 Correct 12 ms 23764 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 14 ms 23892 KB Output is correct
2 Correct 15 ms 23844 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 14 ms 23892 KB Output is correct
2 Correct 15 ms 24432 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 22 ms 25164 KB Output is correct
2 Correct 32 ms 28660 KB Output is correct
3 Correct 29 ms 25292 KB Output is correct
4 Incorrect 22 ms 24488 KB Output isn't correct
5 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 38 ms 30372 KB Output is correct
2 Correct 46 ms 33480 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 94 ms 40132 KB Output is correct
2 Correct 109 ms 50364 KB Output is correct
3 Correct 119 ms 64844 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 140 ms 52964 KB Output is correct
2 Correct 194 ms 89060 KB Output is correct
3 Correct 215 ms 102836 KB Output is correct
4 Correct 277 ms 131072 KB Output is correct
# Verdict Execution time Memory Grader output
1 Incorrect 359 ms 79472 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 340 ms 131072 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -