oj.uz

답안 #1059337

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1059337 2024-08-14T21:26:50 Z Zicrus Islands (IOI08_islands) C++17
80 / 100
 533 ms 131072 KB 
#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
 
ll n;
vector<vector<pair<ll, ll>>> adj; // length, node
vector<ll> rep;
vector<bool> vst;
vector<ll> comps;
vector<bool> cyc;
vector<pair<ll, ll>> nxt; // length, node
vector<ll> globToCyc;
 
ll cycRoot;
ll dfsCyc(ll cur, ll par) {
    vst[cur] = true;
    vector<pair<ll, ll>> numE;
    for (auto &e : adj[cur]) {
        if (e.second == par) {
            numE.push_back(e);
        }
    }
    if (numE.size() == 2) {
        vst[numE[0].second] = true;
        cyc[cur] = cyc[numE[0].second] = true;
        cycRoot = cur;
        nxt[cur] = numE[0];
        nxt[numE[0].second] = {numE[1].first, cur};
        return -1;
    }
    for (auto &e : adj[cur]) {
        if (e.second == par) continue;
        if (vst[e.second] && !cyc[e.second]) {
            cyc[cur] = true;
            nxt[cur] = e;
            cycRoot = e.second;
            return e.second;
        }
        else {
            ll val = dfsCyc(e.second, cur);
            if (val == -1) continue;
            cyc[cur] = true;
            nxt[cur] = e;
            return cur == val ? -1 : val;
        }
    }
    return -1;
}
 
ll mxLen = 0;
ll dfsTreeSz(ll cur, ll par) {
    ll mx = 0;
    ll nxtMx = 0;
    for (auto &e : adj[cur]) {
        if (e.second == par) continue;
        if (cyc[e.second]) continue;
        ll val = e.first + dfsTreeSz(e.second, cur);
        if (val > mx) {
            nxtMx = mx;
            mx = val;
        }
        else if (val > nxtMx) {
            nxtMx = val;
        }
    }
    mxLen = max(mxLen, nxtMx + mx);
    return mx;
}
 
ll solveComp(ll root) {
    mxLen = 0;
    dfsCyc(root, -1);
    ll totLen = 0;
    ll x = cycRoot;
    vector<ll> cycle;
    do {
        globToCyc[x] = cycle.size();
        cycle.push_back(x);
        totLen += nxt[x].first;
        x = nxt[x].second;
    } while (x != cycRoot);
    vector<ll> treeSz(cycle.size());
    for (int i = 0; i < cycle.size(); i++) {
        treeSz[i] = dfsTreeSz(cycle[i], -1);
    }
    ll res = mxLen;
    ll mxId = cycle[0], mxVal = treeSz[0];
    ll lenSum = 0;
    ll cur = cycle[0];
    for (int i = 0; i < cycle.size(); i++) {
        if (cur == cycle[i] || mxId == i) {
            cur = cycle[i];
            lenSum = nxt[cur].first;
            cur = nxt[cur].second;
            mxId = globToCyc[cur];
            mxVal = lenSum + treeSz[globToCyc[cur]];
        }
        while (cur != cycle[i]) {
            ll val = lenSum + treeSz[globToCyc[cur]];
            if (val >= mxVal) {
                mxVal = val;
                mxId = globToCyc[cur];
            }
            lenSum += nxt[cur].first;
            cur = nxt[cur].second;
        }
        res = max(res, treeSz[i] + mxVal);
        lenSum -= nxt[cycle[i]].first;
        mxVal -= nxt[cycle[i]].first;
    }
    return res;
}
 
void dfs1(ll cur, ll rep1) {
    vst[cur] = true;
    rep[cur] = rep1;
    for (auto &e : adj[cur]) {
        if (vst[e.second]) continue;
        dfs1(e.second, rep1);
    }
}
 
int main() {
    cin >> n;
    adj = vector<vector<pair<ll, ll>>>(n);
    cyc = vector<bool>(n);
    nxt = vector<pair<ll, ll>>(n);
    globToCyc = vector<ll>(n);
    for (int i = 0; i < n; i++) {
        ll a, l;
        cin >> a >> l;
        adj[i].push_back({l, a-1});
        adj[a-1].push_back({l, i});
    }
    rep = vector<ll>(n);
    vst = vector<bool>(n);
    for (int i = 0; i < n; i++) {
        if (vst[i]) continue;
        dfs1(i, i);
        comps.push_back(i);
    }
    vst = vector<bool>(n);
 
    ll res = 0;
    for (auto &e : comps) {
        res += solveComp(e);
    }
    cout << res;
}

Compilation message

islands.cpp: In function 'll solveComp(ll)':
islands.cpp:84:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   84 |     for (int i = 0; i < cycle.size(); i++) {
      |                     ~~^~~~~~~~~~~~~~
islands.cpp:91:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   91 |     for (int i = 0; i < cycle.size(); i++) {
      |                     ~~^~~~~~~~~~~~~~

Subtask #1
33.0 / 33.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 344 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 344 KB Output is correct

Subtask #2
3.0 / 3.0

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

Subtask #3
4.0 / 4.0

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

Subtask #4
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 2392 KB Output is correct
2 Correct 17 ms 6564 KB Output is correct
3 Correct 11 ms 2652 KB Output is correct
4 Correct 5 ms 1628 KB Output is correct

Subtask #5
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 23 ms 8796 KB Output is correct
2 Correct 40 ms 13732 KB Output is correct

Subtask #6
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 87 ms 24460 KB Output is correct
2 Correct 87 ms 35660 KB Output is correct
3 Correct 122 ms 51020 KB Output is correct

Subtask #7
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 156 ms 44876 KB Output is correct
2 Correct 203 ms 85328 KB Output is correct
3 Correct 265 ms 102480 KB Output is correct
4 Correct 330 ms 131072 KB Output is correct

Subtask #8
0 / 10.0

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

Subtask #9
0 / 10.0

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