Submission #126731

#TimeUsernameProblemLanguageResultExecution timeMemory
126731eriksuenderhauf보물 찾기 (CEOI13_treasure2)C++11
90 / 100
12 ms1144 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], sm[MAXN][MAXN], sm2[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 && mx != max(bl, max(tr, br)))
		t[i][j] = 1;
	else if (mx == bl && mx != max(tl, max(tr, br)))
		t[i][j] = 2;
	else if (mx == tr && mx != max(max(tl,bl), br))
		t[i][j] = 3;
	else if (mx == br && mx != max(max(tl,bl), tr))
		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);
	if (t[i][j] == 4)
		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);
	return 0;
}
 
void findTreasure(int N) {
	n = N;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++)
			g(i, j);
	for (int i = n; i > 0; i--)
		for (int j = n; j > 0; j--)
			if (t[i][j] == 4)
				a[i][j] = f(i, j);
	for (int i = 1; i <= n; i++)
		for (int j = n; j > 0; j--)
			if (t[i][j] == 3)
				a[i][j] = f(i, j);
	for (int i = n; i > 0; i--)
		for (int j = 1; j <= n; j++)
			if (t[i][j] == 2) 
				a[i][j] = f(i, j);
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= n; j++)
			if (t[i][j] == 1)
				a[i][j] = f(i, j);
	for (int i = n; i > 0; i--)
		for (int j = 1; j <= n; j++)
			sm[i][j] = sm[i+1][j] + a[i][j];
	for (int i = n; i > 0; i--)
		for (int j = n; j > 0; j--)
			sm2[i][j] = sm2[i][j+1] + a[i][j];

	if (n & 1) {
		int idx = (n+1) / 2;
		for (int i = 1; i <= n; i++) {
			if (i == idx) continue;
			if (i < idx) {
				a[idx][i] = qry(idx, i, n, n) - qry(idx, i+1, n, n) - sm[idx][i];
				a[i][idx] = qry(i, idx, n, n) - qry(i+1, idx, n, n) - sm2[i][idx];
			} else {
				a[idx][i] = qry(idx, 1, n, i) - qry(idx, 1, n, i-1) - sm[idx][i];
				a[i][idx] = qry(1, idx, i, n) - qry(1, idx, i-1, n) - sm2[i][idx];
			}
		}
		a[idx][idx] = qry(1,1,idx,idx);
		for (int i = 1; i <= idx; i++)
			for (int j = 1; j <= idx; j++)
				if (!(i == idx && j == idx))
					a[idx][idx] -= a[i][j];
	}
	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...