제출 #991459

#제출 시각아이디문제언어결과실행 시간메모리
991459amine_arouaSplit the Attractions (IOI19_split)C++17
11 / 100
52 ms12116 KiB
#include <cstdio>
#include <cassert>
#include <bits/stdc++.h>
#define Int long long
#define fore(i , n) for(int i = 0 ; i<n;i++)
#define pb push_back
#define forn(i , x , y) for(int i = x ; i >= y; i--)
using namespace std;
vector<int> ans;
vector<bool> vis;
vector<vector<int>>adj;
int a , b , c;
bool acc = 0;
int cnt = 1;
void dfs(int x , int p)
{
    if(cnt == b)
    {
        acc = 1;
    }
    if(cnt <= b)
        ans[x] = 2;
    else
    {
        return;
    }
    for(auto u : adj[x])
    {
        if(u == p)
            continue;
        cnt++;
        if(cnt > b)
            return;
        dfs(u , x);
    }
}
bool cmp(const int &i , const int &j)
{
    return (int)adj[i].size() < (int)adj[j].size();
}
struct DSU
{
    vector<int> e;
    DSU(int n)
    {
        e.assign(n , -1);
    }
    int get(int x)
    {
        return (e[x] < 0 ? x : e[x] = get(e[x]));
    }
    bool unite(int u , int v)
    {
        u = get(u);
        v = get(v);
        if(u == v)
            return 0;
        if(e[u] > e[v])
            swap(u , v);
        e[u]+=e[v];
        e[v] = u;
        return 1;
    }
};
vector<int> find_split(int n, int _a , int _b , int _c , vector<int> p , vector<int> q)
{
    a = _a , b=_b , c = _c;
    if(b > c)
        swap(b , c);
    ans.assign(n , 0);
    vis.assign(n , 0);
    adj.assign(n , {});
    DSU dsu(n);
    int m = (int)p.size();
    fore(i , m)
    {
        if(dsu.unite(p[i] , q[i]))
        {
            adj[p[i]].pb(q[i]);
            adj[q[i]].pb(p[i]);
        }
    }
    fore(i , n)
    {
        if((int)adj[i].size() == 1)
        {
            int node = i;
            ans.assign(n , 3);
            ans[node] = 1;
            acc = 0;
            for(auto u : adj[node])
            {
                cnt = 1;
                dfs(u , node);
                if(acc)
                    break;
            }
            assert(acc == 1);
            if(acc == 1)
            {
                if(_b  > _c)
                {
                    for(auto &x : ans) {
                        if (x == 2)
                            x = 3;
                        else if (x == 3)
                            x = 2;
                    }
                }
                return ans;
            }
            break;
        }
    }
    return vector<int>(n , 0);
}
//int main() {
//    int n, m, A, B, C;
//    assert(5 == scanf("%d%d%d%d%d", &n, &m, &A, &B, &C));
//    vector<int> p(m), q(m);
//    for (int i=0; i<m; i++)
//        assert(2 == scanf("%d%d", &p[i], &q[i]));
//    fclose(stdin);
//
//    vector<int> result = find_split(n, A, B, C, p, q);
//
//    for (int i=0; i<(int)result.size(); i++)
//        printf("%s%d", ((i>0)?" ":""), result[i]);
//    printf("\n");
//    fclose(stdout);
//    return 0;
//}
#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...