답안 #321295

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
321295 2020-11-11T23:55:48 Z monus1042 보물 찾기 (CEOI13_treasure2) C++17
100 / 100
42 ms 1016 KB
#include "treasure.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define ii pair<int,int>

int ans[105][105];
int HOR[105], VER[105];
int area, included, line, ri, le;

map< tuple<int, int, int, int>, int > dp;

int askit(int r1, int c1, int r2, int c2){
	tuple<int,int,int,int> tup(r1, c1, r2, c2);
	if (dp.find(tup) != dp.end() ) return dp[tup];
	return dp[tup] = countTreasure(r1, c1, r2, c2);
}

int f(vector< pair<int,int> > v, int r, int c, int N){
	int result = 0, knew = 0, red = 0, want = 0;
	if (v[3].second == 'A'){
		for (int i=1; i<=r-1; i++){
			for (int j=1; j<=c; j++){
				knew += ans[i][j];
			}
		}
		red = askit(1, 1, r, c);
		want = red - knew;

		knew = 0;
		for (int j=1; j<=c - 1; j++) knew += ans[r][j];
		result = want - knew;
	}else if (v[3].second == 'B'){
		for (int i=1; i<=r-1; i++){
			for (int j=c+1; j<=N; j++){
				knew += ans[i][j];
			}
		}
		if (c+1 <= N) red = askit(1, c+1, r, N);
		want = red - knew;

	    result = included - want;
	}else if (v[3].second == 'C'){
		for (int i=1; i<=r-1; i++){
			for (int j=c+1; j<=N; j++){
				knew += ans[i][j];
			}
		}
		knew = ri - knew;
		if (c+1 <= N && r+1 <= N) red = askit(r+1, c+1, N, N);

		want = knew - red;
		result = included - want;
	}else{ // D
		for (int i=1; i<=r-1; i++){
			for (int j=1; j<=c; j++){
				knew += ans[i][j];
			}
		}
		knew = le - knew;
		if (r+1 <= N) red = askit(r+1, 1, N, c);

		want = knew - red;
		knew = 0;
		for (int j=1; j<=c - 1; j++) knew += ans[r][j];
		result = want - knew;
	}
	return result;
}

void findTreasure (int N) {
    //int cnt = countTreasure(1, 1, N, N);
    //if(cnt > 0) Report (1, 1);
	area = askit(1, 1, N, N);
	for (int i=1; i<=N - 1; i++){
		if ( i > (N-i) ) VER[i] = askit(1, 1, i, N);
		else VER[i] = area - askit(i+1, 1, N, N);
	}
	for (int i=1; i<=N - 1; i++){
		if ( i > (N-i) ) HOR[i] = askit(1, 1, N, i);
		else HOR[i] = area - askit(1, i+1, N, N);
	}
	VER[N] = HOR[N] = area;
	
	for (int i=1; i<=N; i++){
		for (int j=1; j<=N; j++){
			int below = VER[N] - VER[i];
			int sup = VER[i-1];
			line = area - sup - below;
			
			included = line;
			for (int k=1; k<j; k++) included -= ans[i][k];
			
			ri = HOR[N] - HOR[j];
			le = HOR[j];

			vector< pair<int, int> > srt;
			int dx, dy;
			dx = j, dy = i;
			srt.pb(ii(dx*dy, 'A'));
			dx = N - j, dy = i;
			srt.pb(ii(dx*dy, 'B'));
			dx = N - j, dy = N - i;
			srt.pb(ii(dx*dy, 'C'));
			dx = j, dy = N - i;
			srt.pb(ii(dx*dy, 'D'));

			sort(srt.begin(), srt.end());
		    ans[i][j] = f(srt, i, j, N);
		}
	}

	for (int i=1; i<=N; i++){
		for (int j=1; j<=N; j++){
			if (ans[i][j]) Report(i,j);
		}
	}
}
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 492 KB Output is correct - N = 5, K = 289, score = 10
2 Correct 1 ms 364 KB Output is correct - N = 10, K = 4475, score = 10
3 Correct 1 ms 364 KB Output is correct - N = 15, K = 22289, score = 10
4 Correct 1 ms 364 KB Output is correct - N = 16, K = 28928, score = 10
5 Correct 5 ms 620 KB Output is correct - N = 55, K = 4005289, score = 10
6 Correct 9 ms 620 KB Output is correct - N = 66, K = 8305803, score = 10
7 Correct 14 ms 748 KB Output is correct - N = 77, K = 15383161, score = 10
8 Correct 21 ms 876 KB Output is correct - N = 88, K = 26244416, score = 10
9 Correct 42 ms 1016 KB Output is correct - N = 99, K = 42032201, score = 10
10 Correct 33 ms 1016 KB Output is correct - N = 100, K = 43760000, score = 10