oj.uz

답안 #944931

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
944931 2024-03-13T08:20:22 Z thinknoexit Transport (COCI19_transport) C++17
0 / 130
 311 ms 15472 KB 
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 100100;
vector<pair<int, int>> adj[N];
ll ans = 0;
bool vis[N];
int sz[N], m;
ll now[N], fw[N], a[N];
vector<ll> f, c;
int dfs_sz(int v, int p = -1) {
    sz[v] = 1;
    for (auto& [x, w] : adj[v]) {
        if (x == p || vis[x]) continue;
        sz[v] += dfs_sz(x, v);
    }
    return sz[v];
}
int centroid(int v, int p, int s) {
    for (auto& [x, w] : adj[v]) {
        if (x == p || vis[x]) continue;
        if (sz[x] > s) return centroid(x, v, s);
    }
    return v;
}
void init(int v, int p, ll sum, ll d) {
    if (d >= 0) ans++;
    f.push_back(-d);
    sum += a[v];
    for (auto& [x, w] : adj[v]) {
        if (x == p || vis[x]) continue;
        init(x, v, sum - w, min(d, sum - w));
    }
}
void add(int v, int p, ll sum, ll d) {
    int idx = upper_bound(c.begin(), c.end(), -d) - c.begin();
    for (int i = idx;i <= m;i += i & -i) fw[i]++;
    sum += a[v];
    for (auto& [x, w] : adj[v]) {
        if (x == p || vis[x]) continue;
        add(x, v, sum - w, min(d, sum - w));
    }
}
void rem(int v, int p, ll sum, ll d) {
    int idx = upper_bound(c.begin(), c.end(), -d) - c.begin();
    for (int i = idx;i <= m;i += i & -i) fw[i]--;
    sum += a[v];
    for (auto& [x, w] : adj[v]) {
        if (x == p || vis[x]) continue;
        rem(x, v, sum - w, min(d, sum - w));
    }
}
void cal(int v, int p, ll sum, ll mn) {
    if (mn >= 0) {
        int idx = upper_bound(c.begin(), c.end(), sum) - c.begin();
        for (int i = idx;i > 0;i -= i & -i) ans += fw[i];
    }
    for (auto& [x, w] : adj[v]) {
        if (x == p || vis[x]) continue;
        int W = a[x] - w;
        cal(x, v, sum + W, min(mn + W, sum + W));
    }
}
void decom(int v) {
    v = centroid(v, v, dfs_sz(v) / 2);
    vis[v] = 1;
    init(v, v, 0, 0);
    ans--;
    // compress
    sort(f.begin(), f.end());
    for (auto& x : f) {
        if (c.empty() || x != c.back()) c.push_back(x);
    }
    m = c.size();
    add(v, v, 0, 0);
    for (auto& [x, w] : adj[v]) {
        if (vis[x]) continue;
        rem(x, v, a[v] - w, min(0ll, a[v] - w));
        cal(x, v, a[x] - w, min(0ll, a[x] - w));
        add(x, v, a[v] - w, min(0ll, a[v] - w));
    }
    rem(v, v, 0, 0);
    f.clear();
    c.clear();
    for (auto& [x, w] : adj[v]) {
        if (!vis[x]) decom(x);
    }
}
int main() {
    cin.tie(nullptr)->sync_with_stdio(false);
    int n;
    cin >> n;
    for (int i = 1;i <= n;i++) cin >> a[i];
    for (int i = 1;i < n;i++) {
        int u, v, w;
        cin >> u >> v >> w;
        adj[u].push_back({ v, w });
        adj[v].push_back({ u, w });
    }
    decom(1);
    cout << ans;
    return 0;
}

Subtask #1
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 5720 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 6 ms 5980 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 66 ms 10972 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 100 ms 12756 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 130 ms 15472 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 71 ms 7772 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 65 ms 8920 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 176 ms 9420 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #9
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 242 ms 10696 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #10
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 311 ms 12244 KB Output isn't correct
2 Halted 0 ms 0 KB -