oj.uz

Submission #491500

# Submission time Handle Problem Language Result Execution time Memory
491500 2021-12-02T18:12:28 Z blue Harbingers (CEOI09_harbingers) C++17
90 / 100
 185 ms 26016 KB 
#include <iostream>
#include <vector>
#include <cassert>
using namespace std;

using vi = vector<int>;
using ll = long long;
using vll = vector<ll>;

#define pb push_back

const int maxN = 100'000;
const int lgN = 17;

ll INF = 1'000'000'000'000'000'000LL;

vector<pair<int, ll>> edge[1+maxN];

vll S(1+maxN), V(1+maxN);
vi P(1+maxN);
vll D(1+maxN, 0);

vi ord;

void dfs(int u, int p)
{
    ord.push_back(u);
    P[u] = p;
    for(auto x: edge[u])
    {
        int v = x.first;
        ll d = x.second;
        if(v == p) continue;
        D[v] = D[u] + d;
        dfs(v, u);
    }
}





struct line
{
    ll a;
    ll b;

    ll eval(ll x)
    {
        return a*x + b;
    }
};

bool intersect_comp(line A, line B, line C, line D)
{
    //A.a * x + A.b == B.a * x + B.b, x = (B.b - A.b)/(A.a - B.a)

    ll mul = 1;
    if(A.a - B.a < 0) mul *= -1;
    if(C.a - D.a < 0) mul *= -1;
    return mul * (B.b - A.b) * (C.a - D.a) <= mul * (D.b - C.b) * (A.a - B.a);
}

vector<line> cht(1+maxN);
int cht_anc[1+maxN][1+lgN];

void insert_line(int u, line l)
{
    // cerr << "\n\n\n\n insertion: u = " << u << ", " << "line = " << l.a << ' ' << l.b << '\n';
    cht[u] = l;
    int a = P[u];
    // cerr << "init a = " << a << '\n';
    for(int e = lgN; e >= -1; e--)
    {
        int v = (e == -1 ? a : cht_anc[a][e]);
        if(v == 1) continue;
        int w = cht_anc[v][0];

        // cerr << "checking " << w << ' ' << v << '\n';

        if(!intersect_comp(cht[w], cht[v], cht[v], l))
            a = w;
    }

    cht_anc[u][0] = a;
    for(int e = 1; e <= lgN; e++)
        cht_anc[u][e] = cht_anc[ cht_anc[u][e-1] ][e-1];
}

ll query(int u, ll x)
{
    // cerr << "\n\n\n\n querying " << x << " at " << u << '\n';
    for(int e = lgN; e >= -1; e--)
    {
        int v = (e == -1 ? u : cht_anc[u][e]);
        if(v == 1) continue;
        int w = cht_anc[v][0];

        // cerr << "query " << u << ' ' << x << " : " << "checking " << w << ", " << v << '\n';

        ll mul = 1;
        if(cht[w].a - cht[v].a < 0) mul = -1;

        if(x * (cht[w].a - cht[v].a) * mul <= (cht[v].b - cht[w].b) * mul)
            u = w;
    }

    return min(cht[u].eval(x), cht[cht_anc[u][0]].eval(x));
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int N;
    cin >> N;

    for(int i = 1; i <= N-1; i++)
    {
        int u, v;
        ll d;
        cin >> u >> v >> d;

        edge[u].pb({v, d});
        edge[v].pb({u, d});
    }

    for(int i = 2; i <= N; i++) cin >> S[i] >> V[i];


    dfs(1, 1);


    for(int i = 1; i <= N; i++)
        for(int e = 0; e <= lgN; e++)
            cht_anc[i][e] = i;

    insert_line(1, {0, 0});

    vll dp(1+N, INF);
    dp[1] = 0;
    for(int u: ord)
    {
        if(u == 1) continue;

        dp[u] = ll(S[u]) + ll(V[u])*ll(D[u]) + query(P[u], V[u]);
        insert_line(u, {-D[u], dp[u]});
    }

    // cerr << cht_anc[1][0] << ' ' << cht_anc[2][0] << ' ' << cht_anc[3][0] << '\n';

    for(int i = 2; i <= N; i++) cout << dp[i] << ' ';
    cout << '\n';
}

Subtask #1
90.0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 6988 KB Output is correct
2 Correct 5 ms 7500 KB Output is correct
3 Incorrect 44 ms 14900 KB Output isn't correct
4 Correct 82 ms 18760 KB Output is correct
5 Correct 91 ms 22420 KB Output is correct
6 Correct 114 ms 26016 KB Output is correct
7 Correct 69 ms 18132 KB Output is correct
8 Correct 169 ms 23644 KB Output is correct
9 Correct 185 ms 24392 KB Output is correct
10 Correct 145 ms 23612 KB Output is correct