제출 #876426

#제출 시각아이디문제언어결과실행 시간메모리
876426Elvin_FritlArranging Shoes (IOI19_shoes)C++17
10 / 100
681 ms1048576 KiB
#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
 
const int N = 3e5 + 545 , inf = 1e9 + 12 , mod = 1e9 + 7;
 
      #include "shoes.h"
 
struct segtree {
	ll tree[N*3] ;
	void built(int v,int l,int r) {
		if(l == r) {
			tree[v] = 1;
			return;
		}
		int mid = (l + r)/2;
		built(v*2 , l , mid);
		built(v*2 + 1 , mid+1 , r);
		tree[v] = tree[v*2] + tree[v*2 + 1];
	}
	void update(int v,int l,int r,int ml,int val) {
		if(r < ml || ml < l) {
			return;
		}
		if(l == r && ml == l) {
			tree[v] += val;
			return;
		}
		int mid = (l + r)/2;
		update(v , l , mid , ml , val);
		update(v , mid + 1 , r , ml , val);
		tree[v] = tree[v*2] + tree[v*2 + 1];
	}
	ll get(int v,int l,int r,int ml,int mr) {
		if(r < ml && mr < l) {
			return 0;
		}
		if(ml <= l && r <= mr) {
			return tree[v];
		}
		int mid = (l + r)/2;
		return (get(v , l , mid , ml , mr) + get(v , mid + 1 , r , ml ,mr));
	}
};
 
segtree S;
 
ll count_swaps(vector<int> v) {
	int n = v.size()/2;
	set<pair<ll , ll>> s;
	v.push_back(0LL);
	for(int i = 2*n;i>=1;i--){
		v[i] = v[i-1];
		s.insert({v[i] , i});
	}
	///cerr << "true1\n";
	S.built(1,1,2*n);
	ll res = 0;
	for(int i=1;i<=n;i++) {
		///cerr << "true2\n";
		if(s.find({v[i] , i}) != s.end()){
			///cerr << "true2\n";
			auto it = s.lower_bound({-v[i] , i});
			if((*it).second - 2 >= i) {
				res += S.get(1,1,2*n,i + 1,(*it).second-1);
			}
			if(v[i] > 0) {
				res++;
			}
			S.update(1 , 1 , 2*n , i, -1);
			S.update(1 , 1 , 2*n , (*it).second, -1);
			if(s.find({v[i], i}) != s.end()){
				s.erase(s.find({v[i], i}));
			}
			if(s.find({-v[i], (ll)(*it).second}) != s.end()){
				s.erase(s.find({-v[i], (ll)(*it).second}));
			}
		}
	}
	return res;
}
#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...