Submission #406160

#TimeUsernameProblemLanguageResultExecution timeMemory
406160urd05Mechanical Doll (IOI18_doll)C++14
6 / 100
122 ms12340 KiB
#include "doll.h"
#include <bits/stdc++.h>
using namespace std;

vector<int> v;
vector<int> x;
vector<int> y;
vector<int> nt[100001];
int now=-1;

void func(int st,vector<int> vec) {
    if (vec.empty()) {
        return;
    }
    if (vec.size()==1) {
        if (st>=0) {
            v[st]=vec[0];
        }
        else {
            x[-st-1]=st;
            y[-st-1]=vec[0];
        }
        return;
    }
    if (vec.size()==2) {
        if (st>=0) {
            v[st]=now;
            x[-now-1]=vec[0];
            y[-now-1]=vec[1];
            now--;
        }
        else {
            x[-st-1]=vec[0];
            y[-st-1]=vec[1];
        }
        return;
    }
    int nt1,nt2;
    if (st>=0) {
        v[st]=now;
        now--;
        x[-now-2]=now;
        nt1=now;
        now--;
        y[-now-3]=now;
        nt2=now;
        now--;
    }
    else {
        x[st]=now;
        nt1=now;
        now--;
        y[st]=now;
        nt2=now;
        now--;
    }
    vector<int> v1;
    vector<int> v2;
    for(int i=0;i<vec.size();i++) {
        if (i%2==0) {
            v1.push_back(vec[i]);
        }
        else {
            v2.push_back(vec[i]);
        }
    }
    func(nt1,v1);
    func(nt2,v2);
}

void create_circuit(int m,vector<int> a) {
    int n=a.size();
    v.resize(m+1);
    x.resize(200000);
    y.resize(200000);
    nt[0].push_back(a[0]);
    for(int i=1;i<n;i++) {
        nt[a[i-1]].push_back(a[i]);
    }
    nt[a[n-1]].push_back(0);
    for(int i=0;i<=m;i++) {
        if (nt[i].empty()) {
            v[i]=0;
        }
        func(i,nt[i]);
    }
    x.resize(-now-1);
    y.resize(-now-1);
    answer(v,x,y);
}

Compilation message (stderr)

doll.cpp: In function 'void func(int, std::vector<int>)':
doll.cpp:59:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   59 |     for(int i=0;i<vec.size();i++) {
      |                 ~^~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...