Submission #708034

#TimeUsernameProblemLanguageResultExecution timeMemory
708034MODDIArranging Shoes (IOI19_shoes)C++14
50 / 100
116 ms49960 KiB
	#include "shoes.h"
	#include <bits/stdc++.h>
	#define ll long long
	#define pii pair<int,int>
	#define vi vector<int>
	#define vl vector<ll>
	#define mp make_pair
	#define pb push_back
	#define fi first
	#define se second;
	using namespace std;
	int tree[4 * 100100], lazy[4 * 100100];
	multiset<int> l_pos[100010], r_pos[100010];
	void update(int node, int l, int r, int L, int R, int add){
		if(lazy[node]){
			tree[node] += (r - l + 1) * lazy[node];
			if(l != r){
				lazy[node * 2] += lazy[node];
				lazy[node * 2 + 1] += lazy[node];
			}	
			lazy[node] = 0;
		}
		if(R < l || r < L)	return;
		if(L <= l && r <= R){
			tree[node] += (r - l + 1) * add;
			if(l != r){
				lazy[node * 2 + 1] += add;
				lazy[node * 2] += add;
			}
			return;
		}
		int mid = (l + r) / 2;
		update(node * 2, l, mid, L, R, add);
		update(node * 2 + 1, mid + 1, r, L, R, add);
		tree[node] = tree[node * 2] + tree[node * 2 + 1];
	}
	int query(int node, int l, int r, int L, int R){
		if(lazy[node] != 0){
			tree[node] += (r - l + 1) * lazy[node];
			if(l != r){
				lazy[node * 2] += lazy[node];
				lazy[node * 2 + 1] += lazy[node];
			}
			lazy[node] = 0;
		}
		if(r < L || l > R)	return 0;
		if(L <= l && r <= R)	return tree[node];
		
		int mid = (l + r) / 2;
		int left = query(node * 2, l, mid, L, R);
		int right = query(node * 2 + 1, mid + 1, r, L, R);
		return left + right;
	}
	long long count_swaps(std::vector<int> s) {
		ll ans = 0;
		vi arr = s;
		int n = s.size();
		for(int i = 0; i < n; i++){
			if(s[i] < 0){
				l_pos[abs(s[i])].insert(i);
			}
			else
				r_pos[s[i]].insert(i);
		}
		memset(tree, 0, sizeof tree);
		memset(lazy, 0, sizeof lazy);
		bool vis[n];
		memset(vis, false, sizeof vis);
		for(int i = 0; i < n; i++){
			if(vis[i])	continue;
			if(arr[i] < 0){
				int cur_pos = i + query(1, 0, n-1, i, i);
				int p_pos = *r_pos[abs(arr[i])].begin();
				int get = query(1, 0, n-1, p_pos, p_pos) + p_pos;
	//			cout<<cur_pos<<" "<<get<<endl;
				ans += (get - cur_pos - 1);
				update(1, 0, n-1, i, p_pos-1, 1);
				vis[p_pos] = true;
				vis[i] = true;
				r_pos[abs(arr[i])].erase(p_pos);
				l_pos[abs(arr[i])].erase(i);
			}	
			else{
				int cur_pos = i + query(1, 0, n-1, i, i);
				int p_pos = *l_pos[arr[i]].begin();
				int get = query(1, 0, n-1, p_pos, p_pos) + p_pos;
	//			cout<<cur_pos<<" "<<get<<endl;
				ans += (get - cur_pos);
	//			cout<<get - cur_pos<<" "<<get<<" "<<cur_pos<<endl;
				update(1, 0, n-1, i, p_pos-1, 1);
				vis[p_pos] = true;
				vis[i] = true;
				l_pos[arr[i]].erase(p_pos);
				r_pos[arr[i]].erase(i);
			}
		}
		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...