oj.uz

Submission #843567

# Submission time Handle Problem Language Result Execution time Memory
843567 2023-09-04T06:20:59 Z Tuanlinh123 Factories (JOI14_factories) C++17
100 / 100
 2099 ms 292928 KB 
#include "factories.h"
#include <bits/stdc++.h>
#define taskname ""
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define ll long long
#define ld long double
#define pb push_back
#define ff first
#define ss second
#define pii pair<int, int>
#define vi vector<int>
#define vll vector<ll>
#define vvi vector<vi>
#define vii vector<pii>
#define isz(x) (int)x.size()
using namespace std;
 
const int mxN = 1e6 + 5;
const ll oo = 1e18;
 
int n;
vector<vii> G;
 
vi vtn;
vector<vector<pair<ll, ll>>> vt;
 
ll dis[mxN];
pii sparse[22][mxN];
int timer, ecnt, ap[mxN], dep[mxN], cur[mxN], tin[mxN], tout[mxN];
 
void dfs(int v, int p) {
    tin[v] = ++timer;
    dep[v] = dep[p] + 1;
    sparse[0][++ecnt] = {dep[v], v};
    ap[v] = ecnt;
 
    for(auto &[u, w] : G[v]) {
        if(u == p) continue;
        dis[u] = dis[v] + w;
        dfs(u, v);
        sparse[0][++ecnt] = {dep[v], v};
    }
 
    tout[v] = timer;
}
 
void build_sparse() {
    for(int p = 1, i = 1; p << 1 <= ecnt; p <<= 1, ++i)
        for(int j = 1; j <= ecnt - (p << 1) + 1; ++j)
            sparse[i][j] = min(sparse[i - 1][j], sparse[i - 1][j + p]);
}
 
bool is_ancestor(int u, int v) {
    return tin[u] <= tin[v] && tout[u] >= tout[v];
}
 
int LCA(int u, int v) {
    int l = ap[u], r = ap[v];
    if(l > r) swap(l, r);
    int len = r - l + 1, lg_len = __lg(len);
    return min(sparse[lg_len][l], sparse[lg_len][r - (1 << lg_len) + 1]).ss;
}
 
void Init(int N, int A[], int B[], int D[]) {
    n = N;
    G.resize(n); vt.resize(n);
    for(int i = 0; i < N - 1; ++i) {
        G[A[i]].emplace_back(B[i], D[i]);
        G[B[i]].emplace_back(A[i], D[i]);
    }
    dfs(0, 0); build_sparse();
}

long long Query(int S, int X[], int T, int Y[]) {
    queue<pii> q;
 
    for(int i = 0; i < S; ++i) {
        cur[X[i]] = 1;
        q.emplace(X[i], 0);
        vtn.emplace_back(X[i]);
    }
    for(int i = 0; i < T; ++i) {
        cur[Y[i]] = 2;
        vtn.emplace_back(Y[i]);
    }
 
    sort(all(vtn), [&](int u, int v) {
        return tin[u] < tin[v];
    });
 
    for(int i = 0; i < S + T - 1; ++i) {
        vtn.emplace_back(LCA(vtn[i], vtn[i + 1]));
    }
 
    sort(all(vtn), [&](int u, int v) {
        return tin[u] < tin[v];
    });
    vtn.erase(unique(all(vtn)), vtn.end());
 
    auto add_vt = [&](int u, int v, ll w) {
        vt[u].emplace_back(v, w);
        vt[v].emplace_back(u, w);
    };
 
    stack<int> s;
    for(int i = 0; i < isz(vtn); ++i) {
        while(not s.empty() && not is_ancestor(s.top(), vtn[i])) s.pop();
        if(s.empty()) s.emplace(vtn[i]);
        else {
            add_vt(s.top(), vtn[i], dis[vtn[i]] - dis[s.top()]);
            s.emplace(vtn[i]);
        }
    }
 
    ll res = oo;
 
    auto dfs = [&](auto self, int v, int p) -> vll {
        vll cur_dis(2, oo);
        if(cur[v]) cur_dis[cur[v] - 1] = 0;
        for(auto &[u, w] : vt[v]) {
            if(u == p) continue;
            vll tmp = self(self, u, v);
            for(int i = 0; i < 2; ++i)
                cur_dis[i] = min(cur_dis[i], tmp[i] + w);
        }
        res = min(res, accumulate(all(cur_dis), 0LL));
        return cur_dis;
    };
 
    dfs(dfs, vtn[0], 0);
 
    for(int i = 0; i < S; ++i)
        cur[X[i]] = 0;
    for(int i = 0; i < T; ++i)
        cur[Y[i]] = 0;
    for (int i:vtn)
        vt[i].clear();
    vtn.clear();
 
    return res;
}

Subtask #1
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 16 ms 43612 KB Output is correct
2 Correct 583 ms 58776 KB Output is correct
3 Correct 573 ms 58784 KB Output is correct
4 Correct 574 ms 58876 KB Output is correct
5 Correct 543 ms 58960 KB Output is correct
6 Correct 371 ms 58728 KB Output is correct
7 Correct 566 ms 58828 KB Output is correct
8 Correct 584 ms 58836 KB Output is correct
9 Correct 537 ms 59408 KB Output is correct
10 Correct 360 ms 58704 KB Output is correct
11 Correct 585 ms 58796 KB Output is correct

Subtask #2
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 43608 KB Output is correct
2 Correct 903 ms 256152 KB Output is correct
3 Correct 904 ms 260892 KB Output is correct
4 Correct 685 ms 255004 KB Output is correct
5 Correct 888 ms 292928 KB Output is correct
6 Correct 962 ms 261224 KB Output is correct
7 Correct 732 ms 110788 KB Output is correct
8 Correct 513 ms 109824 KB Output is correct
9 Correct 558 ms 115028 KB Output is correct
10 Correct 684 ms 111132 KB Output is correct

Subtask #3
67.0 / 67.0

# Verdict Execution time Memory Grader output
1 Correct 16 ms 43612 KB Output is correct
2 Correct 583 ms 58776 KB Output is correct
3 Correct 573 ms 58784 KB Output is correct
4 Correct 574 ms 58876 KB Output is correct
5 Correct 543 ms 58960 KB Output is correct
6 Correct 371 ms 58728 KB Output is correct
7 Correct 566 ms 58828 KB Output is correct
8 Correct 584 ms 58836 KB Output is correct
9 Correct 537 ms 59408 KB Output is correct
10 Correct 360 ms 58704 KB Output is correct
11 Correct 585 ms 58796 KB Output is correct
12 Correct 7 ms 43608 KB Output is correct
13 Correct 903 ms 256152 KB Output is correct
14 Correct 904 ms 260892 KB Output is correct
15 Correct 685 ms 255004 KB Output is correct
16 Correct 888 ms 292928 KB Output is correct
17 Correct 962 ms 261224 KB Output is correct
18 Correct 732 ms 110788 KB Output is correct
19 Correct 513 ms 109824 KB Output is correct
20 Correct 558 ms 115028 KB Output is correct
21 Correct 684 ms 111132 KB Output is correct
22 Correct 1972 ms 264488 KB Output is correct
23 Correct 1638 ms 263848 KB Output is correct
24 Correct 2099 ms 268724 KB Output is correct
25 Correct 1962 ms 270136 KB Output is correct
26 Correct 1575 ms 263448 KB Output is correct
27 Correct 1788 ms 290516 KB Output is correct
28 Correct 1269 ms 260540 KB Output is correct
29 Correct 1547 ms 262276 KB Output is correct
30 Correct 1518 ms 261364 KB Output is correct
31 Correct 1552 ms 261968 KB Output is correct
32 Correct 887 ms 119372 KB Output is correct
33 Correct 578 ms 112300 KB Output is correct
34 Correct 891 ms 110536 KB Output is correct
35 Correct 882 ms 110752 KB Output is correct
36 Correct 936 ms 111576 KB Output is correct
37 Correct 948 ms 111072 KB Output is correct