Submission #961238

# Submission time Handle Problem Language Result Execution time Memory
961238 2024-04-11T18:21:49 Z MinaRagy06 Paint By Numbers (IOI16_paint) C++17
59 / 100
5 ms 8824 KB
#include <bits/stdc++.h>
#include "paint.h"
#ifdef MINA
#include "grader.cpp"
#endif
using namespace std;
#define ll long long

const int N = 200'005, K = 105;
bool prfdp[N][K], sufdp[N][K];
string solve_puzzle(string s, vector<int> c) {
	int n = s.size(), k = c.size();
	int prf[n + 1]{};
	for (int i = 0; i < n; i++) {
		prf[i + 1] = prf[i] + (s[i] == '_');
	}
	prfdp[0][0] = 1;
	for (int i = 1; i <= n; i++) {
		for (int j = 0; j <= k; j++) {
			if (j - 1 >= 0) {
				if (i - c[j - 1] >= 0 && prf[i] - prf[i - c[j - 1]] == 0) {
					if (j - 1 == 0) {
						prfdp[i][j] = prfdp[i - c[j - 1]][j - 1];
					} else if (s[i - c[j - 1] - 1] != 'X' && i - c[j - 1] - 1 >= 0) {
						prfdp[i][j] = prfdp[i - c[j - 1] - 1][j - 1];
					}
				}
			}
			if (s[i - 1] != 'X') {
				prfdp[i][j] |= prfdp[i - 1][j];
			}
		}
	}
	sufdp[n][k] = 1;
	for (int i = n - 1; i >= 0; i--) {
		for (int j = 0; j <= k; j++) {
			if (j + 1 <= k) {
				if (i + c[j] <= n && prf[i + c[j]] - prf[i] == 0) {
					if (j + 1 == k) {
						sufdp[i][j] = sufdp[i + c[j]][j + 1];
					} else if (i + c[j] + 1 <= n && s[i + c[j]] != 'X') {
						sufdp[i][j] = sufdp[i + c[j] + 1][j + 1];
					}
				}
			}
			if (s[i] != 'X') {
				sufdp[i][j] |= sufdp[i + 1][j];
			}
		}
	}
	for (int j = 0; j <= k; j++) {
		int cur = 0, cnt = 0;
		for (int i = 1; i <= n; i++) {
			if (prfdp[i][j]) {
				cur++;
			} else {
				if (cur) {
					cnt++;
				}
				cur = 0;
			}
		}
		assert(cnt <= 1);
	}
	for (int j = 0; j <= k; j++) {
		int cur = 0, cnt = 0;
		for (int i = 0; i < n; i++) {
			if (sufdp[i][j]) {
				cur++;
			} else {
				if (cur) {
					cnt++;
				}
				cur = 0;
			}
		}
		assert(cnt <= 1);
	}
	string ans;
	ans.resize(n);
	for (int i = 0; i < n; i++) {
		if (s[i] != '.') {
			ans[i] = s[i];
			continue;
		}
		bool ok[2]{};
		for (int j = 0; j < k; j++) {
			for (int l = i; l >= max(0, i - c[j] + 1); l--) {
				if (l + c[j] <= n && prf[l + c[j]] - prf[l] == 0) {
					bool vl = 0, vr = 0;
					if (j == 0) {
						vl = prfdp[l][j];
					} else if (l - 1 >= 0 && s[l - 1] != 'X') {
						vl = prfdp[l - 1][j];
					}
					if (j + 1 == k) {
						vr = sufdp[l + c[j]][j + 1];
					} else if (l + c[j] + 1 <= n && s[l + c[j]] != 'X') {
						vr = sufdp[l + c[j] + 1][j + 1];
					}
					ok[1] |= vl && vr;
				}
			}
		}
		for (int j = 0; j <= k; j++) {
			ok[0] |= prfdp[i][j] && sufdp[i + 1][j];
		}
		if (ok[0] && ok[1]) {
			ans[i] = '?';
		} else if (ok[0]) {
			ans[i] = '_';
		} else if (ok[1]) {
			ans[i] = 'X';
		} else {
			assert(0);
		}
	}
	return ans;
}
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
8 Correct 1 ms 4444 KB n = 20, m = 5
9 Correct 1 ms 4444 KB n = 18, m = 3
10 Correct 1 ms 4444 KB n = 17, m = 2
11 Correct 1 ms 4444 KB n = 20, m = 2
12 Correct 1 ms 4444 KB n = 17, m = 4
13 Correct 1 ms 4444 KB n = 17, m = 6
14 Correct 1 ms 4444 KB n = 17, m = 1
15 Correct 1 ms 4444 KB n = 17, m = 4
16 Correct 1 ms 4440 KB n = 13, m = 3
17 Correct 1 ms 4444 KB n = 18, m = 4
18 Correct 1 ms 4444 KB n = 20, m = 10
19 Correct 1 ms 4444 KB n = 19, m = 10
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
8 Correct 1 ms 4444 KB n = 20, m = 5
9 Correct 1 ms 4444 KB n = 18, m = 3
10 Correct 1 ms 4444 KB n = 17, m = 2
11 Correct 1 ms 4444 KB n = 20, m = 2
12 Correct 1 ms 4444 KB n = 17, m = 4
13 Correct 1 ms 4444 KB n = 17, m = 6
14 Correct 1 ms 4444 KB n = 17, m = 1
15 Correct 1 ms 4444 KB n = 17, m = 4
16 Correct 1 ms 4440 KB n = 13, m = 3
17 Correct 1 ms 4444 KB n = 18, m = 4
18 Correct 1 ms 4444 KB n = 20, m = 10
19 Correct 1 ms 4444 KB n = 19, m = 10
20 Correct 1 ms 4440 KB n = 100, m = 5
21 Correct 1 ms 4440 KB n = 90, m = 3
22 Correct 1 ms 4444 KB n = 86, m = 2
23 Correct 1 ms 4444 KB n = 81, m = 4
24 Correct 1 ms 4444 KB n = 89, m = 10
25 Correct 1 ms 4444 KB n = 81, m = 23
26 Correct 1 ms 4444 KB n = 86, m = 8
27 Correct 1 ms 4444 KB n = 53, m = 22
28 Correct 1 ms 4444 KB n = 89, m = 35
29 Correct 1 ms 4440 KB n = 63, m = 25
30 Correct 1 ms 4444 KB n = 100, m = 50
31 Correct 1 ms 4444 KB n = 99, m = 50
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
8 Correct 1 ms 4444 KB n = 20, m = 5
9 Correct 1 ms 4444 KB n = 18, m = 3
10 Correct 1 ms 4444 KB n = 17, m = 2
11 Correct 1 ms 4444 KB n = 20, m = 2
12 Correct 1 ms 4444 KB n = 17, m = 4
13 Correct 1 ms 4444 KB n = 17, m = 6
14 Correct 1 ms 4444 KB n = 17, m = 1
15 Correct 1 ms 4444 KB n = 17, m = 4
16 Correct 1 ms 4440 KB n = 13, m = 3
17 Correct 1 ms 4444 KB n = 18, m = 4
18 Correct 1 ms 4444 KB n = 20, m = 10
19 Correct 1 ms 4444 KB n = 19, m = 10
20 Correct 1 ms 4440 KB n = 100, m = 5
21 Correct 1 ms 4440 KB n = 90, m = 3
22 Correct 1 ms 4444 KB n = 86, m = 2
23 Correct 1 ms 4444 KB n = 81, m = 4
24 Correct 1 ms 4444 KB n = 89, m = 10
25 Correct 1 ms 4444 KB n = 81, m = 23
26 Correct 1 ms 4444 KB n = 86, m = 8
27 Correct 1 ms 4444 KB n = 53, m = 22
28 Correct 1 ms 4444 KB n = 89, m = 35
29 Correct 1 ms 4440 KB n = 63, m = 25
30 Correct 1 ms 4444 KB n = 100, m = 50
31 Correct 1 ms 4444 KB n = 99, m = 50
32 Correct 1 ms 4440 KB n = 13, m = 4
33 Correct 1 ms 4444 KB n = 86, m = 2
34 Correct 1 ms 4440 KB n = 88, m = 2
35 Correct 1 ms 4444 KB n = 86, m = 2
36 Correct 1 ms 4444 KB n = 81, m = 6
37 Correct 1 ms 4444 KB n = 98, m = 7
38 Correct 1 ms 4444 KB n = 92, m = 7
39 Correct 1 ms 4696 KB n = 88, m = 21
40 Correct 1 ms 4444 KB n = 90, m = 21
41 Correct 1 ms 4444 KB n = 98, m = 22
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
8 Correct 1 ms 4444 KB n = 20, m = 5
9 Correct 1 ms 4444 KB n = 18, m = 3
10 Correct 1 ms 4444 KB n = 17, m = 2
11 Correct 1 ms 4444 KB n = 20, m = 2
12 Correct 1 ms 4444 KB n = 17, m = 4
13 Correct 1 ms 4444 KB n = 17, m = 6
14 Correct 1 ms 4444 KB n = 17, m = 1
15 Correct 1 ms 4444 KB n = 17, m = 4
16 Correct 1 ms 4440 KB n = 13, m = 3
17 Correct 1 ms 4444 KB n = 18, m = 4
18 Correct 1 ms 4444 KB n = 20, m = 10
19 Correct 1 ms 4444 KB n = 19, m = 10
20 Correct 1 ms 4440 KB n = 100, m = 5
21 Correct 1 ms 4440 KB n = 90, m = 3
22 Correct 1 ms 4444 KB n = 86, m = 2
23 Correct 1 ms 4444 KB n = 81, m = 4
24 Correct 1 ms 4444 KB n = 89, m = 10
25 Correct 1 ms 4444 KB n = 81, m = 23
26 Correct 1 ms 4444 KB n = 86, m = 8
27 Correct 1 ms 4444 KB n = 53, m = 22
28 Correct 1 ms 4444 KB n = 89, m = 35
29 Correct 1 ms 4440 KB n = 63, m = 25
30 Correct 1 ms 4444 KB n = 100, m = 50
31 Correct 1 ms 4444 KB n = 99, m = 50
32 Correct 1 ms 4440 KB n = 13, m = 4
33 Correct 1 ms 4444 KB n = 86, m = 2
34 Correct 1 ms 4440 KB n = 88, m = 2
35 Correct 1 ms 4444 KB n = 86, m = 2
36 Correct 1 ms 4444 KB n = 81, m = 6
37 Correct 1 ms 4444 KB n = 98, m = 7
38 Correct 1 ms 4444 KB n = 92, m = 7
39 Correct 1 ms 4696 KB n = 88, m = 21
40 Correct 1 ms 4444 KB n = 90, m = 21
41 Correct 1 ms 4444 KB n = 98, m = 22
42 Correct 1 ms 4440 KB n = 11, m = 2
43 Correct 1 ms 4444 KB n = 11, m = 2
44 Runtime error 5 ms 8824 KB Execution killed with signal 6
45 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
8 Correct 1 ms 4444 KB n = 20, m = 5
9 Correct 1 ms 4444 KB n = 18, m = 3
10 Correct 1 ms 4444 KB n = 17, m = 2
11 Correct 1 ms 4444 KB n = 20, m = 2
12 Correct 1 ms 4444 KB n = 17, m = 4
13 Correct 1 ms 4444 KB n = 17, m = 6
14 Correct 1 ms 4444 KB n = 17, m = 1
15 Correct 1 ms 4444 KB n = 17, m = 4
16 Correct 1 ms 4440 KB n = 13, m = 3
17 Correct 1 ms 4444 KB n = 18, m = 4
18 Correct 1 ms 4444 KB n = 20, m = 10
19 Correct 1 ms 4444 KB n = 19, m = 10
20 Correct 1 ms 4440 KB n = 100, m = 5
21 Correct 1 ms 4440 KB n = 90, m = 3
22 Correct 1 ms 4444 KB n = 86, m = 2
23 Correct 1 ms 4444 KB n = 81, m = 4
24 Correct 1 ms 4444 KB n = 89, m = 10
25 Correct 1 ms 4444 KB n = 81, m = 23
26 Correct 1 ms 4444 KB n = 86, m = 8
27 Correct 1 ms 4444 KB n = 53, m = 22
28 Correct 1 ms 4444 KB n = 89, m = 35
29 Correct 1 ms 4440 KB n = 63, m = 25
30 Correct 1 ms 4444 KB n = 100, m = 50
31 Correct 1 ms 4444 KB n = 99, m = 50
32 Correct 1 ms 4440 KB n = 13, m = 4
33 Correct 1 ms 4444 KB n = 86, m = 2
34 Correct 1 ms 4440 KB n = 88, m = 2
35 Correct 1 ms 4444 KB n = 86, m = 2
36 Correct 1 ms 4444 KB n = 81, m = 6
37 Correct 1 ms 4444 KB n = 98, m = 7
38 Correct 1 ms 4444 KB n = 92, m = 7
39 Correct 1 ms 4696 KB n = 88, m = 21
40 Correct 1 ms 4444 KB n = 90, m = 21
41 Correct 1 ms 4444 KB n = 98, m = 22
42 Correct 1 ms 4440 KB n = 11, m = 2
43 Correct 1 ms 4444 KB n = 11, m = 2
44 Runtime error 5 ms 8824 KB Execution killed with signal 6
45 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 2 ms 4440 KB n = 13, m = 1
2 Correct 2 ms 4444 KB n = 18, m = 1
3 Correct 1 ms 4444 KB n = 17, m = 1
4 Correct 2 ms 4444 KB n = 1, m = 1
5 Correct 1 ms 4440 KB n = 20, m = 1
6 Correct 2 ms 4444 KB n = 20, m = 1
7 Correct 1 ms 4444 KB n = 20, m = 1
8 Correct 1 ms 4444 KB n = 20, m = 5
9 Correct 1 ms 4444 KB n = 18, m = 3
10 Correct 1 ms 4444 KB n = 17, m = 2
11 Correct 1 ms 4444 KB n = 20, m = 2
12 Correct 1 ms 4444 KB n = 17, m = 4
13 Correct 1 ms 4444 KB n = 17, m = 6
14 Correct 1 ms 4444 KB n = 17, m = 1
15 Correct 1 ms 4444 KB n = 17, m = 4
16 Correct 1 ms 4440 KB n = 13, m = 3
17 Correct 1 ms 4444 KB n = 18, m = 4
18 Correct 1 ms 4444 KB n = 20, m = 10
19 Correct 1 ms 4444 KB n = 19, m = 10
20 Correct 1 ms 4440 KB n = 100, m = 5
21 Correct 1 ms 4440 KB n = 90, m = 3
22 Correct 1 ms 4444 KB n = 86, m = 2
23 Correct 1 ms 4444 KB n = 81, m = 4
24 Correct 1 ms 4444 KB n = 89, m = 10
25 Correct 1 ms 4444 KB n = 81, m = 23
26 Correct 1 ms 4444 KB n = 86, m = 8
27 Correct 1 ms 4444 KB n = 53, m = 22
28 Correct 1 ms 4444 KB n = 89, m = 35
29 Correct 1 ms 4440 KB n = 63, m = 25
30 Correct 1 ms 4444 KB n = 100, m = 50
31 Correct 1 ms 4444 KB n = 99, m = 50
32 Correct 1 ms 4440 KB n = 13, m = 4
33 Correct 1 ms 4444 KB n = 86, m = 2
34 Correct 1 ms 4440 KB n = 88, m = 2
35 Correct 1 ms 4444 KB n = 86, m = 2
36 Correct 1 ms 4444 KB n = 81, m = 6
37 Correct 1 ms 4444 KB n = 98, m = 7
38 Correct 1 ms 4444 KB n = 92, m = 7
39 Correct 1 ms 4696 KB n = 88, m = 21
40 Correct 1 ms 4444 KB n = 90, m = 21
41 Correct 1 ms 4444 KB n = 98, m = 22
42 Correct 1 ms 4440 KB n = 11, m = 2
43 Correct 1 ms 4444 KB n = 11, m = 2
44 Runtime error 5 ms 8824 KB Execution killed with signal 6
45 Halted 0 ms 0 KB -