답안 #1086347

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1086347 2024-09-10T09:49:05 Z TrinhKhanhDung 공장들 (JOI14_factories) C++14
컴파일 오류
0 ms 0 KB
#include <bits/stdc++.h>
#define ll long long
#define fi first
#define se second
#define sz(x) (int)x.size()
#define ALL(v) v.begin(),v.end()
#define MASK(k) (1LL << (k))
#define BIT(x, i) (((x) >> (i)) & 1)
#define oo (ll)1e18
#define INF (ll)1e9
#define MOD (ll)(998244353)

using namespace std;

template<class T1, class T2>
    bool maximize(T1 &a, T2 b){if(a < b){a = b; return true;} return false;}

template<class T1, class T2>
    bool minimize(T1 &a, T2 b){if(a > b){a = b; return true;} return false;}

template<class T1, class T2>
    void add(T1 &a, T2 b){a += b; if(a >= MOD) a -= MOD;}

template<class T1, class T2>
    void sub(T1 &a, T2 b){a -= b; if(a < 0) a += MOD;}

template<class T>
    void cps(T &v){sort(ALL(v)); v.resize(unique(ALL(v)) - v.begin());}

struct seg{
    vector<ll> mi;
    int n;

    seg(int _n = 0){
        n = _n;
        mi.resize(n * 4 + 3, oo);
    }

    void upd(int p, ll c){
        int i = 1, l = 0, r = n;
        while(l < r){
            int m = (l + r) >> 1;
            if(p <= m){
                i = i * 2;
                r = m;
            }
            else{
                i = i * 2 + 1;
                l = m + 1;
            }
        }
        mi[i] = c;
        while(i > 1){
            i >>= 1;
            mi[i] = min(mi[i * 2], mi[i * 2 + 1]);
        }
    }

    ll get(int i, int l, int r, int u, int v){
        if(r < u || v < l) return oo;
        if(u <= l && r <= v) return mi[i];
        int m = (l + r) >> 1;
        return min(get(i * 2, l, m, u, v), get(i * 2 + 1, m + 1, r, u, v));
    }

    ll get(int u, int v){
        return get(1, 0, n, u, v);
    }
};

const int MAX = 5e5 + 10, LOG = 19;

int N, Q;
vector<pair<int, int>> adj[MAX];
int h[MAX], d[MAX][LOG], in[MAX], out[MAX], timer;
ll dist[MAX];

void dfs(int u, int p){
    in[u] = ++timer;
    for(auto o: adj[u]){
        int v = o.fi;
        int c = o.se;
        if(v == p) continue;
        h[v] = h[u] + 1;
        d[v][0] = u;
        for(int i=1; i<LOG; i++){
            d[v][i] = d[d[v][i - 1]][i - 1];
        }
        dist[v] = dist[u] + c;
        dfs(v, u);
    }
    out[u] = timer;
//    cout << u << ' ' << in[u] << ' ' << out[u] << ' ' << h[u] << ' ' << dist[u] << '\n';
}

int lca(int u, int v){
    if(h[u] < h[v]) swap(u, v);
    int delta = h[u] - h[v];
    for(int i=0; i<LOG; i++) if(BIT(delta, i)){
        u = d[u][i];
    }
    if(u == v) return u;
    for(int i=LOG-1; i>=0; i--) if(d[u][i] != d[v][i]){
        u = d[u][i];
        v = d[v][i];
    }
    return d[u][0];
}

void solve(){
    cin >> N >> Q;
    for(int i=1; i<N; i++){
        int u, v, c;
        cin >> u >> v >> c;
        adj[u].push_back(make_pair(v, c));
        adj[v].push_back(make_pair(u, c));
    }
    dfs(0, -1);
    seg it1(N), it2(N);
    while(Q--){
        int n, m; cin >> n >> m;
        vector<int> a;
        for(int i=0; i<n; i++){
            int x; cin >> x;
            a.push_back(x);
            it1.upd(in[x], dist[x]);
        }
        for(int i=0; i<m; i++){
            int x; cin >> x;
            a.push_back(x);
            it2.upd(in[x], dist[x]);
        }
        sort(ALL(a), [&](int u, int v){return in[u] < in[v];});
        for(int i=1; i<n+m; i++){
            a.push_back(lca(a[i], a[i - 1]));
        }
        cps(a);
        sort(ALL(a), [&](int u, int v){return in[u] < in[v];});
        ll ans = oo;
        for(int x: a){
//            cout << x << ' ' <<
            minimize(ans, it1.get(in[x], out[x]) + it2.get(in[x], out[x]) - 2 * dist[x]);
        }
        for(int x: a){
            it1.upd(in[x], oo);
            it2.upd(in[x], oo);
        }
        cout << ans << '\n';
    }
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);
//    freopen("numtable.inp","r",stdin);
//    freopen("numtable.out","w",stdout);

    solve();

    return 0;
}

Compilation message

/usr/bin/ld: /tmp/cco6VLt4.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccBJQoR5.o:factories.cpp:(.text.startup+0x0): first defined here
/usr/bin/ld: /tmp/cco6VLt4.o: in function `main':
grader.cpp:(.text.startup+0x37d): undefined reference to `Init(int, int*, int*, int*)'
/usr/bin/ld: grader.cpp:(.text.startup+0x412): undefined reference to `Query(int, int*, int, int*)'
collect2: error: ld returned 1 exit status