제출 #1339854

#제출 시각아이디문제언어결과실행 시간메모리
1339854nathlol2Arranging Shoes (IOI19_shoes)C++20
100 / 100
194 ms273368 KiB
#include "shoes.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e5 + 5;
ll ans, ol;
vector<pair<int, int>> v;
queue<int> l[N], r[N];

struct fenwick{
	int bit[N];
	void upd(int i, int x){
		for(; i<N; i += i & -i) bit[i] += x;
	}
	ll qry(int i){
		ll res = 0;
		for(; i>0; i -= i & -i) res += bit[i];
		return res;
	}
}fw;

ll count_swaps(vector<int> s){
	int n = s.size();
	for(int i = 0;i<n;i++){
		if(s[i] < 0){
			if(!r[-s[i]].empty()) v.push_back({r[-s[i]].front(), i}), r[-s[i]].pop(), ++ans;
			else l[-s[i]].push(i);
		}else{
			if(!l[s[i]].empty()) v.push_back({l[s[i]].front(), i}), l[s[i]].pop();
			else r[s[i]].push(i);
		}
	}
	sort(v.begin(), v.end());
	for(int i = 1;i<N;i++) fw.upd(i, 1);
	for(auto &[x, y] : v) ++x, ++y;
	for(int i = 0;i<v.size();i++){
		ll f = fw.qry(v[i].second - 1) - fw.qry(v[i].first);
		ans += f;
		fw.upd(v[i].first, -1);
		fw.upd(v[i].second, -1);
	}
	return ans - ol;
}
#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...