oj.uz

Submission #96683

# Submission time Handle Problem Language Result Execution time Memory
96683 2019-02-10T19:45:10 Z keko37 Transport (COCI19_transport) C++14
130 / 130
 349 ms 18028 KB 
#include<bits/stdc++.h>

using namespace std;

typedef long long llint;

const int MAXN = 100005;

llint n, root, cc, sol;
int val[MAXN];
llint par[MAXN], siz[MAXN], dp1[MAXN], dp2[MAXN], curr[MAXN], moze[MAXN];
bool cen[MAXN];
vector < pair <int, int> > v[MAXN];
vector <llint> q, t;

void dfs (int x, int rod) {
    par[x] = rod;
    siz[x] = 1;
    for (auto it : v[x]) {
        int sus = it.first;
        if (sus == rod || cen[sus]) continue;
        dfs(sus, x);
        siz[x] += siz[sus];
    }
}

int nadi_centroid (int x) {
    int mx = 0, ind = 0;
    if (par[x] != 0) {
        mx = siz[root] - siz[x];
        ind = par[x];
    }
    for (auto it : v[x]) {
        int sus = it.first;
        if (sus == par[x] || cen[sus]) continue;
        if (siz[sus] > mx) {
            mx = siz[sus];
            ind = sus;
        }
    }
    if (mx * 2 > siz[root]) return nadi_centroid(ind);
    return x;
}

void calc_dp (int x, int rod) {
    if (rod == 0) {
        dp1[x] = 0;
        dp2[x] = 0;
        curr[x] = val[x];
    }
    if (rod != 0 && dp1[x] == 0) sol++;
    q.push_back(dp1[x]);
    for (auto it : v[x]) {
        int sus = it.first;
        int dist = it.second;
        if (sus == rod || cen[sus]) continue;
        dp1[sus] = max(dp1[x], -(curr[x] - dist));
        curr[sus] = curr[x] - dist + val[sus];
        dp2[sus] = max(0LL, dp2[x]-(val[sus] - dist));
        calc_dp(sus, x);
    }
}

void napravi (int x, int rod) {
    t.push_back(dp1[x]);
    for (auto it : v[x]) {
        int sus = it.first;
        if (sus == rod || cen[sus]) continue;
        napravi(sus, x);
    }
}

void upit (int x, int rod) {
    if (dp2[x] == 0) {
        sol += upper_bound(q.begin(), q.end(), curr[x] - val[cc]) - q.begin() - 1;
        sol -= upper_bound(t.begin(), t.end(), curr[x] - val[cc]) - t.begin() - 1;
    }
    for (auto it : v[x]) {
        int sus = it.first;
        if (sus == rod || cen[sus]) continue;
        upit(sus, x);
    }
}

void decompose (int x) {
    root = x;
    dfs(root, 0);
    cc = nadi_centroid(root);
    cen[cc] = 1;
    q.clear();
    calc_dp(cc, 0);
    sort(q.begin(), q.end());
    for (auto it : v[cc]) {
        int sus = it.first;
        if (cen[sus]) continue;
        t.clear();
        napravi(sus, cc);
        sort(t.begin(), t.end());
        upit(sus, cc);
    }
    int cvor = cc;
    for (auto it : v[cvor]) {
        int sus = it.first;
        if (cen[sus]) continue;
        decompose(sus);
    }
}

int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin >> n;
    for (int i=1; i<=n; i++) {
        cin >> val[i];
    }
    for (int i=0; i<n-1; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        v[a].push_back(make_pair(b, c));
        v[b].push_back(make_pair(a, c));
    }
    decompose(1);
    cout << sol;
    return 0;
}

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 3064 KB Output is correct
2 Correct 8 ms 3192 KB Output is correct

Subtask #2
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 3320 KB Output is correct
2 Correct 10 ms 3576 KB Output is correct

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 78 ms 9440 KB Output is correct
2 Correct 68 ms 8820 KB Output is correct

Subtask #4
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 101 ms 12020 KB Output is correct
2 Correct 136 ms 13228 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 142 ms 15344 KB Output is correct
2 Correct 174 ms 18028 KB Output is correct

Subtask #6
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 74 ms 5364 KB Output is correct
2 Correct 35 ms 5112 KB Output is correct

Subtask #7
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 65 ms 7284 KB Output is correct
2 Correct 112 ms 6848 KB Output is correct

Subtask #8
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 172 ms 8560 KB Output is correct
2 Correct 150 ms 8936 KB Output is correct

Subtask #9
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 275 ms 10348 KB Output is correct
2 Correct 251 ms 10776 KB Output is correct

Subtask #10
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 349 ms 12780 KB Output is correct
2 Correct 278 ms 12012 KB Output is correct