Submission #419908

#	Time	Username	Problem	Language	Result	Execution time	Memory
419908		cfalas	Seats (IOI18_seats)	C++14	11 / 100	4073 ms	124304 KiB

seats

#include "seats.h"
#include<bits/stdc++.h>
using namespace std;
#define MOD 1000100
#define MID ((l+r)/2)
#define ll long long
#define F first
#define S second
typedef pair<int, int> ii;

#define FOR(i,n) for(int i=0;i<((int)(n));i++)
#define FORi(i,a,b) for(int i=((int)(a));i<((int)(b));i++)
#define FOA(v, a) for(auto v : a)

int n, m;

static int rl, rr;
static ii val;
static int x;
static pair<ii, ii> RAND = {{MOD, MOD}, {0,0}};

struct seg_tree{
	seg_tree *left=0;
	seg_tree *right=0;
	pair<ii, ii> v = RAND;

	void update(int l=0, int r=MOD){
		//cout<<"B "<<l<<" "<<r<<" "<<mina.F<<" "<<mina.S<<endl;
		if(l==r && l==x){
			v = {val, val};
		}
		else if(l>x || r<x) assert(0);
		else{
			pair<ii, ii> ans1 = RAND;
			pair<ii, ii> ans2 = RAND;
			if(x<=MID){
				if(!left) left = new seg_tree;
				left->update(l, MID);
			}
			else{
				if(!right) right = new seg_tree;
				right->update(MID+1, r);
			}
			if(left) ans1 = left->v;
			if(right) ans2 = right->v;

			v = {{min(ans1.F.F, ans2.F.F), min(ans1.F.S, ans2.F.S)}, {max(ans1.S.F, ans2.S.F), max(ans1.S.S, ans2.S.S)}};
		}
	}
	pair<ii,ii> query(int l=0, int r=MOD){
		if(rl<=l && r<=rr) return v;
		else if(r<rl || rr<l) return RAND;
		else{
			if(!left) left = new seg_tree;
			if(!right) right = new seg_tree;
			pair<ii, ii> ans1 = RAND;
			ans1 = left->query(l, MID);
			pair<ii, ii> ans2 = RAND;
			ans2 = right->query(MID+1, r);

			ii mina = {min(ans1.F.F, ans2.F.F), min(ans1.F.S, ans2.F.S)};
			ii maxa = {max(ans1.S.F, ans2.S.F), max(ans1.S.S, ans2.S.S)};
			return {mina, maxa};

		}
	}
};


seg_tree seg; 

pair<ii, ii> query(int l, int r){
	rl = l;
	rr = r;
	return seg.query();
}
void update(ii v, int xx){
	val = v;
	x = xx;
	seg.update();
}

vector<ii> pos;
void give_initial_chart(int H, int W, std::vector<int> R, std::vector<int> C) {
	cerr<<"STARTED"<<endl;
	n = H, m = W;
	pos.resize(n*m);
	FOR(i, n*m){
		pos[i] = {R[i], C[i]};
		update({R[i], C[i]}, i);
	}
	cerr<<"DONE"<<endl;
}

int swap_seats(int a, int b) {
	int r=0;
	int i=0;
	if(n+m<=2000){
		pair<ii, ii> temp1 = query(a, a);
		pair<ii, ii> temp2 = query(b, b);
		update(temp1.F, b);
		update(temp2.F, a);
		while(i<n*m){
			//cerr<<i<<" ";
			pair<ii, ii> ans = query(0, i);
			int q =  (ans.S.F - ans.F.F+1) * (ans.S.S - ans.F.S+1);
			if(q==i+1) r++, i++;
			else i = q-1;
		}
	}
	else{
		pair<ii, ii> val = RAND;
		swap(pos[a], pos[b]);
		while(i<n*m){
			//cerr<<i<<" ";
			val.F.F = min(val.F.F, pos[i].F);
			val.F.S = min(val.F.S, pos[i].S);
			val.S.F = max(val.S.F, pos[i].F);
			val.S.S = max(val.S.S, pos[i].S);
			int q =  (val.S.F - val.F.F+1) * (val.S.S - val.F.S+1);
			if(q==i+1) r++;
			i++;
		}

	}
	//cerr<<cnt<<endl;
	return r;
}

• Subtask 1: 5 / 5
• Subtask 2: 6 / 6
• Subtask 3: 0 / 20
• Subtask 4: 0 / 6
• Subtask 5: 0 / 33
• Subtask 6: 0 / 30