제출 #983674

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
983674		mariaclara	자동 인형 (IOI18_doll)	C++17	100 / 100	97 ms	12180 KiB

doll

#include "doll.h"
#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
const int MAXN = 4e5+5; 
#define all(x) x.begin(), x.end()
#define sz(x) (int)x.size()
#define mk make_pair
#define pb push_back
#define fr first
#define sc second

int N;
vector<int> ans, S_x, S_y, state;
int create(int qnt) {
    if(qnt == 1) return 1;
    int a = create(qnt/2), b = create(qnt/2);
    S_x.pb(a);
    S_y.pb(b);
    return -sz(S_y);
}

void create_circuit(int M, vector<int> A) {
    A.pb(0);
    N = sz(A);
    
    vector<int> pot;
    for(int i = 20; i >= 0; i--)
        if(N >= (1<<i)) pot.pb(i), N -= (1<<i);

    N = sz(A);

    reverse(all(pot));

    for(int i = 0, last = 0; i < sz(pot); i++) {
        int k = create((1<<pot[i]));
        if(i > 0) {
            S_x.pb(k);
            S_y.pb(last);
            k = -sz(S_x);
        }
        if(i == sz(pot)-1) break;
        int add = pot[i+1] - pot[i] - !!i;
        while(add--) {
            S_y.pb(k);
            S_x.pb(-sz(S_y));
            k = -sz(S_x);
        }
        last = -sz(S_x);
    }

    int S = sz(S_x);
    vector<int> ans(M+1, -S), state(S,0);

    for(int i = 0; i < S; i++)
        if(S_x[i] == -i-1) S_x[i] = -S;

    // for(int i = 0; i < S; i++)
    //     cout << S_x[i] << " " << S_y[i] << endl;

    int x = -S, qtd = 0;
    while(qtd < N) {
        if(state[-x-1] == 0) {
            state[-x-1] = 1;
            if(S_x[-x-1] == 1) S_x[-x-1] = A[qtd], qtd++, x = -S;
            else x = S_x[-x-1]; 
        } 
        else {
            state[-x-1] = 0;
            if(S_y[-x-1] == 1) S_y[-x-1] = A[qtd], qtd++, x = -S;
            else x = S_y[-x-1]; 
        }
    }

    return answer(ans, S_x, S_y);
}

• Subtask 1: 2 / 2
• Subtask 2: 4 / 4
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 18 / 18
• Subtask 6: 56 / 56