CSES - Datatähti Open 2021 - Results
Submission details
Task:Polygonal Chain
Sender:qpwoeirut
Submission time:2021-01-30 09:59:47 +0200
Language:C++ (C++17)
Status:READY
Result:7
Feedback
groupverdictscore
#1ACCEPTED7
#20
#30
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2, 4, 5details
#2ACCEPTED0.01 s1, 2, 4, 5details
#3ACCEPTED0.01 s1, 2, 4, 5details
#4ACCEPTED0.01 s1, 2, 4, 5details
#5ACCEPTED0.01 s1, 2, 4, 5details
#6--2, 4, 5details
#7ACCEPTED0.01 s2, 4, 5details
#8--2, 4, 5details
#9--2, 4, 5details
#10ACCEPTED0.03 s2, 4, 5details
#110.01 s4, 5details
#120.02 s5details
#130.01 s3, 4, 5details
#140.01 s3, 4, 5details
#15ACCEPTED0.01 s1, 2, 4, 5details
#160.01 s4, 5details
#170.01 s4, 5details
#180.02 s5details
#19ACCEPTED0.01 s1, 2, 4, 5details
#20ACCEPTED0.01 s2, 4, 5details
#21ACCEPTED0.01 s2, 4, 5details
#22ACCEPTED0.33 s2, 4, 5details
#230.01 s4, 5details
#240.01 s4, 5details
#250.01 s4, 5details
#260.02 s5details
#270.02 s5details
#280.02 s5details
#290.02 s5details
#30ACCEPTED0.01 s1, 2, 4, 5details

Code

//polygonal_chain.cpp created at 01/29/21 22:20:43

#include <bits/stdc++.h>

using namespace std;

const int MN = 100005;

int N;
int A[MN];
string dir;

bool intersect(const int x1, int y1, int y2, int x3, int x4, const int y3) {
    if (x3 > x4) swap(x3, x4);
    if (y1 > y2) swap(y1, y2);
    //cerr << x1 << ' ' << y1 << ' ' << y2 << ' ' << x3 << ' ' << x4 << ' ' << y3 << endl;
    return (x3 <= x1 && x1 <= x4) && (y1 <= y3 && y3 <= y2);
}

int ord[MN], pos[MN];
int vert_seg[MN][3], hor_seg[MN][3];
int ans[MN];
bool check() {
    //for (int i=0; i<N; ++i) { cerr << ord[i] << ' '; } cerr << endl;
    int x = 0;
    
    int vidx = 0, hidx = 0;
    for (int i=0; i<N; ++i) {
        hor_seg[hidx][0] = x;
        x += A[i] * ((i & 1) ? -1 : +1);
        hor_seg[hidx][1] = x;
        hor_seg[hidx][2] = pos[i];
        ++hidx;

        if (i+1 < N) {
            const int p = pos[i];
            const int tar = pos[i+1];

            vert_seg[vidx][0] = x;
            vert_seg[vidx][1] = p;
            vert_seg[vidx][2] = tar;
            if ((dir[i] == 'D') ^ (p < tar)) return false;
            ++vidx;
            ans[i] = abs(p - tar);
        }
    }

    for (int i=0; i<hidx; ++i) {
        for (int j=0; j<vidx; ++j) {
            if (i == j || i == j + 1) continue;
            //cerr << i << ' ' << j << endl;
            if (intersect(vert_seg[j][0], vert_seg[j][1], vert_seg[j][2], hor_seg[i][0], hor_seg[i][1], hor_seg[i][2])) return false;
        }
    }

    cout << "YES\n";
    for (int i=0; i<N-1; ++i) {
        if (i) cout << ' ';
        cout << ans[i];
    }
    cout << endl;
    return true;
}

bool gen(const int idx, const int prev, const int inc) {
    if (idx == N) {
        return check();
    }
    for (int i=prev+inc; 0 <= i && i < N; i+=inc) {
        if (ord[i] == -1) {
            ord[i] = idx;
            pos[idx] = i;
            if (gen(idx+1, i, dir[idx] == 'D' ? +1 : -1)) return true;
            ord[i] = -1;
        }
    }

    return false;
}

void brute() {
    fill(ord, ord+N, -1);
    dir.push_back('D'); // dummy
    if (!gen(0, -1, 1)) {
        cout << "NO\n";
    }
}

int main() {
    cin.tie(0)->sync_with_stdio(0);

    cin >> N;
    for (int i=0; i<N; ++i) {
        cin >> A[i];
    }
    cin >> dir;

    if (N <= 15) {
        brute();
        return 0;
    } else {
        assert(false);
    }
}

Test details

Test 1

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
2
2 10
D

correct output
YES

user output
YES
1

Test 2

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
8
5 8 7 5 6 5 3 4
DUUUDDD

correct output
YES
1 5 1 1 3 1 1 

user output
YES
1 5 1 1 3 1 1

Test 3

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
8
9 8 8 10 10 8 9 10
DDDUUUD

correct output
YES
1 1 1 4 1 1 7 

user output
YES
1 1 1 4 1 1 7

Test 4

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
8
9 10 8 8 9 9 7 8
DDDDUUU

correct output
NO

user output
NO

Test 5

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
8
10 2 8 3 10 2 10 10
DDUUUUD

correct output
YES
1 1 3 1 1 1 7 

user output
YES
1 5 1 1 1 1 5

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
(empty)

Test 7

Group: 2, 4, 5

Verdict: ACCEPTED

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
YES
1 1 3 1 1 1 1 3 1 1 3 3 1 1

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
(empty)

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
(empty)

Test 10

Group: 2, 4, 5

Verdict: ACCEPTED

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

correct output
NO

user output
NO

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

Test 12

Group: 5

Verdict:

input
100000
999999998 999999996 999999994 ...

correct output
YES
99997 99995 99993 99991 99989 ...

user output
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

Test 15

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
5
6 7 7 6 6
UDUU

correct output
YES
1 4 1 1 

user output
YES
1 4 1 1

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

Test 19

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
8
1 3 1 2 5 1 1 2
DUUUDUU

correct output
NO

user output
NO

Test 20

Group: 2, 4, 5

Verdict: ACCEPTED

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
YES
1 1 5 1 1 3 5 1 1 1 1 5 2 1

Test 21

Group: 2, 4, 5

Verdict: ACCEPTED

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
YES
1 1 1 1 10 9 8 7 6 1 4 1 1 1

Test 22

Group: 2, 4, 5

Verdict: ACCEPTED

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
YES
1 1 1 1 5 3 1 1 9 1 1 3 1 1

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

Test 25

Group: 4, 5

Verdict:

input
1000
1445 1363 1749 1084 262408 263...

correct output
NO

user output
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

Test 26

Group: 5

Verdict:

input
100000
209655 9167 9389 191291 198294...

correct output
NO

user output
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

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
(empty)

Error:
code: input/code.cpp:102: int main(): Assertion `false' failed.

Test 30

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
8
2 3 2 3 5 6 7 8
UDDDUDU

correct output
YES
1 2 1 1 5 6 7 

user output
YES
1 2 1 1 5 6 7