제출 #751648

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
751648		minhcool	코끼리 (Dancing Elephants) (IOI11_elephants)	C++17	26 / 100	9079 ms	19016 KiB

elephants

#include "elephants.h"
#include<bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using namespace std;

#define ll long long
#define fi first
#define se second
#define pb push_back
//#define mp make_pair

typedef pair<int, int> ii;
typedef pair<ii, ii> iii;
typedef pair<ii, ii> iiii;

const int N = 1e6 + 5;

const int oo = 1e9 + 7, mod = 1e9 + 7;

mt19937 rng(1);

ll rnd(ll l, ll r){
	ll temp = rng() % (r - l + 1);
	return abs(temp) + l;
}

set<ii> mls;

int n, l, arr[N];

pair<int, int> nxt[N][21];

void rebuild(){
	arr[n] = oo;
	vector<ii> v;
	for(int i = 0; i < n; i++) v.pb({arr[i], i});
	sort(v.begin(), v.end());
	for(int i = 0; i <= n; i++){
		//cout << arr[i] + l + 1 << "\n";
		vector<ii>::iterator it = lower_bound(v.begin(), v.end(), make_pair(arr[i] + l + 1, -oo));
		nxt[i][0] = (it == v.end() ? make_pair(oo, n) : (*it));
		//cout << nxt[i][0].fi << " " << nxt[i][0].se << "\n";
	}
	for(int i = 1; i <= 17; i++){
		for(int j = 0; j <= n; j++){
			nxt[j][i] = nxt[nxt[j][i - 1].se][i - 1];
			//cout << i << " " << j << " " << nxt[j][i].fi << " " << nxt[j][i].se << "\n";
		}
	}
}

int counter = 0;

set<int> poss;

unordered_set<int> ins;
gp_hash_table<int, int> mp1, mp2;

bool all(int pos){
	return (mp1[pos] == mp2[pos]);
}

void init(int N, int L, int X[]){
	n = N, l = L;
	for(ll i = 0; i < n; i++) arr[i] = X[i];
	for(ll i = 0; i < n; i++) mls.insert({arr[i], i});
	for(int i = 0; i < n; i++) mp1[arr[i]]++;
	rebuild();
	poss.insert(-1), poss.insert(1e9 + 1);
}

int cnt = 0;
int update(int i, int y){
	counter++;
	mls.erase({arr[i], i});
	mp1[arr[i]]--;
	if(ins.find(i) != ins.end()) mp2[arr[i]]--;
	poss.insert(arr[i]);
	arr[i] = y;
	ins.insert(arr[i]);
	mp2[arr[i]]++;
	poss.insert(arr[i]);
	mp1[arr[i]]++;
	mls.insert({arr[i], i});
	/*
	int ans = 0, lst = -1;
	for(auto it : mls){
		if(lst > it) continue;
		ans++;
		lst = it + l;
	}*/
	int ans = 0;
	int lst = -1;
	bool ck = 1;
	for(auto it : poss){
	//	cout << counter << " " << it << " " << lst << " " << arr[lst] << " " << ans << " " << all(arr[lst]) <<  "\n";
		//cout << ans << " " << lst << "\n";
		if(lst < 0){
			set<ii>::iterator it2 = mls.lower_bound({it, -oo});
			if((it2 != mls.end() && (*it2).fi == it) && (it > (-oo + l))){
				ans++;
				it2 = mls.lower_bound({-oo + l + 1, -oo});
				if(it2 == mls.end()) break;
				lst = (*it2).se;
				//lst = it;
			}
			else{
				it2 = mls.lower_bound({-oo + l + 1, -oo});
				if(it2 == mls.end()) break;
				ans++;
				lst = (*it2).se;
			}
		}
		else if(all(arr[lst])){
			set<ii>::iterator it2 = mls.lower_bound({arr[lst] + l + 1, -oo});
			if(it2 == mls.end()) break;
			ans++;
			lst = (*it2).se;
		}
		else{
			for(int i = 17; i >= 0; i--){
			//	cout << i << " " << nxt[lst][i].fi << " " << nxt[lst][i].se << " " << it << "\n";
				if(nxt[lst][i].fi >= it) continue;
				//if(!ck) continue;
				ans += (1LL << i);
			//	assert(arr[nxt[lst][i].se] == nxt[lst][i].fi);
				lst = nxt[lst][i].se;
			}
		//cout << ans << " " << lst << " " << arr[lst] << " " << it << "\n";
			set<ii>::iterator it2 = mls.lower_bound({it, -oo});
			//cout << arr[lst] + l << " " << ans << "\n";
			if((it2 != mls.end() && (*it2).fi == it) && (it > (arr[lst] + l))){
				ans++;
				it2 = mls.lower_bound({it + l + 1, -oo});
			//	cout << it + l + 1 << " " << (it2 == mls.end()) << "\n";
				if(it2 == mls.end()) break;
				ans++;
				//cout << "OK " << it << " " << it + l + 1 << " " << (*it2).se << "\n";
				lst = (*it2).se;
				//lst = it;
			}
			else{
		//		cout << "OKAY\n";
				it2 = mls.lower_bound({arr[lst] + l + 1, -oo});
			//	cout << arr[lst] + l + 1 << " " << (it2 == mls.end()) << "\n";
		//		cout << (it2 == mls.end()) << "\n";
				if(it2 == mls.end()) break;
				ans++;
			//	cout << "OK " << arr[lst] + l + 1 << " " << (*it2).se << "\n";
				lst = (*it2).se;
			}
		//	cout << ans << "\n";
			//cout << lst << "\n";
		}
		ck = 1;
	}
	cnt++;
	//if(!(cnt % 1000)) cout << cnt << " " << "OK\n";
	if(!(counter % 100)){
		poss.clear();
		mp2.clear();
		ins.clear();
		rebuild();
		poss.insert(-1), poss.insert(1e9 + 1);
	}
	return ans;
}

/*
void process(){

}

signed main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	ll t;
	cin >> t;
	while(t--) process();
}
*/

컴파일 시 표준 에러 (stderr) 메시지

elephants.cpp: In function 'int update(int, int)':
elephants.cpp:95:7: warning: variable 'ck' set but not used [-Wunused-but-set-variable]
   95 |  bool ck = 1;
      |       ^~

• Subtask 1: 10 / 10
• Subtask 2: 16 / 16
• Subtask 3: 0 / 24
• Subtask 4: 0 / 47
• Subtask 5: 0 / 3