Submission #1453846


Source Code Expand

#include <cstdio>
#include <vector>
#include <algorithm>
#include <array>
#include <set>
#include <map>
#include <queue>
#include <tuple>
#include <unordered_set>
#include <unordered_map>
#include <functional>
#include <cassert>
#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 ll = long long;
using namespace std;
template <class T> inline void setmin(T & a, T const & b) { a = min(a, b); }

template <typename UnaryPredicate>
ll binsearch(ll l, ll r, UnaryPredicate p) { // [l, r), p is monotone
    assert (l < r);
    -- l;
    while (r - l > 1) {
        ll m = (l + r) / 2;
        (p(m) ? r : l) = m;
    }
    return r; // = min { x in [l, r) | p(x) }, or r
}

bool pred(vector<vector<pair<int, int> > > const & g, ll limit) {
    function<vector<pair<ll, ll> > (int, int)> dfs = [&](int i, int parent) {
        if (g[i].size() == 1) {
            vector<pair<ll, ll> > result;
            result.emplace_back(0, 0);
            return result;
        }
        vector<pair<ll, ll> > left, right;
        for (auto edge : g[i]) {
            int j, value; tie(j, value) = edge;
            if (j != parent) {
                auto x = dfs(j, i);
                auto & y = left.empty() ? left : right;
                assert (y.empty());
                y = move(x);
                for (int i = 0; i < y.size(); ++ i) {
                    y[i].first  += value;
                    y[i].second += value;
                    if (y[i].first > limit or y[i].second > limit) {
                        swap(y[i], y[y.size() - 1]);
                        y.pop_back();
                        -- i;
                    }
                }
            }
        }
        vector<pair<ll, ll> > result;
        if (left.size() < right.size()) {
            left.swap(right);
        }
        for (auto l : left) {
            ll l1, l2; tie(l1, l2) = l;
            for (auto r : right) {
                ll r1, r2; tie(r1, r2) = r;
                auto push = [&](ll a, ll b, ll c) {
                    if (c <= limit) {
                        if (a < b) swap(a, b);
                        result.emplace_back(a, b);
                    }
                };
                push(l1, r1, l2 + r2);
                push(l1, r2, l2 + r1);
                push(l2, r1, l1 + r2);
                push(l2, r2, l1 + r1);
            }
        }
        whole(sort, result);
        result.erase(whole(unique, result, [&](auto x, auto y) { return x.second <= y.second; }), result.end());
        return result;
    };
    return not dfs(0, -1).empty();
}

constexpr ll inf = ll(1e18)+9;
int main() {
    int n; scanf("%d", &n);
    vector<vector<pair<int, int> > > g(n);
    repeat (i, n - 1) {
        int a, v; scanf("%d%d", &a, &v);
        g[a - 1].emplace_back(i + 1, v);
        g[i + 1].emplace_back(a - 1, v);
    }
    ll result = binsearch(0, inf, [&](ll limit) {
        return pred(g, limit);
    });
    printf("%lld\n", result);
    return 0;
}

Submission Info

Submission Time
Task E - Shik and Travel
User kimiyuki
Language C++14 (Clang 3.8.0)
Score 0
Code Size 3183 Byte
Status WA
Exec Time 5257 ms
Memory 475060 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 1400
Status
AC × 3
WA × 1
AC × 43
WA × 7
TLE × 4
Set Name Test Cases
Sample example0.txt, example1.txt, example2.txt, example3.txt
All 000.txt, 001.txt, 002.txt, 003.txt, 004.txt, 005.txt, 006.txt, 007.txt, 008.txt, 009.txt, 010.txt, 011.txt, 012.txt, 013.txt, 014.txt, 015.txt, 016.txt, 017.txt, 018.txt, 019.txt, 020.txt, 021.txt, 022.txt, 023.txt, 024.txt, 025.txt, 026.txt, 027.txt, 028.txt, 029.txt, 030.txt, 031.txt, 032.txt, 033.txt, 034.txt, 035.txt, 036.txt, 037.txt, 038.txt, 039.txt, 040.txt, 041.txt, 042.txt, 043.txt, 044.txt, 045.txt, 046.txt, 047.txt, 048.txt, 049.txt, example0.txt, example1.txt, example2.txt, example3.txt
Case Name Status Exec Time Memory
000.txt AC 17 ms 384 KB
001.txt AC 16 ms 384 KB
002.txt AC 17 ms 384 KB
003.txt AC 17 ms 384 KB
004.txt AC 16 ms 384 KB
005.txt AC 18 ms 384 KB
006.txt AC 18 ms 384 KB
007.txt AC 16 ms 384 KB
008.txt WA 17 ms 384 KB
009.txt AC 16 ms 384 KB
010.txt AC 1053 ms 8448 KB
011.txt AC 1033 ms 8448 KB
012.txt AC 1039 ms 8448 KB
013.txt AC 1066 ms 8448 KB
014.txt AC 1048 ms 8448 KB
015.txt AC 1080 ms 8448 KB
016.txt AC 1057 ms 8448 KB
017.txt AC 1064 ms 8448 KB
018.txt AC 1072 ms 8448 KB
019.txt AC 1060 ms 8448 KB
020.txt AC 1048 ms 8448 KB
021.txt AC 1041 ms 8448 KB
022.txt AC 1052 ms 8448 KB
023.txt AC 1088 ms 8448 KB
024.txt AC 1078 ms 8448 KB
025.txt AC 1076 ms 8448 KB
026.txt AC 1033 ms 8448 KB
027.txt AC 1068 ms 8448 KB
028.txt AC 1067 ms 8448 KB
029.txt AC 1014 ms 10496 KB
030.txt AC 952 ms 8448 KB
031.txt AC 980 ms 8448 KB
032.txt AC 1004 ms 8448 KB
033.txt AC 968 ms 8448 KB
034.txt AC 973 ms 8448 KB
035.txt AC 1047 ms 8448 KB
036.txt AC 1013 ms 8448 KB
037.txt AC 1047 ms 8448 KB
038.txt AC 963 ms 8448 KB
039.txt AC 964 ms 8448 KB
040.txt TLE 5257 ms 475060 KB
041.txt TLE 5257 ms 471300 KB
042.txt TLE 5257 ms 470404 KB
043.txt TLE 5257 ms 471264 KB
044.txt WA 918 ms 33024 KB
045.txt AC 903 ms 20736 KB
046.txt WA 912 ms 24320 KB
047.txt WA 890 ms 21760 KB
048.txt WA 913 ms 31104 KB
049.txt WA 1 ms 256 KB
example0.txt AC 1 ms 256 KB
example1.txt WA 1 ms 256 KB
example2.txt AC 1 ms 256 KB
example3.txt AC 1 ms 256 KB