#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(int v, int r, int c, int N) {
int result = 0, knew = 0, red = 0, want = 0;
if (v == 0) {
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==1) {
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 == 2) {
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;
int data[4];
int A, B, C, D;
dx = j, dy = i;
data[0] = dx * dy;
dx = N - j, dy = i;
data[1] = dx * dy;
dx = N - j, dy = N - i;
data[2] = dx * dy;
dx = j, dy = N - i;
data[3] = dx * dy;
A = data[0], B = data[1], C = data[2], D = data[3];
for (int i = 0; i < 4; i++)
{
for (int j = i + 1; j < 4; j++)
{
if (data[i] > data[j])
{
int temp;
temp = data[i];
data[i] = data[j];
data[j] = temp;
}
}
}
if (A == data[0])
{
ans[i][j] = f(0, i, j, N);
}
else if (B == data[0])
{
ans[i][j] = f(1, i, j, N);
}
else if (C == data[0])
{
ans[i][j] = f(2, i, j, N);
}
else
{
ans[i][j] = f(3, i, j, N);
}
}
}
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
if (ans[i][j]) Report(i, j);
}
}
}
Compilation message
treasure.cpp: In function 'void findTreasure(int)':
treasure.cpp:110:17: warning: variable 'D' set but not used [-Wunused-but-set-variable]
110 | int A, B, C, D;
| ^
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Partially correct |
1 ms |
364 KB |
Output is partially correct - N = 5, K = 449, score = 8 |
| 2 |
Partially correct |
1 ms |
364 KB |
Output is partially correct - N = 10, K = 8075, score = 1 |
| 3 |
Partially correct |
1 ms |
364 KB |
Output is partially correct - N = 15, K = 42869, score = 1 |
| 4 |
Partially correct |
1 ms |
364 KB |
Output is partially correct - N = 16, K = 55808, score = 1 |
| 5 |
Partially correct |
2 ms |
620 KB |
Output is partially correct - N = 55, K = 8335549, score = 1 |
| 6 |
Partially correct |
4 ms |
620 KB |
Output is partially correct - N = 66, K = 17366283, score = 1 |
| 7 |
Partially correct |
6 ms |
748 KB |
Output is partially correct - N = 77, K = 32283737, score = 1 |
| 8 |
Partially correct |
10 ms |
876 KB |
Output is partially correct - N = 88, K = 55214720, score = 1 |
| 9 |
Partially correct |
14 ms |
1004 KB |
Output is partially correct - N = 99, K = 88621205, score = 1 |
| 10 |
Partially correct |
15 ms |
1004 KB |
Output is partially correct - N = 100, K = 92270000, score = 1 |
