Submission #419189

#	Time	Username	Problem	Language	Result	Execution time	Memory
419189		peuch	Seats (IOI18_seats)	C++17	5 / 100	4061 ms	121756 KiB

seats

#include "seats.h"
#include<bits/stdc++.h>
using namespace std;

#pragma GCC optimize("Ofast") 
#pragma GCC target("avx,avx2,fma")

const int MAXN = 1e6 + 10;
const int MAXW = 1e3 + 10;

vector<int> r, c;
int h, w;

vector<vector<int> > seg[2];
vector<pair<int, int> > seg2[2];
int cnt = 0;
int sum = 0;
vector<int> ok;

void update(int pos, int ini, int fim, int id, int val, int x, int k);
int query(int pos, int ini, int fim, int p, int q, int x, int k);
void update2(int pos, int ini, int fim, int id, int val, int k);
pair<int, int> query2(int pos, int ini, int fim, int p, int q, int k);

void give_initial_chart(int H, int W, std::vector<int> R, std::vector<int> C) {
	r = R, c = C;
	h = H, w = W;
	seg[0] = vector<vector<int> > (h, vector<int> (4 * w));
	seg[1] = vector<vector<int> > (w, vector<int> (4 * h));
	seg2[0] = vector<pair<int, int> > (4 * h * w);
	seg2[1] = vector<pair<int, int> > (4 * h * w);
	ok = vector<int> (h * w);
	for(int i = 0; i < r.size(); i++){
		update2(1, 0, h * w - 1, i, r[i], 0);
		update2(1, 0, h * w - 1, i, c[i], 1);
	}
	sum = 0;
	int cur = 0;
	int maxi = 0;
	int maxr = r[0], minr = r[0], maxc = c[0], minc = c[0];
	int ret = 0;
	while(cur < h * w){
		pair<int, int> rVal = query2(1, 0, h * w - 1, 0, cur, 0);
		pair<int, int> cVal = query2(1, 0, h * w - 1, 0, cur, 1);
		minr = rVal.second;
		maxr = rVal.first;
		minc = cVal.second;
		maxc = cVal.first;
		if(cur == (maxr - minr + 1) * (maxc - minc + 1) - 1) ret++, ok[cur] = 1, cur++;
		else cur = (maxr - minr + 1) * (maxc - minc + 1) - 1;
	}
	sum = ret;
}

int swap_seats(int a, int b) {
	swap(r[a], r[b]);
	swap(c[a], c[b]);
	update2(1, 0, h * w - 1, a, r[a], 0);
	update2(1, 0, h * w - 1, a, c[a], 1);
	update2(1, 0, h * w - 1, b, r[b], 0);
	update2(1, 0, h * w - 1, b, c[b], 1);
	if(a > b) swap(a, b);
	int cur = a;
	int maxi = 0;
	int maxr = r[0], minr = r[0], maxc = c[0], minc = c[0];
	while(cur <= b){
		pair<int, int> rVal = query2(1, 0, h * w - 1, 0, cur, 0);
		pair<int, int> cVal = query2(1, 0, h * w - 1, 0, cur, 1);
		minr = rVal.second;
		maxr = rVal.first;
		minc = cVal.second;
		maxc = cVal.first;
		sum -= ok[cur];
		if(cur == (maxr - minr + 1) * (maxc - minc + 1) - 1) ok[cur] = 1;
		else ok[cur] = 0;               
		sum += ok[cur];
		cur++;
	}
	return sum;
}

void update(int pos, int ini, int fim, int id, int val, int x, int k){
	cnt++;
	if(ini > id || fim < id) return;
	if(ini == fim){
		seg[k][x][pos] = val;
		return;
	}
	int mid = (ini + fim) >> 1, e = pos << 1, d = e | 1;
	update(e, ini, mid, id, val, x, k);
	update(d, mid + 1, fim, id, val, x, k);
	seg[k][x][pos] = max(seg[k][x][e], seg[k][x][d]);
}

int query(int pos, int ini, int fim, int p, int q, int x, int k){
	cnt++;
	if(ini > q || fim < p) return 0;
	if(ini >= p && fim <= q) return seg[k][x][pos];
	int mid = (ini + fim) >> 1, e = pos << 1, d = e | 1;
	return max(query(e, ini, mid, p, q, x, k), query(d, mid + 1, fim, p, q, x, k));
}

void update2(int pos, int ini, int fim, int id, int val, int k){
	cnt++;
	if(ini > id || fim < id) return;
	if(ini == fim){
		seg2[k][pos] = make_pair(val, val);
		return;
	}
	int mid = (ini + fim) >> 1, e = pos << 1, d = e | 1;
	update2(e, ini, mid, id, val, k);
	update2(d, mid + 1, fim, id, val, k);
	seg2[k][pos].first = max(seg2[k][e].first, seg2[k][d].first);
	seg2[k][pos].second = min(seg2[k][e].second, seg2[k][d].second);
}

pair<int, int> query2(int pos, int ini, int fim, int p, int q, int k){
	cnt++;
	if(ini > q || fim < p) return make_pair(-1, h * w);
	if(ini >= p && fim <= q) return seg2[k][pos];
	int mid = (ini + fim) >> 1, e = pos << 1, d = e | 1;
	pair<int, int> eVal = query2(e, ini, mid, p, q, k);
	pair<int, int> dVal = query2(d, mid + 1, fim, p, q, k);
	pair<int, int> ret;
	ret.first = max(eVal.first, dVal.first);
	ret.second = min(eVal.second, dVal.second);
	return ret;
}

Compilation message (stderr)

seats.cpp: In function 'void give_initial_chart(int, int, std::vector<int>, std::vector<int>)':
seats.cpp:33:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |  for(int i = 0; i < r.size(); i++){
      |                 ~~^~~~~~~~~~
seats.cpp:39:6: warning: unused variable 'maxi' [-Wunused-variable]
   39 |  int maxi = 0;
      |      ^~~~
seats.cpp: In function 'int swap_seats(int, int)':
seats.cpp:64:6: warning: unused variable 'maxi' [-Wunused-variable]
   64 |  int maxi = 0;
      |      ^~~~

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