Submission #1246098

#TimeUsernameProblemLanguageResultExecution timeMemory
1246098qwushaSplit the Attractions (IOI19_split)C++20
40 / 100
319 ms37024 KiB

#include "split.h"

#include <iostream>
#include <bits/stdc++.h>

#define fi first
#define se second
typedef long long ll;
using namespace std;

int inf = 1e9 + 7;

vector<vector<pair<int, int>>> g;


vector<int> used;
vector<int> active;
int cnt;
int B;

void dfs(int v) {
    if (cnt == B)
        return;
    used[v] = 1;
    cnt++;
    for (auto [u, w] : g[v]) {
        if (!active[w])
            continue;
        if (!used[u]) {
            dfs(u);
        }
    }
}

vector<int> sz;
vector<int> par;

void dfs_sz(int v, int p = -1) {
    used[v] = 1;
    par[v] = p;
    sz[v] = 1;
    for (auto [u, w]: g[v]) {
        if (!used[u]) {
            dfs_sz(u, v);
            sz[v] += sz[u];
        }
    }
}

int centr = 0;
void find_cen(int v, int n) {
    used[v] = 1;
    bool ok = 1;
    for (auto [u, w] : g[v]) {
        if (!used[u] && sz[u] > n / 2) {
            ok = 0;
            find_cen(u, n);
        }
    }
    if (ok)
        centr = v;
}


vector<int> find_split(int n, int a, int b, int c, vector<int> p, vector<int> q) {
    g.assign(n, {});
    int m = p.size();
    active.assign(m, 1);
    map<pair<int, int>, int> edg;
    for (int i = 0; i < m; i++) {
        g[p[i]].push_back({q[i], i});
        g[q[i]].push_back({p[i], i});
        edg[{p[i], q[i]}] = i;
        edg[{q[i], p[i]}] = i;
    }

    bool two = 1;
    int mini = inf, indmi = -1;
    for (int i = 0; i < n; i++) {
        if (g[i].size() > 2)
            two = 0;
        if (g[i].size() < mini) {
            mini = g[i].size();
            indmi = i;
        }
    }
    if (two) {
        vector<int> res(n, 3);
        res[indmi] = 1;
        int last = indmi;
        int cur = g[indmi][0].fi;
        for (int i = 0; i < a - 1; i++) {
            res[cur] = 1;
            int tr = cur;
            cur = g[cur][0].fi + g[cur][1].fi - last;
            last = tr;
        }
        for (int i = 0; i < b; i++) {
            res[cur] = 2;
            int tr = cur;
            cur = g[cur][0].fi + g[cur][1].fi - last;
            last = tr;
        }
        return res;
    } else if (a == 1) {
        used.assign(n, 0);
        cnt = 0;
        B= b;
        dfs(0);
        vector<int> res(n, 3);
        for (int i = 0; i < n; i ++) {
            if (used[i])
                res[i] = 2;
        }
        for (int i = 0; i < n; i++) {
            if (res[i] == 3) {
                res[i] = 1;
                break;
            }
        }
        return res;
    } else {
        used.assign(n, 0);
        sz.assign(n, 0);
        par.assign(n, 0);
        dfs_sz(0);
        used.assign(n, 0);
        centr = 0;
        find_cen(0, n);
        int cent = centr;
        int sa = 1, sb = 2, sc = 3;
        if (a > b) {
            swap(a, b);
            swap(sa, sb);
        }
        if (a > c) {
            swap(a, c);
            swap(sa, sc);
        }
        if (b > c) {
            swap(b, c);
            swap(sb, sc);
        }
        int maxi = n - sz[cent];
        int neig = par[cent];
        for (auto [u,w] : g[cent]) {
            if (u != par[cent]) {
                if (maxi < sz[u]) {
                    maxi = sz[u];
                    neig = u;
                }
            }
        }
        vector<int> res(n, 0);
        if (maxi < a) {
            return res;
        }
        for (int i = 0; i < n; i++)
            res[i] = sc;
        active[edg[{cent, neig}]] = 0;
        B = a;
        cnt = 0;
        used.assign(n, 0);
        dfs(neig);
        for (int i = 0; i < n; i++) {
            if (used[i])
                res[i] = sa;
        }
        B = b;
        cnt = 0;
        used.assign(n, 0);
        dfs(cent);
        for (int i = 0; i < n; i++) {
            if (used[i])
                res[i] = sb;
        }
        return res;
    }

}


/*
signed main() {
    int n, a, b, c;
    cin >> n >> a >> b >> c;
    int m;
    cin >> m;
    vector<int> p(m), q(m);
    for (int i = 0; i < m; i++) {
        cin >> p[i] >> q[i];
    }
    auto res = find_split(n,a,b,c,p,q);
    for (auto el : res) {
        cout << el << ' ';
    }
    cout << endl;

}
*/
#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...