oj.uz

Submission #858515

# Submission time Handle Problem Language Result Execution time Memory
858515 2023-10-08T15:23:52 Z Tenis0206 ICC (CEOI16_icc) C++11
90 / 100
 85 ms 688 KB 
#include <bits/stdc++.h>
#ifndef home
#include "icc.h"
#endif // home

using namespace std;

const int nmax = 100;

#ifdef home

pair<int,int> e[nmax + 5];

void output(string msg)
{
    cout<<msg<<'\n';
    exit(0);
}

int cnt = 1;

int query(int sza, int szb, int a[], int b[])
{
    bool ok = false;
    for(int i=0;i<sza;i++)
    {
        for(int j=0;j<szb;j++)
        {
            for(int nr=1;nr<=cnt;nr++)
            {
                if(e[nr].first==a[i] && e[nr].second==b[j])
                {
                    ok = true;
                    break;
                }
                if(e[nr].first==b[j] && e[nr].second==a[i])
                {
                    ok = true;
                    break;
                }
            }
        }
    }
    return ok;
}

void setRoad(int a, int b)
{
    bool ok = false;
    if(a==e[cnt].first && b==e[cnt].second)
    {
        ok = true;
    }
    if(a==e[cnt].second && b==e[cnt].first)
    {
        ok = true;
    }
    if(!ok)
    {
        output("Bad edge");
    }
    ++cnt;
}

#endif // home

int n;

int a[nmax + 5], b[nmax + 5];
int vst[nmax + 5];

int t[nmax + 5];
vector<int> l[nmax + 5];

int rep(int x)
{
    if(t[x]==x)
    {
        return x;
    }
    return rep(t[x]);
}

void uneste(int x, int y)
{
    x = rep(x);
    y = rep(y);
    if(x==y)
    {
        return;
    }
    if(l[x].size() > l[y].size())
    {
        for(auto it : l[y])
        {
            l[x].push_back(it);
        }
        l[y].clear();
        t[y] = x;
    }
    else
    {
        for(auto it : l[x])
        {
            l[y].push_back(it);
        }
        l[x].clear();
        t[x] = y;
    }
}

pair<int,int> find_dif(int nra, int nrb)
{
    int val_a = 0, val_b = 0;
    int st = 1;
    int dr = nra;
    while(st<dr)
    {
        int mij = (st + dr) >> 1;
        int nrv = 0;
        for(int i=st;i<=mij;i++)
        {
            vst[nrv++] = a[i - 1];
        }
        int ok = query(nrv, nrb, vst, b);
        if(ok)
        {
            dr = mij;
        }
        else
        {
            st = mij + 1;
        }
    }
    val_a = a[st - 1];

    st = 1;
    dr = nrb;
    while(st<dr)
    {
        int mij = (st + dr) >> 1;
        int nrv = 0;
        for(int i=st;i<=mij;i++)
        {
            vst[nrv++] = b[i - 1];
        }
        int ok = query(nrv, nra, vst, a);
        if(ok)
        {
            dr = mij;
        }
        else
        {
            st = mij + 1;
        }
    }
    val_b = b[st - 1];

    return {val_a, val_b};
}

void find_new_edge()
{
    vector<int> c;
    for(int i=1;i<=n;i++)
    {
        if(rep(i)==i)
        {
            c.push_back(i);
        }
    }
    pair<int,int> rez = {0,0};
    for(int bit=0;(1<<bit)<c.size();bit++)
    {
        int nra = 0, nrb = 0;
        for(int i=0;i<c.size();i++)
        {
            if((i & (1<<bit)) == 0)
            {
                for(auto it : l[c[i]])
                {
                    a[nra++] = it;
                }
            }
            else
            {
                for(auto it : l[c[i]])
                {
                    b[nrb++] = it;
                }
            }
        }
        int ok = query(nra, nrb, a, b);
        if(ok)
        {
            rez = find_dif(nra, nrb);
            break;
        }
    }
    setRoad(rez.first,rez.second);
    uneste(rez.first, rez.second);
}

void run(int N)
{
    n = N;
    for(int i=1;i<=n;i++)
    {
        t[i] = i;
        l[i].push_back(i);
    }
    for(int i=1;i<n;i++)
    {
        find_new_edge();
    }
}


#ifdef home
int main()
{
    freopen("nr.in","r",stdin);
    freopen("nr.out","w",stdout);
    int nn;
    cin>>nn;
    for(int i=1;i<nn;i++)
    {
        int x,y;
        cin>>x>>y;
        e[i] = {x,y};
    }
    run(nn);
    output("OK");
    return 0;
}
#endif // home

Compilation message

icc.cpp: In function 'void find_new_edge()':
icc.cpp:173:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  173 |     for(int bit=0;(1<<bit)<c.size();bit++)
      |                   ~~~~~~~~^~~~~~~~~
icc.cpp:176:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  176 |         for(int i=0;i<c.size();i++)
      |                     ~^~~~~~~~~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 604 KB Ok! 99 queries used.
2 Correct 4 ms 604 KB Ok! 96 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 604 KB Ok! 527 queries used.
2 Correct 26 ms 600 KB Ok! 661 queries used.
3 Correct 26 ms 604 KB Ok! 666 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 69 ms 616 KB Ok! 1432 queries used.
2 Correct 77 ms 600 KB Ok! 1612 queries used.
3 Correct 84 ms 688 KB Ok! 1622 queries used.
4 Correct 74 ms 604 KB Ok! 1498 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 620 KB Ok! 1580 queries used.
2 Correct 77 ms 604 KB Ok! 1565 queries used.
3 Correct 78 ms 600 KB Ok! 1643 queries used.
4 Correct 73 ms 604 KB Ok! 1494 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 85 ms 600 KB Ok! 1609 queries used.
2 Correct 78 ms 616 KB Ok! 1597 queries used.
3 Correct 77 ms 636 KB Ok! 1643 queries used.
4 Correct 78 ms 604 KB Ok! 1636 queries used.
5 Correct 78 ms 640 KB Ok! 1450 queries used.
6 Correct 74 ms 604 KB Ok! 1528 queries used.

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 79 ms 600 KB Too many queries! 1643 out of 1625
2 Halted 0 ms 0 KB -