Submission #810645

#	Time	Username	Problem	Language	Result	Execution time	Memory
810645		gagik_2007	Mechanical Doll (IOI18_doll)	C++17	53 / 100	92 ms	23600 KiB

doll

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

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

#define ff first
#define ss second

ll ttt;
const ll INF=1e18;
const ll MOD=1e9+7;
const ll N=300007;
ll n,m,k;
ll lg2[N];
int cnt[N];
int val[N];
vector<int>vs[N];
int cc[N];

string dec_to_bin(int num, int len){
    string s="";
    for(int _=0;_<len;_++){
        s+=num%2+'0';
        num/=2;
    }
    return s;
}

void create_circuit(int M, std::vector<int> a) {
    n = a.size();
    m=M;
    lg2[1]=0;
    for(int i=2;i<N;i++){
        lg2[i]=lg2[i/2]+1;
    }
    vector<int>uniq;
    for(int x:a){
        if(cnt[x]==0)uniq.push_back(x);
        cnt[x]++;
    }
    int cur=-1;
    for(int x:uniq){
        val[x]=lg2[cnt[x]-1];
        vs[x].push_back(0);
        if(cnt[x]>1){
            for(int _=0;_<(1<<(val[x]+1))-1;_++){
                vs[x].push_back(cur);
                cur--;
            }
        }
    }
    vector<int>C(M+1),X(-cur-1,-1),Y(-cur-1,-1);
    C[0]=a[0];
    for(int x:uniq){
        if(cnt[x]>1)C[x]=vs[x][1];
        for(int i=1;i<vs[x].size()/2;i++){
            X[-vs[x][i]-1]=vs[x][2*i];
            Y[-vs[x][i]-1]=vs[x][2*i+1];
        }
    }
    a.push_back(0);
    for(int i=0;i<a.size()-1;i++){
        int x=a[i];
        if(cnt[x]==1){
            C[x]=a[i+1];
        }
        else{
            if(cc[x]!=cnt[x]-1){
                string s=dec_to_bin(cc[x],val[x]+1);
                int ind=1;
                for(int j=0;j+1<s.size();j++){
                    ind*=2;
                    if(s[j]=='1')ind+=1;
                }
                if(s.back()=='0'){
                    X[-vs[x][ind]-1]=a[i+1];
                }
                else{
                    Y[-vs[x][ind]-1]=a[i+1];
                }
            }
            else{
                for(int i=1;i<vs[x].size();i++){
                    if(X[-vs[x][i]-1]==-1){
                        X[-vs[x][i]-1]=vs[x][1];
                    }
                    if(Y[-vs[x][i]-1]==-1){
                        Y[-vs[x][i]-1]=vs[x][1];
                    }
                }
                Y[-vs[x].back()-1]=a[i+1];
            }
            cc[x]++;
        }
    }
    answer(C,X,Y);
}

Compilation message (stderr)

doll.cpp: In function 'void create_circuit(int, std::vector<int>)':
doll.cpp:60:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |         for(int i=1;i<vs[x].size()/2;i++){
      |                     ~^~~~~~~~~~~~~~~
doll.cpp:66:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |     for(int i=0;i<a.size()-1;i++){
      |                 ~^~~~~~~~~~~
doll.cpp:75:32: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   75 |                 for(int j=0;j+1<s.size();j++){
      |                             ~~~^~~~~~~~~
doll.cpp:87:30: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   87 |                 for(int i=1;i<vs[x].size();i++){
      |                             ~^~~~~~~~~~~~~

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