結果

問題 No.19 ステージの選択
ユーザー drymouse
提出日時 2024-02-17 16:20:15
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,449 bytes
コンパイル時間 1,128 ms
コンパイル使用メモリ 75,512 KB
最終ジャッジ日時 2025-02-19 15:31:55
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
using namespace std;

int main(void) {
    int N;
    cin >> N;
    int L[N], S[N];
    for (int i = 0; i < N; i++) {
        int level, stage;
        cin >> level >> stage;
        L[i] = level << 1;
        S[i] = stage - 1;
    }
    vector<int> children[N];
    int tansaku[N] = {0};
    for (int i = 0; i < N; i++) {
        children[S[i]].push_back(i);
    }
    int graph_id = 1;
    int sum = 0;
    for (int i = 0; i < N; i++) {
        if (tansaku[i] == 0) {
            int takinobori[N] = {0};
            vector<int> passed;
            takinobori[i] = 1;
            passed.push_back(i);
            int cur = i, next, kon;
            for (;1;) {
                next = S[cur];
                if (next == cur) {
                    kon = cur;
                    break;
                } else if (takinobori[next] == 1) {
                    int min = L[next];
                    int minver = next;
                    int found = 0;
                    int lsize = passed.size();
                    for (int j = 0; j < lsize; j++) {
                        if (found) {
                            if (L[passed[j]] < min) {
                                minver = passed[j];
                                min = L[minver];
                            }
                        }else if (passed[j] == next) {
                            found = 1;
                        }
                    }
                    kon = minver;
                    break;
                } else {
                    takinobori[next] = 1;
                    passed.push_back(next);
                    cur = next;
                }
            }
            //cout << "graph id: " << graph_id << ", kon: " << kon << endl;
            vector<int> que = {kon};
            tansaku[kon] = graph_id;
            sum += L[kon];
            for (;que.size();) {
                int cur = que[0];
                que.erase(que.begin());
                int csize = children[cur].size();
                for (int j = 0; j < csize; j++) {
                    int child = children[cur][j];
                    if (tansaku[child] == 0) {
                        tansaku[child] = graph_id;
                        que.push_back(child);
                        sum += L[child] / 2;
                    }
                }
            }
            graph_id++;
        }
    }

    cout << sum << endl;
    return 0;
}
0