Submission #708038

#TimeUsernameProblemLanguageResultExecution timeMemory
708038MODDIArranging Shoes (IOI19_shoes)C++14
50 / 100
1077 ms11684 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];
	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();
		set<int> idx;
		int nxt[n];
		memset(nxt, -1, sizeof nxt);
		for(int i = n - 1; i >= 0; i--){
			for(auto t : idx){
				if(arr[i] * -1 == arr[t]){
					nxt[i] = t;
				}
			}
			if(nxt[i] > 0)
				idx.erase(nxt[i]);
			else
				idx.insert(i);
		}
		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 = nxt[i];
				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;
			}	
			else{
				int cur_pos = i + query(1, 0, n-1, i, i);
				int p_pos = nxt[i];
				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;
			}
		}
		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...