Submission #242902

#	Time	Username	Problem	Language	Result	Execution time	Memory
242902		uacoder123	Arranging Shoes (IOI19_shoes)	C++14	100 / 100	588 ms	32416 KiB

shoes

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

using namespace std;
#define F first
#define FOR(i,a,b) for (auto i = (a); i <= (b); ++i)
#define NFOR(i,a,b) for(auto i = (a); i >= (b); --i)
#define all(x) (x).begin(), (x).end()
#define sz(x) lli(x.size())
#define mp(i,a) make_pair(i,a)
#define pb(a) push_back(a)
#define bit(x,b) (x&(1LL<<b))
 
typedef long long int lli;
typedef pair <lli,lli> ii;
typedef pair <lli,ii> iii;
typedef vector <lli> vi;
lli segt[8*100000]={};
void up(lli node,lli l,lli r,lli in)
{
  if(l==r)
  {
    segt[node]=1;
    return;
  }
  lli m=(l+r)/2;
  if(in<=m)
    up(2*node+1,l,m,in);
  else
    up(2*node+2,m+1,r,in);
  segt[node]=segt[2*node+1]+segt[2*node+2];
}
lli qu(lli node,lli l,lli r,lli s,lli e)
{
  if(l>=s&&r<=e)
    return(segt[node]);
  if(l>e||r<s)
    return(0);
  lli m=(l+r)/2;
  lli q1=qu(2*node+1,l,m,s,e),q2=qu(2*node+2,m+1,r,s,e);
  return(q1+q2);
}
long long count_swaps(std::vector<int> s) 
{
  lli n=s.size(),f;
  lli ans=0;
  set<ii>inva,vain;
  for(lli i=0;i<n;++i)
  {
    f=s[i];
    if(f<0)
      f=abs(f)+n/2;
    inva.insert(mp(i,f));
    vain.insert(mp(f,i));
  }
  while(inva.size())
  {
    ii it,f=(*inva.begin());
    inva.erase(inva.begin());
    vain.erase(mp(f.second,f.F));
    f.F+=qu(0,0,n-1,f.F,n-1);
    if(f.second>n/2)
      it=(*vain.lower_bound(mp(f.second-n/2,0)));
    else
      it=(*vain.lower_bound(mp(f.second+n/2,0)));
    vain.erase(it);
    inva.erase(mp(it.second,it.F));
    up(0,0,n-1,it.second);
    it.second+=qu(0,0,n-1,it.second,n-1)-1;
    ans+=it.second-f.F-1;
    if(f.second<=n/2)
      ans++;
  }
  return(ans);
}

• Subtask 1: 10 / 10
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 15 / 15
• Subtask 5: 20 / 20
• Subtask 6: 15 / 15