Submission #212834

#TimeUsernameProblemLanguageResultExecution timeMemory
212834MarcoMeijerSeats (IOI18_seats)C++14
31 / 100
4098 ms118136 KiB
#include "seats.h"
#include <bits/stdc++.h>
using namespace std;

//macros
typedef long long ll;
typedef pair<int, int> ii;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<ll> vll;
#define REP(a,b,c) for(int a=int(b); a<int(c); a++)
#define RE(a,c) REP(a,0,c)
#define RE1(a,c) REP(a,1,c+1)
#define REI(a,b,c) REP(a,b,c+1)
#define REV(a,b,c) for(int a=int(c-1); a>=int(b); a--)
#define INF 1e9
#define pb push_back
#define fi first
#define se second
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

template<class T, int DEF=0>
class Seg {
public:
	vi SEG;
	int sz;
	T t;

	void resize(int _sz) {
		sz = _sz;
		SEG.resize(sz*4);
	}
	void build(int p=0, int l=0, int r=-1) {
		if(r == -1) r = sz-1;
		if(l == r) {
			SEG[p] = 0;
			return;
		}
		int m=(l+r)/2;
		build(p*2+1,l,m);
		build(p*2+2,m+1,r);
		if(t(SEG[p*2+1], SEG[p*2+2])) 	SEG[p] = SEG[p*2+1];
		else							SEG[p] = SEG[p*2+2];
	}
	int get(int i, int j, int p=0, int l=0, int r=-1) {
		if(r == -1) r = sz-1;
		if(j < l || i > r) return DEF;
		if(i <= l && j >= r) return SEG[p];
		int m=(l+r)/2;
		int a=get(i,j,p*2+1,l,m);
		int b=get(i,j,p*2+2,m+1,r);
		if(t(a,b))	return a;
		else		return b;
	}
	void set(int i, int x, int p=0, int l=0, int r=-1) {
		if(r == -1) r = sz-1;
		if(i < l || i > r) return;
		if(l == r) {
			SEG[p] = x;
			return;
		}
		int m=(l+r)/2;
		set(i,x,p*2+1,l,m);
		set(i,x,p*2+2,m+1,r);
		if(t(SEG[p*2+1], SEG[p*2+2])) 	SEG[p] = SEG[p*2+1];
		else							SEG[p] = SEG[p*2+2];
	}
};

int h, w, n;
vi r, c;
vector<Seg<greater<int>>> maxR;
vector<Seg<greater<int>>> maxC;

void give_initial_chart(int H, int W, vi R, vi C) {
	h=H, w=W, r=R, c=C;
	n = h*w;

	maxR.resize(h);
	maxC.resize(w);
	RE(i,h) maxR[i].resize(w);
	RE(i,w) maxC[i].resize(h);
	RE(i,n) maxR[r[i]].set(c[i],i);
	RE(i,n) maxC[c[i]].set(r[i],i);
}

int swap_seats(int a, int b) {
	if(b < a) swap(a,b);
	swap(r[a], r[b]);
	swap(c[a], c[b]);

	maxR[r[a]].set(c[a],a);
	maxR[r[b]].set(c[b],b);
	maxC[c[a]].set(r[a],a);
	maxC[c[b]].set(r[b],b);

	int cnt	  = 0;
	int cMinR = r[0];
	int cMaxR = r[0];
	int cMinC = c[0];
	int cMaxC = c[0];
	int i 	  = 0;
	while(1) {
		if(i == (cMaxR-cMinR+1)*(cMaxC-cMinC+1)-1) cnt++;

		int mn	= INF;
		int res = -1;

		if(cMinR != 0  ) {
			int val = maxR[cMinR-1].get(cMinC, cMaxC);
			if(val < mn) {
				mn = val;
				res = 0;
			}
		}
		if(cMinC != 0  ) {
			int val = maxC[cMinC-1].get(cMinR, cMaxR);
			if(val < mn) {
				mn = val;
				res = 1;
			}
		}
		if(cMaxR != h-1) {
			int val = maxR[cMaxR+1].get(cMinC, cMaxC);
			if(val < mn) {
				mn = val;
				res = 2;
			}
		}
		if(cMaxC != w-1) {
			int val = maxC[cMaxC+1].get(cMinR, cMaxR);
			if(val < mn) {
				mn = val;
				res = 3;
			}
		}

		if(mn == INF) break;

		if(res == 0) cMinR--;
		if(res == 1) cMinC--;
		if(res == 2) cMaxR++;
		if(res == 3) cMaxC++;

		i = max(i, mn);
	}

	return cnt;
}
#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...