CSES - Datatähti 2024 alku - Results
Submission details
Task:Uolevin kalansaalis
Sender:Mariia
Submission time:2023-11-12 08:45:34 +0200
Language:C++ (C++11)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED37
#2ACCEPTED63
Test results
testverdicttimegroup
#1ACCEPTED0.36 s1, 2details
#2ACCEPTED0.36 s1, 2details
#3ACCEPTED0.35 s1, 2details
#4ACCEPTED0.35 s1, 2details
#5ACCEPTED0.35 s1, 2details
#6ACCEPTED0.35 s1, 2details
#7ACCEPTED0.35 s1, 2details
#8ACCEPTED0.36 s1, 2details
#9ACCEPTED0.36 s1, 2details
#10ACCEPTED0.35 s1, 2details
#11ACCEPTED0.37 s1, 2details
#12ACCEPTED0.35 s1, 2details
#13ACCEPTED0.35 s1, 2details
#14ACCEPTED0.35 s1, 2details
#15ACCEPTED0.36 s1, 2details
#16ACCEPTED0.87 s2details
#17ACCEPTED0.87 s2details
#18ACCEPTED0.93 s2details
#19ACCEPTED0.81 s2details
#20ACCEPTED0.89 s2details
#21ACCEPTED0.83 s2details
#22ACCEPTED0.84 s2details
#23ACCEPTED0.81 s2details
#24ACCEPTED0.84 s2details

Code

#include <iostream>
#include <vector>

using namespace std;

int main() {
    int n, m, k;
    cin >> n >> m >> k;

    //vector<vector<vector<int>>> dp1(min(n, m) + 1, vector<vector<int>>(n, vector<int>(m)));
    int ***dp1;
    dp1 = new int** [501];
    for (int i = 0; i < 501; i++) {
        dp1[i] = new int* [501];
        for (int j = 0; j < 501; j++) {
            dp1[i][j] = new int [501];
        }
    }
    //vector<vector<int>> t(500, vector<int> (500, 0));
    int **t;
    t = new int* [501];
    for (int i = 0; i < 501; i++) {
        t[i] = new int [501];
    }

    int SUM = 0;
    for (int i = 0; i < k; i++) {
        int x, y; char type;
        cin >> y >> x >> type;
        y--; x--;
        if (type == 'H') {
            t[y][x] = 1;
        } else {
            t[y][x] = -10;
        }
        SUM += t[y][x];
    }

    int mini = 1e9;
    for (int y = 0; y < n; y++) {
        for (int x = 0; x < m; x++) {
            dp1[1][y][x] = t[y][x];
            if (dp1[1][y][x] < mini) {
                mini = dp1[1][y][x];
            }
        }
    }

    //vector<vector<int>> pr(500, vector<int> (500, 0));
    int **pr;
    pr = new int* [501];
    for (int i = 0; i < 501; i++) {
        pr[i] = new int [501];
    }

    for (int j = 0; j < n; j++) {
        pr[j][0] = t[j][0];
        for (int i = 1; i < m; i++) {
            pr[j][i] = pr[j][i - 1] + t[j][i];
        }
    }

    for (int s = 2; s <= min(n, m); s++) {
        for (int y = 0; y + s - 1 < n; y++) {
            for (int x = 0; x + s - 1 < m; x++) {
                if (y + 1 >= n) {
                    dp1[s][y][x] = 1e9;
                    continue;
                }
                if (x + y % 2 >= m) {
                    dp1[s][y][x] = 1e9;
                    continue;
                }
                if (dp1[s - 1][y + 1][x + y % 2] == 1e9) {
                    dp1[s][y][x] = 1e9;
                    continue;
                }
                int sum = 0;
                if (x == 0) {
                    sum = pr[y][x + s - 1];
                } else {
                    sum = pr[y][x + s - 1] - pr[y][x - 1];
                }
                dp1[s][y][x] = dp1[s - 1][y + 1][x + y % 2] + sum;

                //cout << dp1[s][y][x] << '\n';
                if (dp1[s][y][x] < mini) {
                    mini = dp1[s][y][x];
                }
            }
            //cout << '\n';
        }
    }

    for (int s = 2; s <= min(n, m); s++) {
        for (int y = 0; y < n; y++) {
            for (int x = 0; x < m; x++) {
                dp1[s][y][x] = 0;
            }
        }
    }

    for (int y = 0; y < n; y++) {
        for (int x = 0; x < m; x++) {
            dp1[1][y][x] = t[y][x];
            if (dp1[1][y][x] < mini) {
                mini = dp1[1][y][x];
            }
        }
    }

    for (int s = 2; s <= min(n, m); s++) {
        for (int y = n - 1; y - s + 1 >= 0; y--) {
            for (int x = 0; x + s - 1 < m; x++) {
                if (y - 1 < 0) {
                    dp1[s][y][x] = 1e9;
                    continue;
                }
                if (x + y % 2 >= m) {
                    dp1[s][y][x] = 1e9;
                    continue;
                }
                if (dp1[s - 1][y - 1][x + y % 2] == 1e9) {
                    dp1[s][y][x] = 1e9;
                    continue;
                }
                int sum = 0;
                if (x == 0) {
                    sum = pr[y][x + s - 1];
                } else {
                    sum = pr[y][x + s - 1] - pr[y][x - 1];
                }
                dp1[s][y][x] = dp1[s - 1][y - 1][x + y % 2] + sum;

                if (dp1[s][y][x] < mini) {
                    mini = dp1[s][y][x];
                }
            }
        }
    }

    if (n == 1 && m == 1) {
        cout << 0;
    } else {
        cout << SUM - mini;
    }

    for (int i = 0; i < 500; i++) {
        for (int j = 0; j < 500; j++) {
            delete [] dp1[i][j];
        }
        delete [] dp1[i];
    }
    delete [] dp1;

    for (int i = 0; i < 500; i++) {
        delete [] t[i];
    }
    delete [] t;

    for (int i = 0; i < 500; i++) {
        delete [] pr[i];
    }
    delete [] pr;
}

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