Submission #1339829

#TimeUsernameProblemLanguageResultExecution timeMemory
1339829nathlol2Arranging Shoes (IOI19_shoes)C++20
50 / 100
1103 ms137752 KiB
#include "shoes.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 5;
ll ans, f, 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 = 0;i<v.size() - 1;i++){
		if(v[i].second - v[i + 1].first > 0){
			ans += v[i].second - v[i + 1].first;
			for(int j = i + 1;j<v.size();j++){
				if(v[j].first < v[i].second) ++v[j].first;
				if(v[j].second < v[i].second) ++v[j].second;
			}
		}
	}
	// for(int i = v.size() - 1;i>=0;i--){
	// 	ol += (fw.qry(2 * n) - fw.qry(v[i].second));
	// 	fw.upd(v[i].second, 1);
	// }
	// if(ol > 0) return ans - ol + 1;
	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...