제출 #157159

#제출 시각아이디문제언어결과실행 시간메모리
157159fedoseevtimofeyConstruction of Highway (JOI18_construction)C++14
100 / 100
683 ms111204 KiB
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;

const int N = 1e5 + 7;
vector <int> g[N];
int sz[N], st[N], en[N], par[N], in[N], out[N], h[N];

int A[N], B[N], C[N];
deque <pair <int, int>> col[N];

void dfs(int u = 0, int p = -1) {
    par[u] = p;
    for (int i = 0; i < g[u].size(); ++i) {
        if (g[u][i] == p) {
            g[u].erase(g[u].begin() + i);
            break;
        }
    }
    sz[u] = 1;
    for (int v : g[u]) {
        dfs(v, u);
        sz[u] += sz[v];
    }
    for (int i = 1; i < (int)g[u].size(); ++i) {
        if (sz[g[u][i]] > sz[g[u][0]]) swap(g[u][i], g[u][0]);
    }
}

vector <int> e;

int dfs_hld(int u = 0, int head = 0) {
    e.push_back(u);
    in[u] = (int)e.size() - 1;
    st[u] = head;
    if (par[u] != -1) h[u] = h[par[u]] + 1;
    else h[u] = 0;
    if (!g[u].empty()) {
        en[u] = dfs_hld(g[u][0], head);
        for (int i = 1; i < (int)g[u].size(); ++i) {
            dfs_hld(g[u][i], g[u][i]);
        }
    } else {
        en[u] = u;
    }
    out[u] = (int)e.size() - 1;
    return en[u];
}

struct Fenwick {
    vector <int> f;
    int n;
    void modify(int i, int val) {
        i = n - i - 1;
        for (; i < n; i |= i + 1) f[i] += val;
    }
    int get(int r) {
        int res = 0;
        r = n - r - 1;
        --r;
        for (; r >= 0; r &= r + 1, --r) res += f[r];
        return res;
    }
    Fenwick(int nn) {
        n = nn;
        f.resize(n);
    }
};

bool anc(int a, int b) {
    return in[a] <= in[b] && out[b] <= out[a];
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0); cout.setf(ios::fixed); cout.precision(20);
    #ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    #endif
    int n;
    cin >> n;
    for (int i = 0; i < n; ++i) cin >> C[i];
    for (int i = 0; i + 1 < n; ++i) {
        cin >> A[i] >> B[i];
        --A[i], --B[i];
        g[A[i]].push_back(B[i]);
        g[B[i]].push_back(A[i]);
    }
    dfs();
    dfs_hld();
    col[0].push_back({C[0], 0});
    for (int id = 0; id + 1 < n; ++id) {
        int ans = 0;
        int u = B[id];
        vector <pair <int, int>> hv;
        while (u != -1) {
            int pth = st[u];
            vector <pair <int, int>> cur;
            for (int i = 0; i < (int)col[pth].size(); ++i) {
                if (anc(col[pth][i].second, u)) {
                    cur.push_back({col[pth][i].first, h[col[pth][i].second] - (i == 0 ? h[pth] - 1: h[col[pth][i - 1].second])});
                } else {
                    cur.push_back({col[pth][i].first, h[u] - (i == 0 ? h[pth] - 1: h[col[pth][i - 1].second])});
                    break;
                }
            }
            reverse(cur.begin(), cur.end());
            for (auto p : cur) hv.push_back(p);
            while (!col[pth].empty() && anc(col[pth][0].second, B[id])) col[pth].pop_front();
            col[pth].push_front({C[B[id]], u});
            u = par[st[u]];
        }   
        reverse(hv.begin(), hv.end());
        vector <int> comp;
        for (auto p : hv) comp.push_back(p.first);
        sort(comp.begin(), comp.end());
        comp.resize(unique(comp.begin(), comp.end()) - comp.begin());
        for (auto &p : hv) p.first = lower_bound(comp.begin(), comp.end(), p.first) - comp.begin();
        Fenwick f(comp.size());
        for (auto p : hv) {
            ans += p.second * f.get(p.first);
            f.modify(p.first, p.second);
        }
        cout << ans << '\n';
    }
}   

컴파일 시 표준 에러 (stderr) 메시지

construction.cpp: In function 'void dfs(int, int)':
construction.cpp:17:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < g[u].size(); ++i) {
                     ~~^~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...