| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 209906 | Segtree | Arranging Shoes (IOI19_shoes) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<unordered_set>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> P;
typedef unordered_set<ll> uset;
#define rep(i,n) for(int i=0;i<n;i++)
#define rrep(i,n) for(int i=0;i<=n;i++)
#define chmin(a,b) a=min(a,b)
#define chmax(a,b) a=max(a,b)
#define all(x) x.begin(),x.end()
#pragma gcc optimize("O3")
#pragma gcc optimize("unroll-loops")
#pragma gcc target("avx2,see4")
#define N 100010
ll dat[N];
void add(ll i,ll x){
for(;i<N;i+=i&-i)dat[i]+=x;
}
ll sum(ll i){
ll res=0;
for(;i>0;i-=i&-i)res+=dat[i];
return res;
}
ll calc_tentou(vector<ll> p){
int n=p.size();
rep(i,n)p[i]++;
rep(i,n)dat[i]=0;
ll ans=0;
rep(i,n){
ans+=i-sum(p[i]);
add(p[i],1);
}
return ans;
}
ll count_swaps(vector<int> s){
int n=s.size()/2;
vector<P> L,R;//左・右の靴の{サイズ、場所}
for(int i=0;i<2*n;i++){
if(s[i]<0)L.push_back(make_pair(-s[i],i));
else R.push_back(make_pair(s[i],i));
}
vector<ll> pl[N];//pl[x]:=サイズxの右の靴のある場所、後ろが小さい
for(auto e:R){
pl[e.first].push_back(e.second);
}
rep(i,N)reverse(all(pl[i]));
vector<ll> p(2*n);//p[i]:=元々iにあった靴がいく場所
for(int i=0;i<n;i++){
p[L[i].second]=i*2;
p[pl[L[i].first].back()]=i*2+1;
pl[L[i].first].pop_back();
}
return calc_tentou(p);
}
int main(){
ll n;
cin>>n;
vector<int> s(2*n);
rep(i,2*n)cin>>s[i];
cout<<count_swaps(s)<<endl;
}