Submission #1305568

#TimeUsernameProblemLanguageResultExecution timeMemory
1305568HasanV11010238Swapping Cities (APIO20_swap)C++20
37 / 100
2097 ms31612 KiB
#include "swap.h"
#include <bits/stdc++.h>
#define INF 1000000003
#define MAX 200000
#define ll long long
using namespace std;
int n;
vector<vector<int>> v;
vector<ll> ans, go, de, pa;
int cur;
struct DSU{
    vector<int> p, le, ri, ro;
    int cmp;
    void init(int n){
        p.assign(n + 1, -1);
        ro.assign(n + 1, 0), le.assign(n + 1, 0), ri.assign(n + 1, 0);
        for (int i = 1; i <= n; i++) ro[i] = i, le[i] = i, ri[i] = i;
        cmp = n;
    }
    int find(int x){
        if (p[x] < 0) return x;
        return p[x] = find(p[x]);
    }
    void merge(int a, int b, ll co){
        int ora = a, orb = b;
        a = find(a), b = find(b);
        if (a == b){
            if (go[ro[a]] == 0) go[ro[a]] = 1, ans[ro[a]] = co;
            return;
        }
        cmp--;
        if (p[a] > p[b]) swap(a, b), swap(ora, orb);
        if (ora == le[a]) swap(le[a], ri[a]);
        if (orb == ri[b]) swap(le[b], ri[b]);
        go[cur] = max(go[ro[a]], go[ro[b]]);
        if (ri[a] != ora || le[b] != orb) go[cur] = 1;
        ri[a] = ri[b];
        ans[cur] = co;
        p[a] += p[b];
        p[b] = a;
        v[cur].push_back(ro[a]);
        v[cur].push_back(ro[b]);
        ro[a] = cur;
        cur++;
    }
};
void dfs(int x, int p){
    //cout<<x<<"\n";
    de[x] = de[p] + 1;
    pa[x] = p;
    if (go[x] == 0) ans[x] = -1;
    for (auto el : v[x]){
        if (go[x] == 1 && go[el] == 0) ans[el] = ans[x], go[el] = 1;
        dfs(el, x);
    }
}
DSU ds;
void init(int N, int M, vector<int> U, vector<int> V, vector<int> W) {
    n = N;
    v.assign(2 * n + 1, vector<int>()), ans.assign(2 * n + 1, INF), go.assign(2 * n + 1, 0);
    de.assign(2 * n + 1, 0), pa.assign(2 * n + 1, 0), cur = n + 1;
    vector<vector<int>> so;
    for (int i = 0; i < M; i++){
        so.push_back({W[i], U[i] + 1, V[i] + 1});
    }
    ds.init(n);
    sort(so.begin(), so.end());
    for (auto el : so){
        ds.merge(el[1], el[2], el[0]);
    }
    for (int i = 1; i <= 2 * n - 1; i++){
        /*cout<<go[i]<<" "<<i<<": ";
        for (auto el : v[i]) cout<<el<<" ";
        cout<<"\n";*/
    }
    dfs(cur - 1, 0);
}
int getMinimumFuelCapacity(int x, int y){
    x++, y++;
    if (de[x] > de[y]) swap(x, y);
    while (de[y] > de[x]) y = pa[y];
    while (x != y) x = pa[x], y = pa[y];
    //cout<<x<<"\n";
    return ans[x];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...