結果

問題 No.19 ステージの選択
ユーザー kimiyukikimiyuki
提出日時 2017-01-06 14:28:06
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,334 bytes
コンパイル時間 743 ms
コンパイル使用メモリ 84,888 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-02 12:31:49
合計ジャッジ時間 1,477 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 1 ms
5,376 KB
testcase_10 AC 1 ms
5,376 KB
testcase_11 AC 1 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 1 ms
5,376 KB
testcase_16 AC 1 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 1 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>
#include <functional>
#define repeat(i,n) for (int i = 0; (i) < int(n); ++(i))
#define whole(f,x,...) ([&](decltype((x)) whole) { return (f)(begin(whole), end(whole), ## __VA_ARGS__); })(x)
using namespace std;
int main() {
    // input
    int n; cin >> n;
    vector<int> l(n);
    vector<vector<int> > g(n);
    vector<int> g_inv(n);
    repeat (i,n) {
        int s; cin >> l[i] >> s; -- s;
        g[s].push_back(i);
        g_inv[i] = s;
    }
    // solve
    double ans = whole(accumulate, l, 0) / 2.0;
    vector<bool> used(n);
    function<void (int)> use = [&](int i) {
        if (used[i]) return;
        used[i] = true;
        for (int j : g[i]) if (not used[j]) use(j);
    };
    function<void (int, vector<int> &)> roots = [&](int i, vector<int> & acc) {
        auto it = whole(find, acc, i);
        if (it != acc.end()) {
            acc.erase(acc.begin(), it);
        } else {
            acc.push_back(i);
            roots(g_inv[i], acc);
        }
    };
    repeat (i,n) if (not used[i]) {
        vector<int> acc;
        roots(i, acc);
        int j = *whole(min_element, acc, [&](int i, int j) { return l[i] < l[j]; });
        use(j);
        ans += l[j] / 2.0;
    }
    // output
    printf("%.1f\n", ans);
    return 0;
}
0