CSES - Datatähti Open 2021

Submission details
Task:	Polygonal Chain
Sender:	tutis
Submission time:	2021-01-30 22:12:50 +0200
Language:	C++ (C++11)
Status:	READY
Result:	7

Feedback
group	verdict	score
#1	ACCEPTED	7
#2	WRONG ANSWER	0
#3	WRONG ANSWER	0
#4	WRONG ANSWER	0
#5	WRONG ANSWER	0

Test results
test	verdict	time	group
#1	ACCEPTED	0.01 s	1, 2, 4, 5	details
#2	ACCEPTED	0.01 s	1, 2, 4, 5	details
#3	ACCEPTED	0.01 s	1, 2, 4, 5	details
#4	ACCEPTED	0.01 s	1, 2, 4, 5	details
#5	ACCEPTED	0.01 s	1, 2, 4, 5	details
#6	WRONG ANSWER	0.01 s	2, 4, 5	details
#7	ACCEPTED	0.01 s	2, 4, 5	details
#8	WRONG ANSWER	0.01 s	2, 4, 5	details
#9	WRONG ANSWER	0.01 s	2, 4, 5	details
#10	ACCEPTED	0.01 s	2, 4, 5	details
#11	WRONG ANSWER	0.01 s	4, 5	details
#12	WRONG ANSWER	0.02 s	5	details
#13	WRONG ANSWER	0.01 s	3, 4, 5	details
#14	ACCEPTED	0.01 s	3, 4, 5	details
#15	ACCEPTED	0.01 s	1, 2, 4, 5	details
#16	WRONG ANSWER	0.01 s	4, 5	details
#17	WRONG ANSWER	0.01 s	4, 5	details
#18	TIME LIMIT EXCEEDED	--	5	details
#19	ACCEPTED	0.01 s	1, 2, 4, 5	details
#20	WRONG ANSWER	0.01 s	2, 4, 5	details
#21	WRONG ANSWER	0.01 s	2, 4, 5	details
#22	WRONG ANSWER	0.01 s	2, 4, 5	details
#23	WRONG ANSWER	0.01 s	4, 5	details
#24	WRONG ANSWER	0.01 s	4, 5	details
#25	ACCEPTED	0.01 s	4, 5	details
#26	ACCEPTED	0.02 s	5	details
#27	WRONG ANSWER	0.03 s	5	details
#28	WRONG ANSWER	0.02 s	5	details
#29	WRONG ANSWER	0.03 s	5	details
#30	ACCEPTED	0.01 s	1, 2, 4, 5	details

Code

/*input
4
4 3 2 3
UDD
*/
#pragma GCC optimize ("O3")
#pragma GCC target ("avx2")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
template<typename T>
using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
template<typename T, typename X>
using ordered_map = tree<T, X, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
template<typename T, typename X>
using fast_map = cc_hash_table<T, X>;
//using ull = __uint128_t;
using ull = unsigned long long;
using ll = long long;
using ld = long double;
mt19937_64 rng(123);
const ll mod = 1000000007;
int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	int n;
	cin >> n;
	int P[n];
	for (int i = 0; i < n; i++)
		cin >> P[i];
	string s;
	cin >> s;
	int x1[n];
	int x2[n];
	for (int i = 0; i < n; i++)
	{
		if (i == 0)
			x1[i] = 0;
		else
			x1[i] = x2[i - 1];
		if (i % 2 == 0)
			x2[i] = x1[i] + P[i];
		else
			x2[i] = x1[i] - P[i];
	}
	int p[n];
	iota(p, p + n, 0);
	auto check = [&]()
	{
		bool ok = true;
		int ans[n];
		for (int i = 0; i < n - 1; i++)
		{
			ans[i] = p[i + 1] - p[i];
			assert(p[i + 1] != p[i]);
			if (s[i] == 'U' && ans[i] < 0)
				ok = false;
			if (s[i] == 'D' && ans[i] > 0)
				ok = false;
			ans[i] = abs(ans[i]);
			int x = x2[i];
			int y0 = min(p[i], p[i + 1]);
			int y1 = max(p[i], p[i + 1]);
			for (int j = 0; j < n && ok; j++)
			{
				if (j == i || j == i + 1)
					continue;
				int x_1 = min(x1[j], x2[j]);
				int x_2 = max(x1[j], x2[j]);
				if (x_1 <= x && x <= x_2 && y0 <= p[j] && p[j] <= y1)
				{
					ok = false;
				}
			}
		}
		if (ok)
		{
			cout << "YES\n";
			for (int i = 0; i < n - 1; i++)
				cout << ans[i] << " ";
			cout << "\n";
			exit(0);
		}
	};
	if (n <= 8)
	{
		do
		{
			check();
		} while (next_permutation(p, p + n));
		cout << "NO\n";
		return 0;
	}
	p[0] = 0;
	int mn = 0;
	int mx = 0;
	for (int i = 0; i < n - 1; i++)
	{
		if (s[i] == 'U')
		{
			p[i + 1] = mx + 1;
			mx++;
		}
		else
		{
			mn--;
			p[i + 1] = mn;
		}
	}
	check();
	cout << "NO\n";
}

Test details

Test 1

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`2 2 10 D` view save

correct output
`YES 1` view save

user output
`YES 1` view save

Test 2

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`8 5 8 7 5 6 5 3 4 DUUUDDD` view save

correct output
`YES 1 5 1 1 3 1 1` view save

user output
`YES 1 5 1 1 3 1 1` view save

Test 3

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`8 9 8 8 10 10 8 9 10 DDDUUUD` view save

correct output
`YES 1 1 1 4 1 1 7` view save

user output
`YES 1 1 1 5 1 1 3` view save

Test 4

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`8 9 10 8 8 9 9 7 8 DDDDUUU` view save

correct output
`NO` view save

user output
`NO` view save

Test 5

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`8 10 2 8 3 10 2 10 10 DDUUUUD` view save

correct output
`YES 1 1 3 1 1 1 7` view save

user output
`YES 1 1 3 2 1 1 3` view save

Test 6

Group: 2, 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 1 2 3 1 1 3 1 1 1 10 1 3 1 1` view save

user output
`NO` view save

Test 7

Group: 2, 4, 5

Verdict: ACCEPTED

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

correct output
`YES 1 1 3 1 1 1 1 1 9 1 11 1 13 1` view save

user output
`YES 1 1 3 1 1 1 1 1 9 1 11 1 13 1` view save

Test 8

Group: 2, 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 1 2 3 4 5 6 14 1 6 5 4 3 2 1` view save

user output
`NO` view save

Test 9

Group: 2, 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 13 11 9 7 5 3 1 1 3 5 7 9 11 1...` view save

user output
`NO` view save

Test 10

Group: 2, 4, 5

Verdict: ACCEPTED

input
`15 100 2 99 1 78 4 93 2 100 1 15 ...` view save

correct output
`NO` view save

user output
`NO` view save

Test 11

Group: 4, 5

Verdict: WRONG ANSWER

input
`1000 999997 999995 999993 999991 99...` view save

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

user output
`NO` view save

Test 12

Group: 5

Verdict: WRONG ANSWER

input
`100000 999999998 999999996 999999994 ...` view save

correct output
`YES 99997 99995 99993 99991 99989 ...` view save

user output
`NO` view save

Test 13

Group: 3, 4, 5

Verdict: WRONG ANSWER

input
`1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

correct output
`YES 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

user output
`NO` view save

Test 14

Group: 3, 4, 5

Verdict: ACCEPTED

input
`1000 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

correct output
`NO` view save

user output
`NO` view save

Test 15

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`5 6 7 7 6 6 UDUU` view save

correct output
`YES 1 4 1 1` view save

user output
`YES 1 4 1 1` view save

Test 16

Group: 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 1 1 1 4 1 3 1 1 7 1 19 1 1 3 1...` view save

user output
`NO` view save

Test 17

Group: 4, 5

Verdict: WRONG ANSWER

input
`1000 1000000 1 146324 146324 289287...` view save

correct output
`YES 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

user output
`NO` view save

Test 18

Group: 5

Verdict: TIME LIMIT EXCEEDED

input
`100000 1000000000 1 421262579 4212625...` view save

correct output
`YES 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...` view save

user output
(empty)

Test 19

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`8 1 3 1 2 5 1 1 2 DUUUDUU` view save

correct output
`NO` view save

user output
`NO` view save

Test 20

Group: 2, 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 1 1 5 1 1 3 5 1 1 1 1 5 2 1` view save

user output
`NO` view save

Test 21

Group: 2, 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 1 1 1 1 10 9 8 7 6 1 4 1 1 1` view save

user output
`NO` view save

Test 22

Group: 2, 4, 5

Verdict: WRONG ANSWER

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

correct output
`YES 1 1 1 1 5 1 7 1 3 1 1 3 1 1` view save

user output
`NO` view save

Test 23

Group: 4, 5

Verdict: WRONG ANSWER

input
`1000 3246 3562 197273 197429 197755...` view save

correct output
`YES 1 1 1 4 5 10 3 1 1 3 7 12 1 1 ...` view save

user output
`NO` view save

Test 24

Group: 4, 5

Verdict: WRONG ANSWER

input
`1000 503981 503487 503350 502673 50...` view save

correct output
`YES 999 1 997 1 1 994 989 1 1 1 1 ...` view save

user output
`NO` view save

Test 25

Group: 4, 5

Verdict: ACCEPTED

input
`1000 1445 1363 1749 1084 262408 263...` view save

correct output
`NO` view save

user output
`NO` view save

Test 26

Group: 5

Verdict: ACCEPTED

input
`100000 209655 9167 9389 191291 198294...` view save

correct output
`NO` view save

user output
`NO` view save

Test 27

Group: 5

Verdict: WRONG ANSWER

input
`100000 16295 14904 5103 13337 26939 3...` view save

correct output
`YES 1 1 1 1 5 6 1 1 1 10 11 1 13 1...` view save

user output
`NO` view save

Test 28

Group: 5

Verdict: WRONG ANSWER

input
`100000 1859 174288 15040 4631 4993844...` view save

correct output
`YES 1 3 1 1 99997 99992 1 1 5 1 1 ...` view save

user output
`NO` view save

Test 29

Group: 5

Verdict: WRONG ANSWER

input
`100000 959817 958289 966165 922369 92...` view save

correct output
`YES 1 1 1 1 1 1 1 1 1 1 11 14 1 1 ...` view save

user output
`NO` view save

Test 30

Group: 1, 2, 4, 5

Verdict: ACCEPTED

input
`8 2 3 2 3 5 6 7 8 UDDDUDU` view save

correct output
`YES 1 2 1 1 5 6 7` view save

user output
`YES 3 1 1 2 5 6 7` view save

Datatähti Open 2021

Polygonal Chain

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Tasks

Submissions (tutis)