CSES - Datatähti Open 2021 - Results
Submission details
Task:Polygonal Chain
Sender:voventa
Submission time:2021-02-12 18:29:25 +0200
Language:C++17
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#10.01 s1, 2, 4, 5details
#20.01 s1, 2, 4, 5details
#30.01 s1, 2, 4, 5details
#40.01 s1, 2, 4, 5details
#50.01 s1, 2, 4, 5details
#60.01 s2, 4, 5details
#70.01 s2, 4, 5details
#80.01 s2, 4, 5details
#90.01 s2, 4, 5details
#100.01 s2, 4, 5details
#110.01 s4, 5details
#12--5details
#130.01 s3, 4, 5details
#140.01 s3, 4, 5details
#150.01 s1, 2, 4, 5details
#160.01 s4, 5details
#170.01 s4, 5details
#18--5details
#190.01 s1, 2, 4, 5details
#200.01 s2, 4, 5details
#210.01 s2, 4, 5details
#220.01 s2, 4, 5details
#230.01 s4, 5details
#240.01 s4, 5details
#250.01 s4, 5details
#260.14 s5details
#270.16 s5details
#280.17 s5details
#290.16 s5details
#300.01 s1, 2, 4, 5details

Code

#include <bits/stdc++.h>
#define X first
#define Y second
#define sz(a) (int)a.size()
#define pb push_back
//#define int long long

using namespace std;
typedef long long ll;

void solve();

signed main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    int t = 1;
    //cin >> t;
    while (t--)
        solve();
    return 0;
}

int a[100010];

pair <int, int> xx[100010];

string s;

vector <vector <int>> v;

vector <int> ans, ans2;

void solve() {
    int n, l = 0, r;
    cin >> n;
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
        if (i == 0) {
            r = a[i];
        } else {
            l = r;
            if (i % 2 == 1)
                r = l - a[i];
            else
                r = l + a[i];
        }
        xx[i] = {l, r};
    }
    for (int i = 0; i < n; ++i) {
        cout << xx[i].X << " " << xx[i].Y << endl;
    }
    cin >> s;
    int A, B, C;
    for (int i = 0; i < n; ++i) {
        v.pb({i});
        while (sz(v) > 2) {
            char c1 = s[v[sz(v) - 3].back()], c2 = s[v[sz(v) - 2].back()];
            A = abs(xx[v[sz(v) - 3][0]].X - xx[v[sz(v) - 3].back()].Y);
            B = abs(xx[v[sz(v) - 2][0]].X - xx[v[sz(v) - 2].back()].Y);
            C = abs(xx[v[sz(v) - 1][0]].X - xx[v[sz(v) - 1].back()].Y);
            if ((A < B && B < C) || (A < B && B > C) || (A > B && B > C) || (A == B && B > C) || (A < B && B == C)) {
                break;
            }
            if ((A == B && B == C) || (A > B && B == C) ) {
                if (c1 != c2) {
                    cout << "NO";
                    return;
                } else {
                    if (c1 == s[i] || i == n - 1) {
                        for (auto i : v[sz(v) - 2]) {
                            v[sz(v) - 3].pb(i);
                        }
                        for (auto i : v[sz(v) - 1]) {
                            v[sz(v) - 3].pb(i);
                        }
                        v.pop_back();
                        v.pop_back();
                        continue;
                    } else
                        break;
                }
            }
            if (c1 != c2) {
                cout << "NO";
                return;
            }
            for (auto i : v[sz(v) - 2]) {
                v[sz(v) - 3].pb(i);
            }
            for (auto i : v[sz(v) - 1]) {
                v[sz(v) - 3].pb(i);
            }
            v.pop_back();
            v.pop_back();
        }
    }
    int t = sz(v);
    int x = 0, y = 0;
    for (int i = 0; i < sz(v); i++) {
        int k = 1;
        for (int j = 0; j < sz(v[i]) - 1; ++j) {
            if (j != sz(v[i]) - 2 && s[x + j] != s[x + j + 1]) {
                ans.pb(k);
            } else
                ans.pb(1);
            k++;
        }
        if (i < sz(v) - 1 && abs(xx[v[i][0]].X - xx[v[i].back()].Y) >= abs(xx[v[i + 1][0]].X - xx[v[i + 1].back()].Y)) {
            t = i + 1;
            ans.pb(1000000);
            break;
        }
        x += sz(v[i]);
        y += sz(v[i]);
        if (sz(v) == i + 1)
            break;
        if (i == 0 || s[x - 1] == s[x - 1 - sz(v[i])]) {
            ans.pb(1);
            y++;
        } else {
            ans.pb(y + 2);
            y += 2;
        }
    }
    x = 0;
    y = 0;
    for (int i = sz(v) - 1; i >= t; --i) {
        int k = 1;
        for (int j = sz(v[i]) - 1; j > 0; --j) {
            if (j != 1 && s[n - 2 - x - j] != s[n - 2 - x - j - 1]) {
                ans.pb(k);
            } else
                ans.pb(1);
            k++;
        }
        x += sz(v[i]);
        y += sz(v[i]);
        if (t == i)
            break;
        if (s[n - 2 - x] == s[n - 2 - x + sz(v[i])])
            ans2.pb(1);
        else {
            ans2.pb(y + 2);
            y++;
        }
        y++;
    }
    reverse(ans2.begin(), ans2.end());
    cout << "YES\n";
    for (auto i : ans) {
        cout << i << " ";
    }
    for (auto i : ans2) {
        cout << i << " ";
    }
    return;
}

Test details

Test 1

Group: 1, 2, 4, 5

Verdict:

input
2
2 10
D

correct output
YES

user output
0 2
2 -8
YES

Test 2

Group: 1, 2, 4, 5

Verdict:

input
8
5 8 7 5 6 5 3 4
DUUUDDD

correct output
YES
1 5 1 1 3 1 1 

user output
0 5
5 -3
-3 4
4 -1
-1 5
...

Test 3

Group: 1, 2, 4, 5

Verdict:

input
8
9 8 8 10 10 8 9 10
DDDUUUD

correct output
YES
1 1 1 4 1 1 7 

user output
0 9
9 1
1 9
9 -1
-1 9
...

Test 4

Group: 1, 2, 4, 5

Verdict:

input
8
9 10 8 8 9 9 7 8
DDDDUUU

correct output
NO

user output
0 9
9 -1
-1 7
7 -1
-1 8
...

Test 5

Group: 1, 2, 4, 5

Verdict:

input
8
10 2 8 3 10 2 10 10
DDUUUUD

correct output
YES
1 1 3 1 1 1 7 

user output
0 10
10 8
8 16
16 13
13 23
...

Test 6

Group: 2, 4, 5

Verdict:

input
15
73 74 97 82 19 50 26 51 56 93 ...

correct output
YES
1 2 3 1 1 3 1 1 1 10 1 3 1 1 

user output
0 73
73 -1
-1 96
96 14
14 33
...

Test 7

Group: 2, 4, 5

Verdict:

input
15
95 71 97 77 98 76 100 62 96 69...

correct output
YES
1 1 3 1 1 1 1 1 9 1 11 1 13 1 

user output
0 95
95 24
24 121
121 44
44 142
...

Test 8

Group: 2, 4, 5

Verdict:

input
15
79 81 84 86 88 90 92 92 91 89 ...

correct output
YES
1 2 3 4 5 6 14 1 6 5 4 3 2 1 

user output
0 79
79 -2
-2 82
82 -4
-4 84
...

Test 9

Group: 2, 4, 5

Verdict:

input
15
97 90 87 83 79 76 74 23 24 76 ...

correct output
YES
13 11 9 7 5 3 1 1 3 5 7 9 11 1...

user output
0 97
97 7
7 94
94 11
11 90
...

Test 10

Group: 2, 4, 5

Verdict:

input
15
100 2 99 1 78 4 93 2 100 1 15 ...

correct output
NO

user output
0 100
100 98
98 197
197 196
196 274
...

Test 11

Group: 4, 5

Verdict:

input
1000
999997 999995 999993 999991 99...

correct output
YES
997 995 993 991 989 987 985 98...

user output
0 999997
999997 2
2 999995
999995 4
4 999993
...

Test 12

Group: 5

Verdict:

input
100000
999999998 999999996 999999994 ...

correct output
YES
99997 99995 99993 99991 99989 ...

user output
(empty)

Test 13

Group: 3, 4, 5

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
YES
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
0 1
1 0
0 1
1 0
0 1
...

Test 14

Group: 3, 4, 5

Verdict:

input
1000
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

correct output
NO

user output
0 1
1 0
0 1
1 0
0 1
...

Test 15

Group: 1, 2, 4, 5

Verdict:

input
5
6 7 7 6 6
UDUU

correct output
YES
1 4 1 1 

user output
0 6
6 -1
-1 6
6 0
0 6
...

Test 16

Group: 4, 5

Verdict:

input
30
15 12 9 88 10 26 78 23 67 14 9...

correct output
YES
1 1 1 4 1 3 1 1 7 1 19 1 1 3 1...

user output
0 15
15 3
3 12
12 -76
-76 -66
...

Test 17

Group: 4, 5

Verdict:

input
1000
1000000 1 146324 146324 289287...

correct output
YES
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
0 1000000
1000000 999999
999999 1146323
1146323 999999
999999 1289286
...

Test 18

Group: 5

Verdict:

input
100000
1000000000 1 421262579 4212625...

correct output
YES
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
(empty)

Test 19

Group: 1, 2, 4, 5

Verdict:

input
8
1 3 1 2 5 1 1 2
DUUUDUU

correct output
NO

user output
0 1
1 -2
-2 -1
-1 -3
-3 2
...

Test 20

Group: 2, 4, 5

Verdict:

input
15
3 1 33 13 1 11 32 8 1 19 15 25...

correct output
YES
1 1 5 1 1 3 5 1 1 1 1 5 2 1 

user output
0 3
3 2
2 35
35 22
22 23
...

Test 21

Group: 2, 4, 5

Verdict:

input
15
10 2 39 41 42 34 31 28 26 24 2...

correct output
YES
1 1 1 1 10 9 8 7 6 1 4 1 1 1 

user output
0 10
10 8
8 47
47 6
6 48
...

Test 22

Group: 2, 4, 5

Verdict:

input
15
27 4 6 23 26 37 40 38 44 27 3 ...

correct output
YES
1 1 1 1 5 1 7 1 3 1 1 3 1 1 

user output
0 27
27 23
23 29
29 6
6 32
...

Test 23

Group: 4, 5

Verdict:

input
1000
3246 3562 197273 197429 197755...

correct output
YES
1 1 1 4 5 10 3 1 1 3 7 12 1 1 ...

user output
0 3246
3246 -316
-316 196957
196957 -472
-472 197283
...

Test 24

Group: 4, 5

Verdict:

input
1000
503981 503487 503350 502673 50...

correct output
YES
999 1 997 1 1 994 989 1 1 1 1 ...

user output
0 503981
503981 494
494 503844
503844 1171
1171 503792
...

Test 25

Group: 4, 5

Verdict:

input
1000
1445 1363 1749 1084 262408 263...

correct output
NO

user output
0 1445
1445 82
82 1831
1831 747
747 263155
...

Test 26

Group: 5

Verdict:

input
100000
209655 9167 9389 191291 198294...

correct output
NO

user output
0 209655
209655 200488
200488 209877
209877 18586
18586 216880
...

Test 27

Group: 5

Verdict:

input
100000
16295 14904 5103 13337 26939 3...

correct output
YES
1 1 1 1 5 6 1 1 1 10 11 1 13 1...

user output
0 16295
16295 1391
1391 6494
6494 -6843
-6843 20096
...

Test 28

Group: 5

Verdict:

input
100000
1859 174288 15040 4631 4993844...

correct output
YES
1 3 1 1 99997 99992 1 1 5 1 1 ...

user output
0 1859
1859 -172429
-172429 -157389
-157389 -162020
-162020 499222419
...

Test 29

Group: 5

Verdict:

input
100000
959817 958289 966165 922369 92...

correct output
YES
1 1 1 1 1 1 1 1 1 1 11 14 1 1 ...

user output
0 959817
959817 1528
1528 967693
967693 45324
45324 972854
...

Test 30

Group: 1, 2, 4, 5

Verdict:

input
8
2 3 2 3 5 6 7 8
UDDDUDU

correct output
YES
1 2 1 1 5 6 7 

user output
0 2
2 -1
-1 1
1 -2
-2 3
...