oj.uz

답안 #629737

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
629737 2022-08-15T02:20:52 Z dooompy Islands (IOI08_islands) C++17
70 / 100
 953 ms 131076 KB 
#include <bits/stdc++.h>

using namespace std;

using ll = long long;

struct edge {
    int to; ll cost; int idx;
};

vector<edge> adj[1000005];
bool seen[1000005], seenedge[1000005], inloop[1000005];
int par[1000005];

int cyclestart, cycleend;

void dfs(int node) {
    seen[node] = true;
    for (auto nxt : adj[node]) {
        if (seenedge[nxt.idx]) continue;
        seenedge[nxt.idx] = true;
        if (seen[nxt.to]) {
            // cycle found
            cyclestart = node, cycleend = nxt.to;
        } else {
            par[nxt.to] = node;
            dfs(nxt.to);
        }
    }
}

ll globmax = 0;

ll dfs_tree(int node, int par = -1) {
    ll ans = 0;

    for (auto nxt : adj[node]) {
        if (nxt.to == par) continue;
        if (inloop[nxt.to]) continue;

        ll temp = dfs_tree(nxt.to, node);

        globmax = max(globmax, ans + temp + nxt.cost);
        ans = max(ans, temp + nxt.cost);
    }

    return ans;
}

int main() {
    int n; cin >> n;
    for (int i = 1; i <= n; i++) {
        int to; ll c; cin >> to >> c;
        adj[i].push_back({to, c, i});
        adj[to].push_back({i, c, i});
    }

    ll total = 0;

    for (int i = 1; i <= n; i++) {
        if (seen[i]) continue;
        dfs(i);

        vector<int> loop;

        while (cyclestart != cycleend) {
            loop.push_back(cyclestart);
            cyclestart = par[cyclestart];
        }

        loop.push_back(cyclestart);

        for (auto cur : loop) {
            inloop[cur] = true;
        }

        ll ans = 0;

        ll prevInMax = 0;
        ll prevOutMax = 0;

        map<int, ll> cursum;
        ll totalsum = 0;
        map<int, bool> seensecond;
        for (int i = 0; i <= loop.size(); i++) {
            if (i == 0) continue;
            if (i == loop.size()) {
                for (auto nxt : adj[loop[0]]) {
                    if (nxt.to == loop.back() && !seensecond[nxt.idx]) {
                        totalsum += nxt.cost;
                        break;
                    }
                }
                break;
            }
            for (auto nxt : adj[loop[i]]) {
                if (nxt.to == loop[i-1] && !seensecond[nxt.idx]) {
                    seensecond[nxt.idx] = true;
                    cursum[loop[i]] += nxt.cost;
                    totalsum += nxt.cost;
                    break;
                }
            }
            cursum[loop[i]] += cursum[loop[i-1]];
        }

        for (int i = 0; i < loop.size(); i++) {
            globmax = 0;
            ll tree = dfs_tree(loop[i]);

            if (i > 0) {
                ans = max({ans, tree + cursum[loop[i]] + prevInMax, totalsum + prevOutMax - cursum[loop[i]] + tree});
            }
            ans = max(ans, globmax);

            prevInMax = max(prevInMax, tree - cursum[loop[i]]);
            prevOutMax = max(prevOutMax, tree + cursum[loop[i]]);



        }

        total += ans;

    }

    cout << total << endl;

}

Compilation message

islands.cpp: In function 'int main()':
islands.cpp:85:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   85 |         for (int i = 0; i <= loop.size(); i++) {
      |                         ~~^~~~~~~~~~~~~~
islands.cpp:87:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   87 |             if (i == loop.size()) {
      |                 ~~^~~~~~~~~~~~~~
islands.cpp:107:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  107 |         for (int i = 0; i < loop.size(); i++) {
      |                         ~~^~~~~~~~~~~~~

Subtask #1
33.0 / 33.0

# 결과 실행 시간 메모리 Grader output
1 Correct 12 ms 23824 KB Output is correct
2 Correct 13 ms 23816 KB Output is correct
3 Correct 13 ms 23912 KB Output is correct
4 Correct 12 ms 23752 KB Output is correct
5 Correct 12 ms 23764 KB Output is correct
6 Correct 12 ms 23752 KB Output is correct
7 Correct 13 ms 23788 KB Output is correct
8 Correct 13 ms 23764 KB Output is correct
9 Correct 14 ms 23708 KB Output is correct
10 Correct 12 ms 23764 KB Output is correct
11 Correct 15 ms 23796 KB Output is correct

Subtask #2
3.0 / 3.0

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

Subtask #3
4.0 / 4.0

# 결과 실행 시간 메모리 Grader output
1 Correct 13 ms 23932 KB Output is correct
2 Correct 17 ms 24416 KB Output is correct

Subtask #4
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 28 ms 25472 KB Output is correct
2 Correct 59 ms 29432 KB Output is correct
3 Correct 30 ms 25624 KB Output is correct
4 Correct 20 ms 24660 KB Output is correct

Subtask #5
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 70 ms 31588 KB Output is correct
2 Correct 111 ms 35512 KB Output is correct

Subtask #6
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 177 ms 44020 KB Output is correct
2 Correct 298 ms 55404 KB Output is correct
3 Correct 371 ms 72788 KB Output is correct

Subtask #7
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 387 ms 60232 KB Output is correct
2 Correct 593 ms 101444 KB Output is correct
3 Correct 953 ms 116744 KB Output is correct
4 Runtime error 695 ms 131076 KB Execution killed with signal 9
5 Halted 0 ms 0 KB -

Subtask #8
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 712 ms 99996 KB Output is correct
2 Runtime error 945 ms 131072 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

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