제출 #1274106

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1274106		almaz	Arranging Shoes (IOI19_shoes)	C++20	50 / 100	374 ms	148352 KiB

shoes

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

// #define int long long
// #define endl '\n'
#define ff first
#define ss second
#define pb push_back
#define all(a) a.begin(), a.end()
#define rall(a) a.rbegin(), a.rend()
#define ar array

const int MOD = 1e9 + 7, N = 2e5 + 5;

long long INF = 1e18;

vector <long long> f;
long long n;

long long get(long long x){
	long long res = 0;
	for(long long i = x;i >= 0;i = (i & (i + 1)) - 1){
		res += f[i];
	}
	return res;
}

long long getsum(long long l ,long long r){
	if(l == 0){
		return get(r);
	}
	return get(r) - get(l - 1);
}

void update(long long x){
	// cout<<x<<endl;
	for(long long i = x;i < n;i = (i | (i + 1))){
		f[i]++;
	}
}

long long count_swaps(std::vector<int> s) {
	n = s.size();
	f.clear();
	f.resize(n);
	map<int, stack <int>> mp;
	for(int i = n - 1;i >= 0;i--){
		mp[s[i]].push(i);
	}
	int ans = 0;
	vector <int> used(n);
	for(int i = 0;i < n;i++){
		if(mp[s[i]].size() == 0 || used[i]) continue;
		int j = mp[-s[i]].top();
		int sum = getsum(i , j);
		// cout<<i << ' '<< j<<endl;
		used[j] = 1;
		// cout<<sum<<endl;
		if(s[i] < 0){
			ans += (j - (i + 1)) - sum;
		}
		else{
			ans += (j - i) - sum;
		}
		mp[s[i]].pop();
		mp[-s[i]].pop();
		update(j);
	}
	// cout<<ans;
	return ans;
}

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