Submission #132887

#	Time	Username	Problem	Language	Result	Execution time	Memory
132887		amiratou	자리 배치 (IOI18_seats)	C++14	11 / 100	4091 ms	56716 KiB

seats

#pragma GCC optimize("Ofast")
#pragma GCC target("sse4")
#include "seats.h"
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define ll long long
#define ii pair<int,int>
#define INF (int)(1e9)
pair<ii,ii> tab[1000002];
int A[1000002],n,h,w;
ii st_x[4000005],st_y[4000005];
vector<int> RR,c;
int ones=0,zeros=0;
void build(int node,int l,int r){
	if(l==r){
		st_x[node]=ii(RR[l],RR[l]);
		st_y[node]=ii(c[l],c[l]);
		return ;
	}
	int med=(l+r)>>1;
	build((node<<1),l,med),build((node<<1)+1,med+1,r);
	st_x[node].fi=max(st_x[(node<<1)].fi,st_x[(node<<1)+1].fi);
	st_x[node].se=min(st_x[(node<<1)].se,st_x[(node<<1)+1].se);
	st_y[node].fi=max(st_y[(node<<1)].fi,st_y[(node<<1)+1].fi);
	st_y[node].se=min(st_y[(node<<1)].se,st_y[(node<<1)+1].se);
}
ii query_x(int node,int l,int r,int i,int j){
	if(l>j||r<i)
		return ii(-1,INF);
	if(l>=i&&r<=j)
		return st_x[node];
	int med=(l+r)>>1;
	ii q1=query_x((node<<1),l,med,i,j),q2=query_x((node<<1)+1,med+1,r,i,j);
	return ii(max(q1.fi,q2.fi),min(q1.se,q2.se));
}
ii query_y(int node,int l,int r,int i,int j){
	if(l>j||r<i)
		return ii(-1,INF);
	if(l>=i&&r<=j)
		return st_y[node];
	int med=(l+r)>>1;
	ii q1=query_y((node<<1),l,med,i,j),q2=query_y((node<<1)+1,med+1,r,i,j);
	return ii(max(q1.fi,q2.fi),min(q1.se,q2.se));
}
void update(int node,int l,int r,int idx,int x,int y){
	if(r<idx||l>idx)
		return ;
	if(l==r){
		st_x[node]=ii(x,x);
		st_y[node]=ii(y,y);
		return ;
	}
	int med=(l+r)>>1;
	update((node<<1),l,med,idx,x,y),update((node<<1)+1,med+1,r,idx,x,y);
	st_x[node].fi=max(st_x[(node<<1)].fi,st_x[(node<<1)+1].fi);
	st_x[node].se=min(st_x[(node<<1)].se,st_x[(node<<1)+1].se);
	st_y[node].fi=max(st_y[(node<<1)].fi,st_y[(node<<1)+1].fi);
	st_y[node].se=min(st_y[(node<<1)].se,st_y[(node<<1)+1].se);
}
void give_initial_chart(int H, int W, vector<int> R, vector<int> C) {
	n=H*W,h=H,w=W;
	RR=R,c=C;
	if(H<=1000&&W<=1000){
		build(1,0,n-1);
		return;
	}
	else{
		for (int i = 0; i < n; ++i)
		{
			tab[i].fi=ii(R[i],C[i]);
			tab[i].se=ii(R[i],C[i]);
			if(i){
				tab[i].fi.fi=max(tab[i].fi.fi,tab[i-1].fi.fi);
				tab[i].fi.se=max(tab[i].fi.se,tab[i-1].fi.se);
				tab[i].se.fi=min(tab[i].se.fi,tab[i-1].se.fi);
				tab[i].se.se=min(tab[i].se.se,tab[i-1].se.se);
			}
			if((tab[i].fi.fi-tab[i].se.fi+1)*(tab[i].fi.se-tab[i].se.se+1)==(i+1))A[i]=1,ones++;
			else A[i]=0,zeros++;
		}
	}
}

int swap_seats(int a, int b) {
	if(h<=1000&&w<=1000){
		swap(RR[a],RR[b]),swap(c[a],c[b]);
		update(1,0,n-1,a,RR[a],c[a]),update(1,0,n-1,b,RR[b],c[b]);
		int ans=0;
		ii X=ii(-1,INF),Y=ii(-1,INF);
		for (int i = 0; i < n;)
		{
			ii q1=query_x(1,0,n-1,0,i),q2=query_y(1,0,n-1,0,i);
			X.fi=max(X.fi,q1.fi),X.se=min(X.se,q1.se),Y.fi=max(Y.fi,q2.fi),Y.se=min(Y.se,q2.se);
			if((X.fi-X.se+1)*(Y.fi-Y.se+1)==(i+1))
				ans++;
			if((X.fi-X.se+1)*(Y.fi-Y.se+1)>i+1)
				i=(X.fi-X.se+1)*(Y.fi-Y.se+1)-1;
			else i++;
		}
		return ans;

	}
	else{
		swap(RR[a],RR[b]),swap(c[a],c[b]);
		for (int i = min(a,b); i <= max(a,b); ++i)
		{
			tab[i].fi=ii(RR[i],c[i]);
			tab[i].se=ii(RR[i],c[i]);
			if(i){
				tab[i].fi.fi=max(tab[i].fi.fi,tab[i-1].fi.fi);
				tab[i].fi.se=max(tab[i].fi.se,tab[i-1].fi.se);
				tab[i].se.fi=min(tab[i].se.fi,tab[i-1].se.fi);
				tab[i].se.se=min(tab[i].se.se,tab[i-1].se.se);
			}
			if((tab[i].fi.fi-tab[i].se.fi+1)*(tab[i].fi.se-tab[i].se.se+1)==(i+1)){
				if(!A[i])
					A[i]=1,ones++,zeros--;
			}
			else {
				if(A[i])
					A[i]=0,ones--,zeros++;
			}
		}
		return ones;
	}
}

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