제출 #448555

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
448555		JooDdae	카멜레온의 사랑 (JOI20_chameleon)	C++17	4 / 100	51 ms	768 KiB

chameleon

#include "chameleon.h"
#include <bits/stdc++.h>
using namespace std;

int nxt[1010];
bool chk[1010];
vector<int> v[1010];

int lb(int u, const vector<int> &x, int l){
 
    int L = l, R = x.size()-1;
    while(L <= R){
        int M = (L+R)/2;
 
        vector<int> Q(1, u);
        for(int i=l;i<=M;i++) Q.push_back(x[i]);
 
        if(Query(Q) == Q.size()) L = M+1;
        else R = M-1;
    }
 
    return L == x.size() ? -1 : R;
}

void addEdge(int a, int b){
    v[a].push_back(b), v[b].push_back(a);
}

void find(int u, const vector<int> &x){
    if(x.size() <= 1) return;

    int bf = 1;
    for(int i=0;i<3;i++){
        int A = lb(u, x, bf);
        if(A < 0) return;
        addEdge(u, x[A+1]), bf = A+2;
    }
}

void solve2(const vector<int> &g, const vector<int> &b){
    vector<int> k = g;
    for(int u : b){
        while(!k.empty() && k[0] < u){
            reverse(k.begin(), k.end()), k.pop_back(), reverse(k.begin(), k.end());
        }
        reverse(k.begin(), k.end()), k.push_back(u), reverse(k.begin(), k.end());
        find(u, k);
        reverse(k.begin(), k.end()), k.pop_back(), reverse(k.begin(), k.end());
    }
}

void Solve(int N) {
    set<int> s;

    for(int i=1;i<=2*N;i++){
        s.insert(i);
        vector<int> q(s.begin(), s.end());

        if(Query(q) != q.size()){
            reverse(q.begin(), q.end());
            find(i, q);

            if(v[i].size() == 3){
                s.erase(i);
            }else{
                for(auto x : v[i]) s.erase(x);
            }
        }
    }

    vector<int> chk2(2*N+1);

    for(int i=1;i<=2*N;i++) if(!chk2[i]){
        vector<int> g, b;
        queue<pair<int, int>> q; chk2[i] = 1, q.push({i, 0});
        while(!q.empty()){
            auto [u, c] = q.front(); q.pop();
            (c ? b : g).push_back(u);
            for(auto x : v[u]) if(!chk2[x]) chk2[x] = 1, q.push({x, !c});
        }

        solve2(g, b);
    }

    // for(int i=1;i<=2*N;i++){
    //     printf("[%d] :",i);
    //     for(int x : v[i]) printf(" %d",x);
    //     puts("");
    // }

    for(int i=1;i<=2*N;i++) if(!chk[i]){
        if(v[i].size() == 1){
            chk[i] = chk[v[i][0]] = 1, Answer(i, v[i][0]);
            continue;
        }

        int &A = v[i][0], &B = v[i][1], &C = v[i][2];
        if(Query({i, B, C}) == 1) swap(A, C);
        else if(Query({i, A, C}) == 1) swap(B, C);
        nxt[i] = C, v[i].pop_back();
    }

    for(int i=1;i<=2*N;i++) if(!chk[i]){
        int x = (chk[v[i][1]] || nxt[v[i][1]] == i ? v[i][0] : v[i][1]);
        chk[i] = chk[x] = 1, Answer(i, x);
    }
}

컴파일 시 표준 에러 (stderr) 메시지

chameleon.cpp: In function 'int lb(int, const std::vector<int>&, int)':
chameleon.cpp:18:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   18 |         if(Query(Q) == Q.size()) L = M+1;
      |            ~~~~~~~~~^~~~~~~~~~~
chameleon.cpp:22:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   22 |     return L == x.size() ? -1 : R;
      |            ~~^~~~~~~~~~~
chameleon.cpp: In function 'void Solve(int)':
chameleon.cpp:59:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |         if(Query(q) != q.size()){
      |            ~~~~~~~~~^~~~~~~~~~~

• Subtask 1: 4 / 4
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 20
• Subtask 5: 0 / 36