Submission #553341

#	Time	Username	Problem	Language	Result	Execution time	Memory
553341		fcmalkcin	Unscrambling a Messy Bug (IOI16_messy)	C++17	100 / 100	4 ms	480 KiB

messy

#include "messy.h"

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

#define ll int
#define pll pair<ll,ll>
#define ff first
#define ss second
//#define endl "\n"
#define pb push_back
#define F(i,a,b) for(ll i=a;i<=b;i++)

const ll maxn=3e5+100;
const ll base=3e18;
const ll mod= 1e9+7 ;

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

ll n;
void dosth(ll l,ll r)
{
    if (l==r) return ;
    ll mid=(l+r)/2;
    for (int t=l;t<=mid;t++)
    {
        string p;
        for (int i=1;i<l;i++) p.pb('1');
        for (int i=l;i<=r;i++)
        {
            if (t==i) p.pb('1');
            else p.pb('0');
        }
        for (int i=r+1;i<=n;i++) p.pb('1');
        add_element(p);
    }
    dosth(l,mid);
    dosth(mid+1,r);
}
ll res[maxn];
void dosth1(ll l,ll r,vector<ll> vt)
{
   if (l==r)
   {
       res[vt[0]]=l;
       return ;
   }
   vector<bool> dd(n+3,0);
   for (auto to:vt)
   {
       dd[to]=1;
   }
   vector<ll> vt1;
   vector<ll> vt2;
   for (int i=0;i<vt.size();i++)
   {
       ll x=vt[i];
       string p;
       for (int i=1;i<=n;i++)
       {
           if (dd[i])
           {
               if (i==x)
               {
                   p.pb('1');
               }
               else
               {
                   p.pb('0');
               }
           }
           else
           {
               p.pb('1');
           }
       }
       if (check_element(p))
       {
           vt1.pb(x);
       }
       else
       {
           vt2.pb(x);
       }
   }
   ll mid=(l+r)/2;
   dosth1(l,mid,vt1);
   dosth1(mid+1,r,vt2);
}
vector<ll> restore_permutation(int n1, int w, int r)
{
    n=n1;
    dosth(1,n);
    compile_set();
    vector<ll> vt;
    for (int i=1;i<=n;i++) vt.pb(i);
    dosth1(1,n,vt);
    vector<ll> ans;
    for (int i=1;i<=n;i++) ans.pb(res[i]-1);
    return ans;
}
/*int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    if (fopen("t.inp","r"))
    {
        freopen("test.inp","r",stdin);
        freopen("test.out","w",stdout);
    }
}*/

Compilation message (stderr)

messy.cpp:15:15: warning: overflow in conversion from 'double' to 'int' changes value from '3.0e+18' to '2147483647' [-Woverflow]
   15 | const ll base=3e18;
      |               ^~~~
messy.cpp: In function 'void dosth1(int, int, std::vector<int>)':
messy.cpp:55:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |    for (int i=0;i<vt.size();i++)
      |                 ~^~~~~~~~~~

• Subtask 1: 20 / 20
• Subtask 2: 18 / 18
• Subtask 3: 11 / 11
• Subtask 4: 21 / 21
• Subtask 5: 30 / 30