CSES - Datatähti 2024 alku - Results
Submission details
Task:Uolevin kalansaalis
Sender:Tipu
Submission time:2023-11-04 21:18:53 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.02 s1, 2details
#20.01 s1, 2details
#30.02 s1, 2details
#40.02 s1, 2details
#50.02 s1, 2details
#60.01 s1, 2details
#70.02 s1, 2details
#80.01 s1, 2details
#90.01 s1, 2details
#100.01 s1, 2details
#110.02 s1, 2details
#120.02 s1, 2details
#130.02 s1, 2details
#140.01 s1, 2details
#150.01 s1, 2details
#160.01 s2details
#170.01 s2details
#180.01 s2details
#190.02 s2details
#200.02 s2details
#210.01 s2details
#220.01 s2details
#230.01 s2details
#240.02 s2details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:261:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  261 |     freopen("input.txt", "r", stdin);
      |     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
struct Hex {
public:
int l = 0;
int r = 0;
int x = 0;
};
const int SIZE = 501;
int n, m, k;
int v[SIZE][SIZE];
Hex d[SIZE][SIZE];
unordered_map<int, pair<bool, bool>> sm[SIZE][SIZE];
int count_l_colum(int i, int j, int l){
bool add = (i % 2 == 0);
if(i-l < 0 || i >= n + 1 || j - int(ceil((l-add)/2.0)) < 0 || j >= m + 1){
return 0;
}
return (d[i][j].l - d[i-l][j - int(ceil((l-add)/2.0))].l);
}
int count_r_colum(int i, int j, int l){
bool add = (i % 2 == 0);
if(i-l < 0 || i >= n + 1 || j < 0 || j + int((l+add)/2.0) >= m + 1){
return 0;
}
return (d[i][j].r - d[i-l][j + int((l+add)/2.0)].r);
}
int count_x_colum(int i, int j, int l){
if(i < 0 || i >= n + 1 || j-l < 0 || j >= m + 1){
return 0;
}
return (d[i][j].x - d[i][j-l].x);
}
int scan(int p){
pair<short, short> a, b;
int safe_sum = 0;
for (int i = n; i >= 1; i--) {
for (int j = 1; j <= m; j++)
{
a = {i, j};
b = {i, j};
int s = count_x_colum(i, j, 1);
int l = 1;
bool up = true;
int safety = 0;
while(safety < 1e6){
safety++;
auto it = sm[a.first][a.second].find(l);
if(l == 1) up = true;
//tsekataan onko muistissa
if(it != sm[a.first][a.second].end()){
if((up && it->second.first) || (!up && it->second.second)){
break;
}
else{
if(up) it->second.first = true;
else it->second.second = true;
}
//lisätään muistikohta
} else{
if (up) sm[a.first][a.second].insert({l, {true, false}});
else sm[a.first][a.second].insert({l, {false, true}});
}
bool add = (b.first % 2 == 0);
if(l == 1){
int su1 = count_l_colum(b.first, b.second + 1, l + 1);
int su2 = count_x_colum(b.first + 1, b.second + add, l + 1);
int su3 = count_r_colum(a.first, a.second - 1, l + 1);
int su = min({su1, su2, su3});
int sd1 = count_l_colum(b.first + 1, b.second + (add - 1), l + 1);
int sd2 = count_x_colum(b.first - l, b.second + int((l+add)/2.0), l + 1);
int sd3 = count_r_colum(b.first + 1, b.second + add, l + 1);
int sd = min({sd1, sd2, sd3});
if(su <= sd && su < 0){
up = true;
l += 1;
if(su1 <= su2 && su1 <= su3){
s+=su1;
b = {b.first, b.second + 1};
continue;
}
if(su2 <= su3 && su2 <= su1){
s += su2;
b = {b.first + 1, b.second + add};
a = {a.first + 1, a.second + (add - 1)};
continue;
}
if(su3 <= su1 && su3 <= su2){
s += su3;
a = {a.first, a.second - 1};
continue;
}
}
else if (sd < 0){
up = false;
l += 1;
if(sd1 <= sd2 && sd1 <= sd3){
s+=sd1;
a = {a.first, a.second - 1};
b = {b.first + 1, b.second + (add - 1)};
continue;
}
if(sd2 <= sd3 && sd2 <= sd1){
s += sd2;
a = {a.first - 1, a.second + (add - 1)};
continue;
}
if(sd3 <= sd1 && sd3 <= sd2){
s += sd3;
b = {b.first + 1, b.second + add};
continue;
}
}
else{
break;
}
}
if(l != 0){
if(up){
int su1 = count_l_colum(b.first, b.second + 1, l + 1);
int su2 = count_x_colum(b.first + 1, b.second + add, l + 1);
int su3 = count_r_colum(a.first, a.second - 1, l + 1);
int su = min({su1, su2, su3});
int ru1 = -1 * count_l_colum(b.first, b.second, l);
int ru2 = -1 * count_x_colum(b.first, b.second, l);
int ru3 = -1 * count_r_colum(a.first, a.second , l);
int ru = min({ru1, ru2, ru3});
if(su <= ru && su < 0){
l += 1;
if(su1 <= su2 && su1 <= su3){
s+=su1;
b = {b.first, b.second + 1};
continue;
}
if(su2 <= su3 && su2 <= su1){
s += su2;
b = {b.first + 1, b.second + add};
a = {a.first + 1, a.second + (add - 1)};
continue;
}
if(su3 <= su1 && su3 <= su2){
s += su3;
a = {a.first, a.second - 1};
continue;
}
}
else if(ru < 0){
l -= 1;
if(ru1 <= ru2 && ru1 <= ru3){
s += ru1;
b = {b.first, b.second - 1};
continue;
}
if(ru2 <= ru3 && ru2 <= ru1){
s += ru2;
b = {b.first - 1, b.second + (add - 1)};
a = {a.first - 1, a.second + add};
continue;
}
if(ru3 <= ru1 && ru3 <= ru2){
s += ru3;
a = {a.first, a.second + 1};
continue;
}
}
else{
break;
}
}
else{
int sd1 = count_l_colum(b.first + 1, b.second + (add - 1), l + 1);
int sd2 = count_x_colum(b.first - l, b.second + int((l+add)/2.0), l + 1);
int sd3 = count_r_colum(b.first + 1, b.second + add, l + 1);
int sd = min({sd1, sd2, sd3});
int rd1 = -1 * count_l_colum(b.first, b.second, l);
int rd2 = -1 * count_x_colum(a.first, a.second + l - 1, l);
int rd3 = -1 * count_r_colum(b.first, b.second, l);
int rd = min({rd1, rd2, rd3});
if(sd <= rd && sd < 0){
l += 1;
if(sd1 <= sd2 && sd1 <= sd3){
s+=sd1;
a = {a.first, a.second - 1};
b = {b.first + 1, b.second + (add - 1)};
continue;
}
if(sd2 <= sd3 && sd2 <= sd1){
s += sd2;
a = {a.first - 1, b.second - int(ceil((l-add)/2.0))};
continue;
}
if(sd3 <= sd1 && sd3 <= sd2){
s += sd3;
b = {b.first + 1, b.second + add};
continue;
}
}
else if(rd < 0){
l -= 1;
if(rd1 <= rd2 && rd1 <= rd3){
s+=rd1;
a = {a.first, a.second + 1};
b = {b.first - 1, b.second + add};
continue;
}
if(rd2 <= rd3 && rd2 <= rd1){
s += rd2;
a = {a.first + 1, b.second - int((l-add)/2.0)};
continue;
}
if(rd3 <= rd1 && rd3 <= rd2){
s += sd3;
b = {b.first - 1, b.second + (add - 1)};
continue;
}
}
else{
break;
}
}
}
}
p = min(s, p);
// cout << "( " << i << ";" << j << ") s:" << s << " p:" << p << " safety: " << safety << endl;
safe_sum += safety;
}
}
// cout << "ss: " << safe_sum << endl;
return p;
}
int main() {
freopen("input.txt", "r", stdin);
cin >> n >> m >> k;
int a, b;
char c;
for (int i = 0; i < k; i++) {
cin >> a >> b >> c;
if(c == 'H'){
v[a][b] = 1;
}else{
v[a][b] = -10;
}
}
int os = 0;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
bool add = 0;
if(i % 2 == 0){
add = 1;
}
os += v[i][j];
d[i][j].l = v[i][j] + d[i-1][j-1+add].l;
d[i][j].r = v[i][j] + d[i-1][j+add].r;
d[i][j].x = v[i][j] + d[i][j-1].x;
}
}
// for (int i = 1; i <= n; i++) {
// for (int j = 1; j <= m; j++) {
// cout << "(" << d[i][j].l << " ; " << d[i][j].r << " ; " << d[i][j].x << ") " ;
// }
// cout << "\n";
// }
int p = 1e6;
p = scan(p);
// cout << p << endl;
// cout << os << endl;
// assert(p >= os);
cout << os - p;
}

Test details

Test 1

Group: 1, 2

Verdict:

input
5 6 13
1 1 K
5 1 K
2 2 H
4 2 H
...

correct output
-16

user output
-1000000

Test 2

Group: 1, 2

Verdict:

input
5 6 7
1 5 K
4 6 K
2 4 H
2 5 H
...

correct output
0

user output
-1000000

Test 3

Group: 1, 2

Verdict:

input
5 6 7
5 5 K
2 6 K
2 4 H
2 5 H
...

correct output
0

user output
-1000000

Test 4

Group: 1, 2

Verdict:

input
10 10 51
3 3 H
6 3 H
9 5 H
5 10 H
...

correct output
50

user output
-1000000

Test 5

Group: 1, 2

Verdict:

input
10 10 52
3 5 H
3 1 H
9 6 H
2 8 H
...

correct output
40

user output
-1000000

Test 6

Group: 1, 2

Verdict:

input
10 10 60
6 10 H
2 8 H
5 8 H
8 10 H
...

correct output
-15

user output
-1000000

Test 7

Group: 1, 2

Verdict:

input
10 10 60
4 7 H
7 4 H
4 10 H
3 6 H
...

correct output
60

user output
-1000000

Test 8

Group: 1, 2

Verdict:

input
10 10 40
9 9 H
5 10 H
5 6 H
4 9 H
...

correct output
2

user output
-1000000

Test 9

Group: 1, 2

Verdict:

input
1 1 0

correct output
0

user output
-1000000

Test 10

Group: 1, 2

Verdict:

input
1 1 1
1 1 K

correct output
0

user output
-1000000

Test 11

Group: 1, 2

Verdict:

input
1 1 1
1 1 H

correct output
0

user output
-1000000

Test 12

Group: 1, 2

Verdict:

input
10 5 32
10 3 H
4 4 H
3 3 H
5 4 H
...

correct output
20

user output
-1000000

Test 13

Group: 1, 2

Verdict:

input
5 10 32
5 9 H
2 4 H
2 9 H
2 5 H
...

correct output
28

user output
-1000000

Test 14

Group: 1, 2

Verdict:

input
10 10 100
2 9 H
5 4 H
5 9 K
6 1 K
...

correct output
-439

user output
-1000000

Test 15

Group: 1, 2

Verdict:

input
10 10 100
8 9 H
5 10 H
5 4 H
3 9 H
...

correct output
88

user output
-1000000

Test 16

Group: 2

Verdict:

input
500 500 125000
125 261 K
84 78 K
11 200 K
481 246 K
...

correct output
-624270

user output
-1000000

Test 17

Group: 2

Verdict:

input
500 500 125100
16 61 H
37 62 H
459 125 H
318 476 H
...

correct output
124020

user output
-1000000

Test 18

Group: 2

Verdict:

input
500 500 249999
22 214 H
356 145 H
341 29 H
393 262 H
...

correct output
249999

user output
-1000000

Test 19

Group: 2

Verdict:

input
500 500 32000
30 81 H
315 34 H
78 112 H
367 166 H
...

correct output
10126

user output
-1000000

Test 20

Group: 2

Verdict:

input
500 500 126745
164 390 H
126 331 H
164 126 H
55 92 H
...

correct output
-104692

user output
-1000000

Test 21

Group: 2

Verdict:

input
500 500 71200
106 191 H
314 189 H
482 485 H
344 401 H
...

correct output
-335853

user output
-1000000

Test 22

Group: 2

Verdict:

input
500 500 67772
421 277 H
428 470 H
169 142 H
256 345 H
...

correct output
-208567

user output
-1000000

Test 23

Group: 2

Verdict:

input
500 500 27434
366 481 H
38 22 H
126 107 H
135 169 H
...

correct output
-57100

user output
-1000000

Test 24

Group: 2

Verdict:

input
500 500 93982
183 13 H
463 230 H
264 351 H
399 290 H
...

correct output
-52800

user output
-1000000