結果

問題 No.19 ステージの選択
ユーザー はまやんはまやん
提出日時 2017-06-13 14:32:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 4,482 bytes
コンパイル時間 3,003 ms
コンパイル使用メモリ 202,236 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-23 10:06:48
合計ジャッジ時間 3,256 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<b;i++)
#pragma GCC optimize ("-O3")
using namespace std; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); }
//---------------------------------------------------------------------------------------------------
struct SCC {
    vector<vector<int> > SC; int NV; vector<int> GR, rev, NUM, vis; vector<vector<int>> E, RE;
    void init(int NV) { GR.resize(NV); rev.resize(NV); vis.resize(NV); E.resize(NV, vector<int>());
        RE.resize(NV, vector<int>());this->NV = NV;}
    void add_edge(int x, int y) { E[x].push_back(y); RE[y].push_back(x); }
    void dfs(int cu) { vis[cu] = 1;
        for (int i = 0; i<E[cu].size(); i++) if (!vis[E[cu][i]]) dfs(E[cu][i]); NUM.push_back(cu); }
    void revdfs(int cu, int ind) { int i; vis[cu] = 1; GR[cu] = ind; SC[ind].push_back(cu);
        rev[cu] = ind; rep(i, 0, RE[cu].size()) if (!vis[RE[cu][i]]) revdfs(RE[cu][i], ind); }
    void scc() { int c = 0; SC.clear(); SC.resize(NV); NUM.clear();
        rep(i, 0, NV) vis[i] = 0; for (int i = 0; i < NV; i++) if (!vis[i]) dfs(i);
        rep(i, 0, NV) vis[i] = 0; for (int i = NUM.size() - 1; i >= 0; i--) if (!vis[NUM[i]]) {
        SC[c].clear(); revdfs(NUM[i], c); sort(SC[c].begin(), SC[c].end()); c++;} SC.resize(c); }};
struct TopologicalSort {
    vector<set<int>> E; TopologicalSort(int N) { E.resize(N); }
    void add_edge(int a, int b) { E[a].insert(b); }
    bool visit(int v, vector<int>& order, vector<int>& color) { color[v] = 1;
        for (int u : E[v]) { if (color[u] == 2) continue; if (color[u] == 1) return false;
        if (!visit(u, order, color)) return false; } order.push_back(v); color[v] = 2; return true; }
    bool sort(vector<int> &order) { int n = E.size(); vector<int> color(n);
        for (int u = 0; u < n; u++) if (!color[u] && !visit(u, order, color)) return false;
        reverse(order.begin(), order.end()); return true; } };
/*---------------------------------------------------------------------------------------------------
　　　　　　　　　　　 ∧＿∧  
　　　　　 ∧＿∧ 　（´<_｀ ）　 Welcome to My Coding Space!
　　　　 （ ´_ゝ`）　/　 ⌒i     
　　　　／　　　＼　 　  |　|     
　　　 /　　 /￣￣￣￣/　　|  
　 ＿_(__ﾆつ/　    ＿/ .| .|＿＿＿＿  
　 　　　＼/＿＿＿＿/　（u　⊃  
---------------------------------------------------------------------------------------------------*/

struct SCCtoDAG {
    int N = -1; vector<vector<int>> E;
    int M; vector<int> mapping;
    vector<set<int>> newE;
    vector<int> tsorder;
    void init(int n) { N = n; E.resize(N, vector<int>()); }
    void add_edge(int a, int b) {
        assert(0 < N && "Not Init!!");
        E[a].push_back(b);
    }
    void build() {
        SCC scc;
        scc.init(N);
        rep(i, 0, N) for(int j : E[i]) scc.add_edge(i, j);
        scc.scc();

        vector<int> dec;
        rep(i, 0, N) dec.push_back(scc.GR[i]);
        sort(dec.begin(), dec.end());
        dec.erase(unique(dec.begin(), dec.end()), dec.end());
        M = dec.size();
        mapping.clear();
        rep(i, 0, N) {
            int id = lower_bound(dec.begin(), dec.end(), scc.GR[i]) - dec.begin();
            mapping.push_back(id);
        }
        newE.resize(M, set<int>());
        rep(i, 0, N) for (int j : E[i]) {
            int ii = mapping[i], jj = mapping[j];
            if (ii == jj) continue;
            newE[ii].insert(jj);
        }

        TopologicalSort ts(M);
        rep(i, 0, M) for(int j : newE[i]) ts.add_edge(i, j);
        tsorder.clear();
        ts.sort(tsorder);
    }

};



int N, L[101], S[101];
vector<int> P[101];
int vis[101], done[101];
//---------------------------------------------------------------------------------------------------
void _main() {
    cin >> N;
    rep(i, 0, N) cin >> L[i] >> S[i];

    SCCtoDAG machine;
    machine.init(N);
    rep(i, 0, N) if (S[i] - 1 != i) machine.add_edge(S[i] - 1, i);
    machine.build();

    rep(i, 0, N) P[machine.mapping[i]].push_back(L[i]);
    rep(i, 0, N) sort(P[i].begin(), P[i].end());

    int ans = 0;
    for (auto i : machine.tsorder) {
        for (auto j : P[i]) {
            ans += j;
        }
        if (!vis[i] && P[i].size()) ans += P[i][0];
        for (int j : machine.newE[i]) vis[j] = 1;
    }

    printf("%d", ans / 2);
    if (ans % 2) printf(".5");
    else printf(".0");
    printf("\n");
}
0