CSES - Datatähti 2024 alku - Results
Submission details
Task:Käännöt
Sender:a256
Submission time:2023-10-30 09:53:08 +0200
Language:C++ (C++20)
Status:READY
Result:32
Feedback
groupverdictscore
#1ACCEPTED13
#2ACCEPTED19
#30
#40
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1details
#2ACCEPTED0.00 s1details
#3ACCEPTED0.00 s1details
#4ACCEPTED0.00 s1details
#5ACCEPTED0.00 s1details
#6ACCEPTED0.00 s2details
#7ACCEPTED0.00 s2details
#8ACCEPTED0.00 s2details
#9ACCEPTED0.00 s2details
#10ACCEPTED0.00 s2details
#11ACCEPTED0.01 s2details
#12ACCEPTED0.01 s2details
#13ACCEPTED0.01 s2details
#14ACCEPTED0.01 s2details
#15ACCEPTED0.01 s2details
#16ACCEPTED0.01 s2details
#17ACCEPTED0.00 s3details
#18ACCEPTED0.00 s3details
#190.00 s3details
#200.00 s3details
#210.00 s3details
#220.00 s3details
#230.00 s3details
#24ACCEPTED0.00 s3details
#250.00 s3details
#26ACCEPTED0.00 s4details
#27ACCEPTED0.00 s4details
#280.00 s4details
#290.00 s4details
#300.00 s4details
#310.00 s4details
#320.00 s4details
#330.00 s4details
#340.00 s4details

Code

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 555

int n,k;
array<int,N> a;
vector<int> v;

void k2(){

	for(int i=0;i<n;++i){
		for(int j=0;j<n-1;++j){
			if(a[j]>a[j+1]){
				v.push_back(j);
				swap(a[j],a[j+1]);
			}
		}
	}
	cout<<"YES\n";
	cout<<v.size()<<'\n';
	for(int e:v) cout<<e+1<<' ';
	cout<<'\n';
}

void k3(){

	for(int i=0;i<5*n;++i){
		for(int j=0;j<n-2;++j){
			if(a[j]>a[j+2]){
				v.push_back(j);
				swap(a[j],a[j+2]);
			}
		}
	}

	if(is_sorted(a.begin(),a.begin()+n)){
		cout<<"YES\n";
		cout<<v.size()<<'\n';
		for(int e:v) cout<<e+1<<' ';
		cout<<'\n';
	} else {
		cout<<"NO\n";
	}
}

int inv(array<int,N> ar){
	int r=0;
	for(int i=0;i<n-1;++i){
		for(int j=i+1;j<n;++j){
			if(a[i]>a[j]) ++r;
		}
	}
	return r;
}

int invab(int l,int r){
	int I=0;
	for(int i=l;i<r-1;++i){
		for(int j=i+1;j<r;++j){
			if(a[i]>a[j]) ++I;
		}
	}
	return I;
}

void brute(){
	array<int,N> A={0},B={0};
	for(int i=0;i<n;++i) B[i]=A[i]=a[i];
	sort(B.begin(),B.begin()+n);
	map<array<int,N>,vector<int>> m;
	m[A]={};

	queue<array<int,N>> q;
	q.push(A);

	while(q.size()){
		array<int,N> ar=q.front(); q.pop();

		auto vc=m[ar];

		for(int i=0;i<n-k+1;++i){
			reverse(ar.begin()+i,ar.begin()+i+k);
			vc.push_back(i);

			if(!m.count(ar)){
				m[ar]=vc;
				q.push(ar);
			}

			vc.pop_back();
			reverse(ar.begin()+i,ar.begin()+i+k);
		}
	}

	//cerr<<m.size()<<'\n';

	if(m.count(B)){
		cout<<"YES\n";
		auto vc=m[B];
		cout<<vc.size()<<'\n';
		for(int e:vc) cout<<e+1<<' ';
		cout<<'\n';
	} else cout<<"NO\n";
}

vector<int> brutea(int l){
	array<int,N> A={0},B={0};
	for(int i=0;i<n;++i) B[i]=A[i]=a[i];
	sort(B.begin(),B.begin()+n);
	map<array<int,N>,vector<int>> m;
	m[A]={};

	queue<array<int,N>> q;
	q.push(A);

	while(q.size()){
		array<int,N> ar=q.front(); q.pop();

		auto vc=m[ar];

		for(int i=l;i<n-k+1;++i){
			reverse(ar.begin()+i,ar.begin()+i+k);
			vc.push_back(i);

			if(!m.count(ar)){
				m[ar]=vc;
				q.push(ar);
			}

			vc.pop_back();
			reverse(ar.begin()+i,ar.begin()+i+k);
		}
	}

	//cerr<<m.size()<<'\n';

	if(!m.count(B)){
		exit(1);
	}
	return m[B];
}

void cycle(int j,int t){
	for(int i=0;i<t;++i){
		v.push_back(j);
		v.push_back(j+1);
		v.push_back(j);
		v.push_back(j+1);
	}
}

void k4(){
	if(n<9){
		brute();
		return;
	}

	int ii=invab(0,n);
	if(ii&1){
		cout<<"NO\n";
		return;
	}

	for(int i=1;i<=n-6;++i){ // loput 6 brute
		int j=0;
		while(a[j]!=i) ++j;

		if(j<k-1){
			reverse(a.begin()+j,a.begin()+j+k);
			v.push_back(j);
			j+=3;
		}

		if(j==n-1){
			reverse(a.begin()+j-3,a.begin()+j+1);
			v.push_back(j-3);
			j-=3;
		}

		while(j!=i+k-2){
			reverse(a.begin()+j-2,a.begin()+j+2);
			v.push_back(j-2);
			--j;
		}

		reverse(a.begin()+j-3,a.begin()+j+1);
		v.push_back(j-3);
		j-=3;

		if(j!=i-1){
			//exit(2);
		}
	}

	auto vv=brutea(n-7);
	for(int e:vv){
		reverse(a.begin()+e,a.begin()+e+k);
		v.push_back(e);
	}
	if(!is_sorted(a.begin(),a.begin()+n)){
		for(int i=0;i<n;++i) cerr<<a[i]<<' ';
		cerr<<'\n';
		//exit(3);
	} else {
		cout<<"YES\n";
		cout<<v.size()<<'\n';
		for(int e:v) cout<<e+1<<' ';
		cout<<'\n';
	}
}

void k5(){
	if(n<=9){
		brute();
		return;
	}
	cout<<"NJET\n";
}

int main(){
	cin.tie(0)->sync_with_stdio(0);
	cin>>n>>k;
	for(int i=0;i<n;++i) cin>>a[i];
	switch(k){
		case 2:
			k2();
			break;
		case 3:
			k3();
			break;
		case 4:
			k4();
			break;
		case 5:
			k5();
			break;
	}
}

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
5 2
1 2 3 4 5

correct output
YES
0

user output
YES
0

Test 2

Group: 1

Verdict: ACCEPTED

input
5 2
2 1 3 4 5

correct output
YES
1
1

user output
YES
1

Test 3

Group: 1

Verdict: ACCEPTED

input
20 2
6 20 18 2 16 13 19 17 8 14 11 ...

correct output
YES
366
2 3 4 5 6 7 8 9 10 11 12 13 14...

user output
YES
132
2 3 4 5 6 7 8 9 10 11 12 13 14...
Truncated

Test 4

Group: 1

Verdict: ACCEPTED

input
100 2
100 92 62 88 12 7 43 31 19 72 ...

correct output
YES
2876
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
2438
1 2 3 4 5 6 7 8 9 10 11 12 13 ...
Truncated

Test 5

Group: 1

Verdict: ACCEPTED

input
100 2
100 99 98 97 96 95 94 93 92 91...

correct output
YES
5248
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
YES
4950
1 2 3 4 5 6 7 8 9 10 11 12 13 ...
Truncated

Test 6

Group: 2

Verdict: ACCEPTED

input
5 3
1 2 3 4 5

correct output
YES
0

user output
YES
0

Test 7

Group: 2

Verdict: ACCEPTED

input
5 3
3 5 4 1 2

correct output
NO

user output
NO

Test 8

Group: 2

Verdict: ACCEPTED

input
5 3
5 2 1 4 3

correct output
YES
8
1 2 1 3 1 2 3 1

user output
YES
2
1 3 

Test 9

Group: 2

Verdict: ACCEPTED

input
20 3
19 14 1 18 3 4 11 20 13 6 17 8...

correct output
YES
52
8 10 12 14 16 18 1 3 5 7 9 11 ...

user output
YES
42
1 3 4 5 7 8 9 10 11 12 13 14 1...
Truncated

Test 10

Group: 2

Verdict: ACCEPTED

input
20 3
9 6 13 18 5 10 3 2 7 20 1 4 19...

correct output
YES
50
10 12 14 16 18 13 15 17 4 6 8 ...

user output
YES
34
3 4 5 6 7 9 10 12 13 14 15 16 ...

Test 11

Group: 2

Verdict: ACCEPTED

input
500 3
53 52 21 76 25 142 5 4 83 176 ...

correct output
YES
15194
334 336 338 340 342 344 346 34...

user output
YES
15174
1 3 5 6 10 11 12 13 14 15 16 1...
Truncated

Test 12

Group: 2

Verdict: ACCEPTED

input
500 3
51 44 147 172 1 28 27 82 233 1...

correct output
YES
15565
366 368 370 372 374 376 378 38...

user output
YES
15543
3 4 5 6 8 10 11 12 13 15 16 17...
Truncated

Test 13

Group: 2

Verdict: ACCEPTED

input
500 3
75 46 179 62 221 14 67 154 89 ...

correct output
YES
15920
454 456 458 460 462 464 466 46...

user output
YES
15902
4 5 7 9 10 11 12 13 14 15 16 1...
Truncated

Test 14

Group: 2

Verdict: ACCEPTED

input
500 3
161 54 285 12 71 142 111 94 97...

correct output
YES
15931
408 410 412 414 416 418 420 42...

user output
YES
15907
2 3 5 6 7 8 9 10 11 13 15 16 1...
Truncated

Test 15

Group: 2

Verdict: ACCEPTED

input
500 3
122 260 455 113 315 276 433 43...

correct output
NO

user output
NO

Test 16

Group: 2

Verdict: ACCEPTED

input
500 3
499 500 497 498 495 496 493 49...

correct output
YES
62264
2 4 6 8 10 12 14 16 18 20 22 2...

user output
YES
62250
1 2 3 4 5 6 7 8 9 10 11 12 13 ...
Truncated

Test 17

Group: 3

Verdict: ACCEPTED

input
5 4
1 2 3 4 5

correct output
YES
0

user output
YES
0

Test 18

Group: 3

Verdict: ACCEPTED

input
5 4
5 1 2 3 4

correct output
YES
4
1 2 1 2

user output
YES
4
1 2 1 2 

Test 19

Group: 3

Verdict:

input
500 4
58 14 107 124 4 113 24 290 56 ...

correct output
YES
15698
389 392 395 398 401 404 407 41...

user output
(empty)

Test 20

Group: 3

Verdict:

input
500 4
113 187 278 242 23 67 48 298 3...

correct output
YES
15004
480 481 480 482 485 488 491 49...

user output
(empty)

Test 21

Group: 3

Verdict:

input
500 4
5 233 199 35 213 354 11 134 30...

correct output
YES
16770
458 461 464 467 470 473 476 47...

user output
(empty)

Test 22

Group: 3

Verdict:

input
500 4
169 47 21 137 57 138 360 147 4...

correct output
YES
15889
497 371 372 371 373 376 379 38...

user output
(empty)

Test 23

Group: 3

Verdict:

input
500 4
493 409 291 313 156 443 496 40...

correct output
YES
22886
480 481 480 482 485 488 491 49...

user output
(empty)

Test 24

Group: 3

Verdict: ACCEPTED

input
500 4
137 99 100 226 326 298 140 340...

correct output
NO

user output
NO

Test 25

Group: 3

Verdict:

input
500 4
500 499 498 497 496 495 494 49...

correct output
YES
41458
1 2 1 2 5 8 11 14 17 20 23 26 ...

user output
(empty)

Error:
terminate called after throwing an instance of 'std::bad_alloc'
  what():  std::bad_alloc

Test 26

Group: 4

Verdict: ACCEPTED

input
5 5
1 2 3 4 5

correct output
YES
0

user output
YES
0

Test 27

Group: 4

Verdict: ACCEPTED

input
5 5
5 4 3 2 1

correct output
YES
1
1

user output
YES
1

Test 28

Group: 4

Verdict:

input
500 5
145 26 285 154 147 314 141 40 ...

correct output
YES
13786
216 220 224 228 232 236 240 24...

user output
NJET

Test 29

Group: 4

Verdict:

input
500 5
137 22 399 292 249 6 51 224 42...

correct output
YES
13465
456 460 464 468 472 476 480 48...

user output
NJET

Test 30

Group: 4

Verdict:

input
500 5
153 52 85 100 329 60 433 468 4...

correct output
YES
13642
377 378 377 380 384 388 392 39...

user output
NJET

Test 31

Group: 4

Verdict:

input
500 5
267 326 95 108 189 32 291 366 ...

correct output
YES
14639
213 214 213 216 220 224 228 23...

user output
NJET

Test 32

Group: 4

Verdict:

input
500 5
15 450 272 80 321 101 247 438 ...

correct output
NO

user output
NJET

Test 33

Group: 4

Verdict:

input
499 5
497 498 499 496 495 494 493 49...

correct output
YES
30886
3 7 11 15 19 23 27 31 35 39 43...

user output
NJET

Test 34

Group: 4

Verdict:

input
500 5
499 500 497 498 495 496 493 49...

correct output
YES
30919
1 4 8 12 16 20 24 28 32 36 40 ...

user output
NJET