제출 #594908

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
594908		TimDee	Mechanical Doll (IOI18_doll)	C++14	20 / 100	109 ms	30120 KiB

doll

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

struct tree {
    int n, size;
    vector<int> t, x, y, state;
    void build(int N, int amt) {
        n=N;
        size=1;
        while ((size<<1)<n) size<<=1;
        t.assign(2*size-1,0), x.assign(2*size-1,0), y.assign(2*size-1,0), state.assign(2*size-1,0);
        for (int i=1; i<=size-1; ++i) {
            x[i-1] = -(2*i);
            y[i-1] = -(2*i+1);
        }
        int cnt=0, p=0;
        while (cnt<2*size-amt) {
            if (p>=size-1) {
                if (state[p]) {
                    y[p]=-1;
                } else {
                    x[p]=-1;
                }
                state[p]^=1;
                ++cnt;
                p=0;
                continue;
            }
            if (!state[p]) {
                state[p]^=1;
                p<<=1; ++p;
            } else {
                state[p]^=1;
                ++p; p<<=1;
            }
        }
    }
};

void create_circuit(int m, vector<int>a) {
    int n=a.size();
    vector<int> cnt(m+1,0);
    int mx=0;
    for (int i=0; i<n; ++i) {
        cnt[a[i]]++;
        mx=max(mx,cnt[a[i]]);
    }
    if (mx==1) {
        vector<int> c(m+1,0);
        c[0]=a[0];
        for (int i=1; i<n; ++i) {
            c[a[i-1]]=a[i];
        }
        vector<int> x(0), y(0);
        answer(c,x,y);
    }/* else if (mx==2) {

    } else if (mx<=4) {

    } */else if (m==1) {

        int size=1;
        while (4*size<n) size<<=1;
        vector<int> x(2*size+1), y(2*size+1), state(2*size+1,0), c = {1, -(2*size)};
        for (int i=1; i<=size-1; ++i) {
            x[i-1] = -(2*i);
            y[i-1] = -(2*i+1);
        }
        x[2*size-1]=-(2*size+1);
        y[2*size-1]=-1;
        x[2*size] = y[2*size] = 1;

        int cnt=0, p=0;
        while (cnt<2*size) {

            //cout<<p<<"->";
            if (p>=size-1) {
                //cout<<"added "<<p<<' '<<cnt<<'\n';
                if (state[p]) {
                    if (cnt<n-1-2*size) y[p]=1;
                    else y[p]=-(2*size);
                } else {
                    if (cnt<n-1-2*size) x[p]=1;
                    else x[p]=-(2*size);
                }
                state[p]^=1;
                ++cnt;
                p=0;
                continue;
            }

            if (!state[p]) {

                state[p]^=1;
                p<<=1; ++p;

            } else {

                state[p]^=1;
                ++p; p<<=1;

            }

        }
        y[2*size-2]=0;
        answer(c,x,y);

    } else {

        vector<int> c(m+1),x,y;
        c[0]=a[0];
        vector<tree> v(m+1);
        for (int i=1; i<=m; ++i) v[i].build(cnt[i],cnt[i]);

        int in=0, p=0;
        while (in<n) {
            int ind = a[in];
            while (p<v[ind].size-1) {
                if (!v[ind].state[p]) {
                    v[ind].state[p]^=1;
                    p<<=1;
                    ++p;
                } else {
                    v[ind].state[p]^=1;
                    ++p;
                    p<<=1;
                }
            }
            if (!v[ind].state[p]) {
                v[ind].x[p]= ++in<n ? a[in] : 0;
                v[ind].state[p]^=1;
            } else {
                v[ind].y[p]= ++in<n ? a[in] : 0;
                v[ind].state[p]^=1;
            }
        }

        for(int i=1; i<=m; ++i) {
            int sz=x.size();
            for (auto X:v[i].x) x.push_back( X>=0 ? X : (X-sz) );
            for (auto Y:v[i].y) y.push_back( Y>=0 ? Y : (Y-sz) );
            c[i]=-1-sz;
        }
        answer(c,x,y);

    }
}

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