CSES - Datatähti 2019 alku - Results
Submission details
Task:Leimasin
Sender:Yytsi
Submission time:2018-10-03 13:56:13 +0300
Language:C++
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1details
#2ACCEPTED0.02 s1details
#30.01 s1details
#4ACCEPTED0.02 s1details
#5ACCEPTED0.02 s1details
#6ACCEPTED0.02 s1details
#7ACCEPTED0.02 s1details
#8ACCEPTED0.02 s1details
#9ACCEPTED0.01 s1details
#10ACCEPTED0.02 s1details
#11ACCEPTED0.03 s1details
#12ACCEPTED0.01 s1details
#13ACCEPTED0.02 s1details
#14ACCEPTED0.03 s1details
#15ACCEPTED0.01 s2details
#160.03 s2details
#170.02 s2details
#18ACCEPTED0.02 s2details
#19ACCEPTED0.02 s2details
#20ACCEPTED0.03 s2details
#210.02 s2details
#22ACCEPTED0.03 s2details
#23ACCEPTED0.02 s2details
#24ACCEPTED0.01 s2details
#25ACCEPTED0.03 s2details
#26ACCEPTED0.01 s2details
#27ACCEPTED0.01 s2details
#28ACCEPTED0.03 s2details
#29ACCEPTED0.02 s3details
#30ACCEPTED0.02 s3details
#31ACCEPTED0.03 s3details
#32ACCEPTED0.03 s3details
#33ACCEPTED0.02 s3details
#34ACCEPTED0.02 s3details
#350.01 s3details
#360.01 s3details
#37ACCEPTED0.02 s3details
#38ACCEPTED0.02 s3details
#39ACCEPTED0.02 s3details
#40ACCEPTED0.01 s3details
#41ACCEPTED0.01 s3details
#42ACCEPTED0.03 s3details

Compiler report

input/code.cpp: In function 'std::vector<int> f(int, std::__cxx11::string)':
input/code.cpp:28:9: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  if (cnt==x.size()) return {};
      ~~~^~~~~~~~~~
input/code.cpp: In function 'void hsh(std::__cxx11::string)':
input/code.cpp:3:37: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define FOR(i, a, b) for (int i=a; i<(b); i++)
                                     ^
input/code.cpp:90:2: note: in expansion of macro 'FOR'
  FOR(i,1,x.size()) {
  ^~~

Code

#include <bits/stdc++.h>
using namespace std;
#define FOR(i, a, b) for (int i=a; i<(b); i++)
#define IO ios_base::sync_with_stdio(0); cin.tie(0)
#define ff first
#define ss second
#define pb push_back
#define INF 2147483647
#define LINF (1LL<<61LL)
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

#define N 1003

string s, L;
int n, m;

void fail() {
	cout<<"-1\n"; exit(0);
}

vector<int> f(int offset, string x) {
	if (x.size()==0) return {};
	int cnt = 0;
	for(char c:x) if(c=='?')cnt++;
	if (cnt==x.size()) return {};
	
	// ...??
	// or
	// ??...
	
	int sz = x.size();
	int idx = -1;
	FOR(i,0,sz-m+1) {
		if (i==0 && x[0]=='?') continue;
		bool found = true;
		FOR(j,0,m) {
			if (x[i+j] == '?') continue;
			if (x[i+j] != L[j]) {found=false; break;}
		}
		if (found) {
			idx = i;
			break;
		}
	}
	
	if (idx == -1) fail();
	
	FOR(j,0,m) x[idx+j] = '?';
	//cout<<x<<" "<<sz<<" offset: "<<offset<<" idx: "<<idx<<"\n";
	
	
	// Go left
	vector<int> left = f(offset+0, x.substr(0,idx+m));
	vector<int> right = f(offset+idx, x.substr(idx,string::npos));
	vector<int> res;
	for (int v:left) res.pb(v);
	for (int v:right) res.pb(v);
	res.pb(idx+offset);
	return res;
}

ll A = 911382323, B = 972663749;
ll pA[N];
ll H[N], HASH[N];
ll pre[N], suff[N];
vector<int> pres;
vector<int> sufs;
vector<int> full;

// 0-indexed
ll rq(int a, int b) {
	if (a == 0) return H[b];
	ll h = (H[b]-H[a-1]*pA[b-a+1])%B;
	if (h<0) h+=B;
	return h;
}

ll ran(int a, int b) {
	if (a==0) return HASH[b];
	ll h = (HASH[b]-HASH[a-1]*pA[b-a+1])%B;
	if (h<0) h+=B;
	return h;
}

void hsh(string x) {
	HASH[0] = x[0];
	FOR(i,1,x.size()) {
		HASH[i] = (HASH[i-1]*A+x[i])%B;
	}
	FOR(l,0,m) {
		ll a = ran(0, l);
		ll b = ran(m-1-l,m-1);
		if (a < 0) a += B;
		if (b < 0) b += B;
		pre[l+1] = a;
		suff[l+1] = b;
	}
}

ll pure(const string& x) {
	ll h = 0;
	for (char c:x) h=(h*A+(ll)c)%B;
	return h;
}

string w = "";
void paint(int p) {
	FOR(j,0,m) w[j+p] = L[j]; 
}

int main() {
	IO;
	cin>>s; cin>>L;
	n = s.size();
	m = L.size();
	FOR(j,0,n) w+="?";
	
	pA[0] = 1;
	H[0] = s[0];
	FOR(i,1,n+1) {
		H[i] = (H[i-1]*A+s[i])%B;
		pA[i] = pA[i-1]*A%B;
	}
	
	hsh(L);
	int i = 0;
	for (;i<n;) {
		// Find block length!
		bool found = false;
		for (int l=m; l>0; l--) {
			if (i+l-1 >= n) continue;
			ll slice = rq(i,i+l-1);
			//cout<<"slice: ("<<i<<","<<i+l-1<<")\n";
			
			if (slice == pre[l]) {
				if (i+m-1 < n) {
					if (l!=m) pres.pb(i+1);
					else full.pb(i+1);
					//cout<<"flip on "<<i+1<<"\n";
					i += l;
					found = true;
					break;
				}
			}
			if (slice == suff[l]) {
				int p = i+l-1;
				if (p-m+1 >= 0) {
					int crd = i-(m-l)+1;
					if (l!=m) sufs.pb(crd);
					else full.pb(crd);
					//cout<<"flip on "<<crd<<"\n";
					i += l;
					found = true;
					break;
				}
			}
		}
		
		if (found) continue;
		// not found :(
		cout<<-1<<"\n";
		exit(0);
	}
	
	//cout<<pres.size()+full.size()+sufs.size()<<"\n";
	
	
	vector<int> res;
	reverse(sufs.begin(), sufs.end());
	for (int pr : pres) res.pb(pr-1);
	for (int sf : sufs) res.pb(sf-1);
	for (int f : full) res.pb(f-1);
	
	for (int v : res) paint(v);
	if (w != s) {
		// Let's ask for confirmation
		FOR(j,0,n) w[j] = '?';
		vector<int> conf = f(0, s);
		for (int v : conf) paint(v);
		if (w == s) {
			// we we're wrong!
			cout<<conf.size()<<"\n";
			for (int v : conf) cout<<v+1<<" ";
			exit(0);
		} else {
			// we we're right
			cout<<"-1\n"; exit(0);
		}
	}
	
	cout<<pres.size()+full.size()+sufs.size()<<"\n";
	reverse(sufs.begin(), sufs.end());
	for (int pr : pres) cout<<pr<<" ";
	for (int sf : sufs) cout<<sf<<" ";
	for (int f : full) cout<<f<<" ";
	
	//cout<<w<<"\n";
}






Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
BBBBBBBBBB
B

correct output
10
10 9 8 7 6 5 4 3 2 1 

user output
10
1 2 3 4 5 6 7 8 9 10 

Test 2

Group: 1

Verdict: ACCEPTED

input
AABBABABAB
AB

correct output
6
1 9 7 5 3 2 

user output
6
1 3 2 5 7 9 

Test 3

Group: 1

Verdict:

input
AABAAABAAA
AABAA

correct output
4
6 5 2 1 

user output
3
5 6 1 

Test 4

Group: 1

Verdict: ACCEPTED

input
BAAAAAABBB
BAAAAAABB

correct output
2
2 1 

user output
2
2 1 

Test 5

Group: 1

Verdict: ACCEPTED

input
AAABBABBAA
AAABBABBAA

correct output
1

user output
1

Test 6

Group: 1

Verdict: ACCEPTED

input
GGGGGGGGGG
G

correct output
10
10 9 8 7 6 5 4 3 2 1 

user output
10
1 2 3 4 5 6 7 8 9 10 

Test 7

Group: 1

Verdict: ACCEPTED

input
QUUQUUQUQU
QU

correct output
6
9 7 5 4 2 1 

user output
6
2 5 1 4 7 9 

Test 8

Group: 1

Verdict: ACCEPTED

input
DWXDWDWXHJ
DWXHJ

correct output
3
1 4 6 

user output
3
1 4 6 

Test 9

Group: 1

Verdict: ACCEPTED

input
FSOCRDGQBB
FSOCRDGQB

correct output
2
2 1 

user output
2
2 1 

Test 10

Group: 1

Verdict: ACCEPTED

input
OETMIMPUPD
OETMIMPUPD

correct output
1

user output
1

Test 11

Group: 1

Verdict: ACCEPTED

input
DOWEUOWUEU
DOWEU

correct output
-1

user output
-1

Test 12

Group: 1

Verdict: ACCEPTED

input
JQZYVSIWTE
JQZVYSIWTE

correct output
-1

user output
-1

Test 13

Group: 1

Verdict: ACCEPTED

input
ABABABABA
ABA

correct output
4
7 5 3 1 

user output
4
3 5 7 1 

Test 14

Group: 1

Verdict: ACCEPTED

input
AAAAAAAAAA
AAAAAAAAAB

correct output
-1

user output
-1

Test 15

Group: 2

Verdict: ACCEPTED

input
BBBBBBBBBBBBBBBBBBBBBBBBBBBBBB...

correct output
100
100 99 98 97 96 95 94 93 92 91...

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

Test 16

Group: 2

Verdict:

input
BABABAAAAAAAAAAAAAAAAAABABAAAA...

correct output
36
87 43 24 1 91 79 69 68 67 66 6...

user output
15
1 24 43 89 3 10 14 33 53 69 91...

Test 17

Group: 2

Verdict:

input
ABABAAAAABABBBBAAAABBBBAABBBBB...

correct output
22
51 50 43 41 31 28 26 24 21 20 ...

user output
5
41 43 50 51 1 

Test 18

Group: 2

Verdict: ACCEPTED

input
AAABABAAAABBBBBABABBAABBABABBA...

correct output
2
1 2 

user output
2
1 2 

Test 19

Group: 2

Verdict: ACCEPTED

input
AABABBBBBBAABBABABBBBBBAABBAAA...

correct output
1

user output
1

Test 20

Group: 2

Verdict: ACCEPTED

input
SSSSSSSSSSSSSSSSSSSSSSSSSSSSSS...

correct output
100
100 99 98 97 96 95 94 93 92 91...

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

Test 21

Group: 2

Verdict:

input
NNNININIMNIMKLMXCNIMKLMXCDEIMK...

correct output
18
1 2 3 74 5 79 58 7 84 64 37 10...

user output
-1

Test 22

Group: 2

Verdict: ACCEPTED

input
VYQFNHMVTKOEYCXWINLKLHVFMEPQEU...

correct output
3
51 2 1 

user output
2
1 51 

Test 23

Group: 2

Verdict: ACCEPTED

input
IISNROLHLOJIWPTVFHFLUQRIROVLYP...

correct output
2
1 2 

user output
2
1 2 

Test 24

Group: 2

Verdict: ACCEPTED

input
WPMEMERJXXADLKONUZPUUFTPSXDHIV...

correct output
1

user output
1

Test 25

Group: 2

Verdict: ACCEPTED

input
LNSBGZAWFJZAWFJWFJLNSBLNSBGZAL...

correct output
-1

user output
-1

Test 26

Group: 2

Verdict: ACCEPTED

input
IPIPYFUMRIPYFUMRLPIIIPYFIPYFUM...

correct output
-1

user output
-1

Test 27

Group: 2

Verdict: ACCEPTED

input
ABABABABABABABABABABABABABABAB...

correct output
49
97 95 93 91 89 87 85 83 81 79 ...

user output
49
3 5 7 9 11 13 15 17 19 21 23 2...
Truncated

Test 28

Group: 2

Verdict: ACCEPTED

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
-1

user output
-1

Test 29

Group: 3

Verdict: ACCEPTED

input
BBBBBBBBBBBBBBBBBBBBBBBBBBBBBB...

correct output
1000
1000 999 998 997 996 995 994 9...

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

Test 30

Group: 3

Verdict: ACCEPTED

input
BBBBBBBBAABBBBBBBBAABBBBBBBAAB...

correct output
218
1 626 607 519 415 5 975 957 92...

user output
218
1 5 11 15 29 41 50 62 81 122 1...
Truncated

Test 31

Group: 3

Verdict: ACCEPTED

input
AABBBABAABABAAABBAAAAAAABBBAAB...

correct output
55
569 639 403 761 663 437 172 90...

user output
55
172 403 437 569 639 663 761 90...
Truncated

Test 32

Group: 3

Verdict: ACCEPTED

input
ABBAAABAAABAAAAABBABABBABBABBB...

correct output
2
2 1 

user output
2
2 1 

Test 33

Group: 3

Verdict: ACCEPTED

input
BAAABBABBBAAAABAAAABBBBABAABAA...

correct output
1

user output
1

Test 34

Group: 3

Verdict: ACCEPTED

input
UUUUUUUUUUUUUUUUUUUUUUUUUUUUUU...

correct output
1000
1000 999 998 997 996 995 994 9...

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

Test 35

Group: 3

Verdict:

input
KSBMRKKSBMRZXBDKSKSBMRZXBDAMRZ...

correct output
178
723 731 1 935 857 820 760 735 ...

user output
-1

Test 36

Group: 3

Verdict:

input
ILYLILYLVJILYLVJZCCQDLFRLSXZDM...

correct output
21
671 54 747 504 113 1 856 764 5...

user output
-1

Test 37

Group: 3

Verdict: ACCEPTED

input
ZZJZNKHDLJBPXIAZNJIIGBEEJFSDAF...

correct output
2
1 2 

user output
2
1 2 

Test 38

Group: 3

Verdict: ACCEPTED

input
FIMWTOLSRKOWYDPCOFUJZMXJEJFKSU...

correct output
1

user output
1

Test 39

Group: 3

Verdict: ACCEPTED

input
AIVHCGUMKSTIYBRNPONXHRFVBKPYHX...

correct output
-1

user output
-1

Test 40

Group: 3

Verdict: ACCEPTED

input
QPMSLIDCLFLBEXGVVQQNSVKJYXGETC...

correct output
-1

user output
-1

Test 41

Group: 3

Verdict: ACCEPTED

input
ABABABABABABABABABABABABABABAB...

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

user output
499
3 5 7 9 11 13 15 17 19 21 23 2...
Truncated

Test 42

Group: 3

Verdict: ACCEPTED

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
-1

user output
-1