제출 #126639

#제출 시각아이디문제언어결과실행 시간메모리
126639eriksuenderhauf보물 찾기 (CEOI13_treasure2)C++11
80 / 100
16 ms1272 KiB
//#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#include "treasure.h"
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
#define mem(a,v) memset((a), (v), sizeof (a))
#define enl printf("\n")
#define case(t) printf("Case #%d: ", (t))
#define ni(n) scanf("%d", &(n))
#define nl(n) scanf("%lld", &(n))
#define nai(a, n) for (int i = 0; i < (n); i++) ni(a[i])
#define nal(a, n) for (int i = 0; i < (n); i++) nl(a[i])
#define pri(n) printf("%d\n", (n))
#define prl(n) printf("%lld\n", (n))
#define pii pair<int, int>
#define pil pair<int, long long>
#define pll pair<long long, long long>
#define vii vector<pii>
#define vil vector<pil>
#define vll vector<pll>
#define vi vector<int>
#define vl vector<long long>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef cc_hash_table<int,int,hash<int>> ht;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> oset;
const double pi = acos(-1);
const int MOD = 1e9 + 7;
const int INF = 1e9 + 7;
const int MAXN = 1e2 + 5;
const double eps = 1e-9;
int a[MAXN][MAXN], n, t[MAXN][MAXN];
map<pair<pii,pii>,int> dp;

int qry(int x1, int y1, int x2, int y2) {
	if (dp.count(mp(mp(x1,y1),mp(x2,y2))))
		return dp[mp(mp(x1,y1),mp(x2,y2))];
	int ret = countTreasure(x1, y1, x2, y2);
	dp[mp(mp(x1,y1),mp(x2,y2))] = ret;
	return ret;
}

void g(int i, int j) {
	int tl = i * j + (i-1) * j + i * (j-1) + (i-1)*(j-1);
	int bl = (n-i+1)*j+(n-i)*j+(n-i+1)*(j-1)+(n-i)*(j-1);
	int tr = i*(n-j+1)+(i-1)*(n-j+1)+i*(n-j)+(i-1)*(n-j);
	int br = (n-i+1)*(n-j+1)+(n-i)*(n-j+1)+(n-i+1)*(n-j)+(n-i)*(n-j);
	int mx = max(max(tl,bl), max(tr, br));
	if (mx == tl)
		t[i][j] = 1;
	else if (mx == bl)
		t[i][j] = 2;
	else if (mx == tr)
		t[i][j] = 3;
	else 
		t[i][j] = 4;
}

int f(int i, int j) {
	if (t[i][j] == 1)
		return qry(1,1,i,j)-qry(1,1,i-1,j)-qry(1,1,i,j-1)+qry(1,1,i-1,j-1);
	if (t[i][j] == 2)
		return qry(i,1,n,j)-qry(i+1,1,n,j)-qry(i,1,n,j-1)+qry(i+1,1,n,j-1);
	if (t[i][j] == 3)
		return qry(1,j,i,n)-qry(1,j,i-1,n)-qry(1,j+1,i,n)+qry(1,j+1,i-1,n);
	return qry(i,j,n,n)-qry(i+1,j,n,n)-qry(i,j+1,n,n)+qry(i+1,j+1,n,n);
}

vii tmp[5];
pair<pii,pii> pos[5];
int sz[5];

void findTreasure(int N) {
	mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
	mem(a, -1);
	n = N;
	for (int l = 1; l <= 4; l++)
		pos[l].fi = mp(n+1,n+1);
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++) {
			g(i, j);
			tmp[t[i][j]].pb(mp(i,j));
			pos[t[i][j]].fi.fi = min(pos[t[i][j]].fi.fi, i);
			pos[t[i][j]].fi.se = min(pos[t[i][j]].fi.se, j);
			pos[t[i][j]].se.fi = max(pos[t[i][j]].se.fi, i);
			pos[t[i][j]].se.se = max(pos[t[i][j]].se.se, j);
			sz[t[i][j]]++;
		}
	for (int l = 1; l <= 4; l++) {
		int cnt = qry(pos[l].fi.fi, pos[l].fi.se, pos[l].se.fi, pos[l].se.se), val = 0;
		shuffle(tmp[l].begin(), tmp[l].end(), rng);
		int fl = 1;
		if (cnt * 2 > sz[l]) {
			fl = 0;
			cnt = sz[l] - cnt;
		}
		while (!tmp[l].empty() && val < cnt) {
			pii cur = tmp[l].back(); tmp[l].pop_back();
			a[cur.fi][cur.se] = f(cur.fi, cur.se);
			if (a[cur.fi][cur.se] == fl)
				val++;
		}
		for (int i = pos[l].fi.fi; i <= pos[l].se.fi; i++)
			for (int j = pos[l].fi.se; j <= pos[l].se.se; j++)
				if (a[i][j] == -1)
					a[i][j] = 1 - fl;
	}
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++)
			if (a[i][j])
				Report(i, j);
}
#Verdict Execution timeMemoryGrader output
Fetching results...