Submission #341247

#TimeUsernameProblemLanguageResultExecution timeMemory
341247YJUArranging Shoes (IOI19_shoes)C++14
0 / 100
532 ms546148 KiB
#include<bits/stdc++.h>
#pragma GCC optimize("unroll-loops,no-stack-protector")
using namespace std;
typedef int ll;
typedef long double ld;
typedef pair<ll,ll> pll;
const ll MOD=1e9+7;
const ll MOD2=998244353;
const ll N=1e5+5;
const ll K=350;
const ld pi=acos(-1);
#define SQ(i) ((i)*(i))
#define REP(i,n) for(ll i=0;i<n;i++)
#define REP1(i,n) for(ll i=1;i<=n;i++)
#define pb push_back
#define mp make_pair
#define X first
#define Y second
#define setp setprecision
#define lwb lower_bound
#define SZ(_a) (ll)_a.size()

ll bit[N];
queue<ll> m[4*N];

void ins(ll id,ll del){
	++id;
	for(;id;id+=(id&-id)){
		bit[id]+=del;
	}
}

ll q(ll id){
	++id;
	ll sum=0;
	for(;id;id-=(id&-id)){
		sum+=bit[id];
	}
	return sum;
}

long long count_swaps(vector<int> s){
	long long ans=0;
	ll n=SZ(s);
	REP(i,n){
		if(SZ(m[-s[i]+n])){
			ans-=(q(i)-q(m[-s[i]+n].front()));
			ans+=i-m[-s[i]+n].front()-1+(s[i]<0?1:0);
			ins(m[-s[i]+n].front(),-1);
			m[-s[i]+n].pop();
		}else{
			ins(i,1);
			m[s[i]+n].push(i);
		}
	}
	return ans;
}
#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...