CSES - Datatähti 2024 alku - Results
Submission details
Task:Uolevin kalansaalis
Sender:Bean Benestrom
Submission time:2023-11-04 11:07:47 +0200
Language:C++17
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED37
#2ACCEPTED63
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2details
#2ACCEPTED0.00 s1, 2details
#3ACCEPTED0.00 s1, 2details
#4ACCEPTED0.00 s1, 2details
#5ACCEPTED0.00 s1, 2details
#6ACCEPTED0.00 s1, 2details
#7ACCEPTED0.00 s1, 2details
#8ACCEPTED0.00 s1, 2details
#9ACCEPTED0.00 s1, 2details
#10ACCEPTED0.00 s1, 2details
#11ACCEPTED0.00 s1, 2details
#12ACCEPTED0.00 s1, 2details
#13ACCEPTED0.00 s1, 2details
#14ACCEPTED0.00 s1, 2details
#15ACCEPTED0.00 s1, 2details
#16ACCEPTED0.54 s2details
#17ACCEPTED0.55 s2details
#18ACCEPTED0.60 s2details
#19ACCEPTED0.49 s2details
#20ACCEPTED0.54 s2details
#21ACCEPTED0.51 s2details
#22ACCEPTED0.51 s2details
#23ACCEPTED0.49 s2details
#24ACCEPTED0.52 s2details

Code

#include <iostream>
// #include <chrono>
// using namespace std::chrono;
 
#define Pike    -1
#define Shrimp  10
int MAP[500][500]{0};
int SELECTION[500]{0};
int WIDTH, HEIGHT;

bool line_sum(int x, int y, int x_direction, int y_direction, unsigned int depth) {    // False if went out of bounds
    if (y < 0 || y >= HEIGHT) return false;
    if (x < 0 || x >= WIDTH)  return false;
    SELECTION[y] += MAP[y][x];
    int _x = x_direction == 1 ? 1 : 0;
    while (depth) {
        y += y_direction;
        if ((y+_x) % 2) x += x_direction;
        if (y < 0 || y >= HEIGHT) return false;
        if (x < 0 || x >= WIDTH)  return false;
        SELECTION[y] += MAP[y][x];
        depth--;
    }
    return true;
}

// x_direction : step direction & opposite of cutting ange
// y_direction : step direction & direction of triangle
int edge_sum(int x, int y, int x_direction, int y_direction, bool row) {
    int largest_sum = -1;
    while (true)
    {
        for (int i=0; i<WIDTH; i++) {
            int sum = 0;
            if (line_sum(x+i*x_direction, y, -x_direction, y_direction, i)) {
                for (int j=0; j<i+1; j++) {                         //                                              y_direction = 1  y_direction = -1
                    sum += SELECTION[y+y_direction*(i-j)];          // -j*y_direction+y+i    y-i+j*y_direction      y+(i-j)          y-(i-j)
                    if (sum > largest_sum) largest_sum = sum;
                }      
            } else { break; }
        }
        std::fill(SELECTION, SELECTION+HEIGHT, 0);
        if (row) { x += x_direction; if (x < 1 || x >= WIDTH-1) break; }
        else     { y += y_direction; if (y < 0 || y >= HEIGHT) break; }
    }
    return largest_sum;
}


int main() {
    // auto start = high_resolution_clock::now();
    int TOTAL = 0;
    int K; 
    std::cin >> HEIGHT >> WIDTH >> K;
 
    int x, y;
    char _type;
    for (int i=0; i<K; i++) {
        std::cin >> y >> x >> _type;
        if (_type == 'K') {MAP[y-1][x-1] = Shrimp; TOTAL -= Shrimp;}
        else              {MAP[y-1][x-1] = Pike; TOTAL -= Pike;}
    }

    // ---

    int largest_sum = -1, sum;
    // Point up
    sum = edge_sum(1, HEIGHT-1, 1, -1, true);
    if (sum > largest_sum) largest_sum = sum;
    sum = edge_sum(0, HEIGHT-1, 1, -1, false);
    if (sum > largest_sum) largest_sum = sum;
    sum = edge_sum(WIDTH-1, HEIGHT-1, -1, -1, false);
    if (sum > largest_sum) largest_sum = sum;
    // Point down
    sum = edge_sum(1, 0, 1, 1, true);
    if (sum > largest_sum) largest_sum = sum;
    sum = edge_sum(0, 0, 1, 1, false);
    if (sum > largest_sum) largest_sum = sum;
    sum = edge_sum(WIDTH-1, 0, -1, 1, false);
    if (sum > largest_sum) largest_sum = sum;

    std::cout << TOTAL + largest_sum;

    // Timing
    // auto stop = high_resolution_clock::now();
    // auto duration = duration_cast<microseconds>(stop - start);
    // std::cout << '\n' << (double)(duration.count()) / 1000000 << '\n';
    
    return 0;
}

/*
500 500 13
1 1 K
5 1 K
2 2 H
4 2 H
5 2 H
2 6 K
3 3 H
3 4 K
3 5 H
3 6 K
4 4 K
2 5 H
5 5 H
*/

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

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

correct output
-16

user output
-16

Test 2

Group: 1, 2

Verdict: ACCEPTED

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

correct output
0

user output
0

Test 3

Group: 1, 2

Verdict: ACCEPTED

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

correct output
0

user output
0

Test 4

Group: 1, 2

Verdict: ACCEPTED

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

correct output
50

user output
50

Test 5

Group: 1, 2

Verdict: ACCEPTED

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

correct output
40

user output
40

Test 6

Group: 1, 2

Verdict: ACCEPTED

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

correct output
-15

user output
-15

Test 7

Group: 1, 2

Verdict: ACCEPTED

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

correct output
60

user output
60

Test 8

Group: 1, 2

Verdict: ACCEPTED

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

correct output
2

user output
2

Test 9

Group: 1, 2

Verdict: ACCEPTED

input
1 1 0

correct output
0

user output
0

Test 10

Group: 1, 2

Verdict: ACCEPTED

input
1 1 1
1 1 K

correct output
0

user output
0

Test 11

Group: 1, 2

Verdict: ACCEPTED

input
1 1 1
1 1 H

correct output
0

user output
0

Test 12

Group: 1, 2

Verdict: ACCEPTED

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

correct output
20

user output
20

Test 13

Group: 1, 2

Verdict: ACCEPTED

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

correct output
28

user output
28

Test 14

Group: 1, 2

Verdict: ACCEPTED

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

correct output
-439

user output
-439

Test 15

Group: 1, 2

Verdict: ACCEPTED

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

correct output
88

user output
88

Test 16

Group: 2

Verdict: ACCEPTED

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

correct output
-624270

user output
-624270

Test 17

Group: 2

Verdict: ACCEPTED

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

correct output
124020

user output
124020

Test 18

Group: 2

Verdict: ACCEPTED

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

correct output
249999

user output
249999

Test 19

Group: 2

Verdict: ACCEPTED

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

correct output
10126

user output
10126

Test 20

Group: 2

Verdict: ACCEPTED

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

correct output
-104692

user output
-104692

Test 21

Group: 2

Verdict: ACCEPTED

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

correct output
-335853

user output
-335853

Test 22

Group: 2

Verdict: ACCEPTED

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

correct output
-208567

user output
-208567

Test 23

Group: 2

Verdict: ACCEPTED

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

correct output
-57100

user output
-57100

Test 24

Group: 2

Verdict: ACCEPTED

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

correct output
-52800

user output
-52800