제출 #153528

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
153528		myungwoo	Arranging Shoes (IOI19_shoes)	C++14	100 / 100	458 ms	275244 KiB

shoes

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

#define MAXN 200005
typedef long long lld;

int N;
int A[MAXN], bit[MAXN];
bool vis[MAXN];
stack <int> L[MAXN], R[MAXN];

void add(int n)
{
	for (;n<=N;n+=n&-n) bit[n]++;
}
int get(int n)
{
	int ret = 0;
	for (;n;n^=n&-n) ret += bit[n];
	return ret;
}

int move(int a, int b)
{
	// a -> b
	vis[a] = 1; add(a);
	return b-(a-get(a-1));
}

lld count_swaps(vector<int> arr)
{
	N = arr.size();
	for (int i=1;i<=N;i++) A[i] = arr[i-1];
	for (int i=1;i<=N;i++){
		if (A[i] < 0) L[-A[i]].push(i);
		else R[A[i]].push(i);
	}
	lld total = 0;
	int pos = N;
	for (int i=N;i;i--) if (!vis[i]){
		int t = abs(A[i]);
		total += move(R[t].top(), pos);
		total += move(L[t].top(), pos-1);
		R[t].pop(); L[t].pop();
		pos -= 2;
	}
	return total;
}

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