답안 #1026066

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1026066 2024-07-17T13:56:00 Z faruk CEOI16_icc (CEOI16_icc) C++17
7 / 100
75 ms 860 KB
// 90 pts
#include "icc.h"
#include <bits/stdc++.h>
#define all(a) a.begin(), a.end()
 
using namespace std;
 
typedef pair<int, int> pii;

vector<vector<int> > components;

bool query(vector<int> a, vector<int> b) {
    int a_niz[a.size()], b_niz[b.size()];
    for (int i = 0; i < a.size(); i++)
        a_niz[i] = a[i];
    for (int i = 0; i < b.size(); i++)
        b_niz[i] = b[i];
    return query(a.size(), b.size(), a_niz, b_niz);
}

void merge_components(int A, int B) {
    int a, b;
    for (int i = 0; i < components.size(); i++)
    {
        if (count(all(components[i]), A))
            a = i;
        if (count(all(components[i]), B))
            b = i;
    }
    if (a > b)
        swap(a, b);
    vector<int> neww_component = components[a];
    neww_component.insert(neww_component.end(), all(components[b]));
    components.erase(components.begin() + b);
    components.erase(components.begin() + a);
    components.push_back(neww_component);
}

vector<int> expand_component_array(vector<int> me) {
    vector<int> neww;
    for (int i : me)
        neww.insert(neww.end(), all(components[i]));
    return neww;
}

bool components_query(vector<int> a, vector<int> b) {
    vector<int> newa = expand_component_array(a), newb = expand_component_array(b);
    return query(newa, newb);
}

pii find_edge_from_true_query(vector<int> a, vector<int> b) {
    pii out;
    int l = 0, r = a.size();
    while (l < r) {
        int mid = (l + r) / 2;
        if (query(vector<int>(a.begin(), a.begin() + mid + 1), b))
            r = mid, out.first = mid;
        else
            l = mid + 1;
    }

    l = 0, r = b.size();
    while (l < r) {
        int mid = (l + r) / 2;
        if (query(a, vector<int>(b.begin(), b.begin() + mid + 1)))
            r = mid, out.second = mid;
        else
            l = mid + 1;
    }
    return pii(a[out.first], b[out.second]);
}

void run(int n) {
    for (int i = 1; i <= n; i++)
        components.push_back(vector<int>(1, i));
    
    for (int i = 0; i < n - 1; i++) {
        int max_bit = __lg(n);
        vector<int> bits(max_bit);
        iota(all(bits), 0);
        random_shuffle(all(bits));

        vector<int> find_a, find_b;
        for (int bit : bits)
        {
            vector<int> cmp1, cmp2;
            for (int i = 0; i < components.size(); i++)
            {
                if (i & (1 << bit))
                    cmp1.push_back(i);
                else
                    cmp2.push_back(i);
            }
            if (components_query(cmp1, cmp2)) {
                find_a = expand_component_array(cmp1), find_b = expand_component_array(cmp2);
                break;
            }
        }
        pii edge = find_edge_from_true_query(find_a, find_b);
        setRoad(edge.first, edge.second);
        merge_components(edge.first, edge.second);
    }
}

Compilation message

icc.cpp: In function 'bool query(std::vector<int>, std::vector<int>)':
icc.cpp:14:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   14 |     for (int i = 0; i < a.size(); i++)
      |                     ~~^~~~~~~~~~
icc.cpp:16:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   16 |     for (int i = 0; i < b.size(); i++)
      |                     ~~^~~~~~~~~~
icc.cpp: In function 'void merge_components(int, int)':
icc.cpp:23:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 |     for (int i = 0; i < components.size(); i++)
      |                     ~~^~~~~~~~~~~~~~~~~~~
icc.cpp: In function 'void run(int)':
icc.cpp:87:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   87 |             for (int i = 0; i < components.size(); i++)
      |                             ~~^~~~~~~~~~~~~~~~~~~
icc.cpp: In function 'void merge_components(int, int)':
icc.cpp:30:5: warning: 'b' may be used uninitialized in this function [-Wmaybe-uninitialized]
   30 |     if (a > b)
      |     ^~
icc.cpp:30:5: warning: 'a' may be used uninitialized in this function [-Wmaybe-uninitialized]
# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 600 KB Ok! 113 queries used.
2 Correct 4 ms 604 KB Ok! 113 queries used.
# 결과 실행 시간 메모리 Grader output
1 Correct 22 ms 632 KB Ok! 544 queries used.
2 Runtime error 1 ms 860 KB Execution killed with signal 11
3 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Correct 75 ms 636 KB Ok! 1457 queries used.
2 Runtime error 1 ms 860 KB Execution killed with signal 11
3 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Runtime error 2 ms 860 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Runtime error 1 ms 860 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -
# 결과 실행 시간 메모리 Grader output
1 Runtime error 1 ms 860 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -