oj.uz

답안 #650684

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
650684 2022-10-14T16:31:15 Z Spade1 Islands (IOI08_islands) C++14
64 / 100
 329 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(i, 0, K, cycle));
        v.pb({w, W});
        return 0;
    }
    else if (ret) {
        auto [val, w] = dfs2(i, prt, K);
        cur = max(cur, dfs3(i, 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(i, 0, K, cycle));
      |                        ~~~~^~~~~~~~~~~~~~~~

Subtask #1
27.0 / 33.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 23764 KB Output is correct
2 Incorrect 12 ms 23740 KB Output isn't correct
3 Correct 14 ms 23928 KB Output is correct
4 Correct 13 ms 23768 KB Output is correct
5 Correct 13 ms 23772 KB Output is correct
6 Correct 13 ms 23764 KB Output is correct
7 Correct 14 ms 23764 KB Output is correct
8 Correct 15 ms 23764 KB Output is correct
9 Correct 15 ms 23764 KB Output is correct
10 Incorrect 14 ms 23732 KB Output isn't correct
11 Correct 13 ms 23764 KB Output is correct

Subtask #2
3.0 / 3.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 23984 KB Output is correct
2 Correct 14 ms 23836 KB Output is correct

Subtask #3
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 14 ms 23892 KB Output is correct
2 Correct 15 ms 24464 KB Output is correct

Subtask #4
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 21 ms 25456 KB Output is correct
2 Correct 30 ms 29136 KB Output is correct
3 Correct 24 ms 25500 KB Output is correct
4 Incorrect 22 ms 24680 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #5
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 36 ms 30800 KB Output is correct
2 Correct 48 ms 34768 KB Output is correct

Subtask #6
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 87 ms 40484 KB Output is correct
2 Correct 102 ms 52540 KB Output is correct
3 Correct 117 ms 68572 KB Output is correct

Subtask #7
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 139 ms 53264 KB Output is correct
2 Correct 195 ms 93944 KB Output is correct
3 Correct 209 ms 107840 KB Output is correct
4 Correct 262 ms 131072 KB Output is correct

Subtask #8
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 308 ms 79904 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 329 ms 131072 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -