#include "rainbow.h"
#include <bits/stdc++.h>
using namespace std;
// #define int long long
#define double long double
using pii = pair<int, int>;
template<typename T>
using prior = std::priority_queue<T, vector<T>, greater<T>>;
template<typename T>
using Prior = std::priority_queue<T>;
#define X first
#define Y second
#define ALL(x) (x).begin(), (x).end()
#define eb emplace_back
#define pb push_back
#define fastIO() ios_base::sync_with_stdio(false), cin.tie(0)
const int maxn = 1 << 18;
int arr[5][maxn];
vector<pii> vec[5];
void init(int R, int C, int sr, int sc, int M, char *S) {
arr[sr][sc] = 1;
for (int i = 0; i < M; ++i) {
if (S[i] == 'N') --sr;
if (S[i] == 'S') ++sr;
if (S[i] == 'W') --sc;
if (S[i] == 'E') ++sc;
arr[sr][sc] = 1;
}
int lst1 = 0, lst2 = 0;
for (int i = 0; i <= C; ++i) {
if ( arr[1][i] and !arr[1][i + 1]) lst1 = i + 1;
if (!arr[1][i] and arr[1][i + 1]) vec[1].eb(lst1, i);
if ( arr[2][i] and !arr[2][i + 1]) lst2 = i + 1;
if (!arr[2][i] and arr[2][i + 1]) vec[2].eb(lst2, i);
}
vec[1].eb(lst1, C + 1);
vec[2].eb(lst2, C + 1);
}
int colour(int ar, int ac, int br, int bc) {
int cnt1L, cnt1R, cnt2L, cnt2R;
if (lower_bound(ALL(vec[1]), pii{ac, 0}) != vec[1].begin() and
(*prev(lower_bound(ALL(vec[1]), pii{ac, 0}))).Y >= ac) {
cnt1L = lower_bound(ALL(vec[1]), pii{ac, 0}) - vec[1].begin() - 1;
}
else {
cnt1L = lower_bound(ALL(vec[1]), pii{ac, 0}) - vec[1].begin();
}
cnt1R = upper_bound(ALL(vec[1]), pii{bc, maxn}) - vec[1].begin() - 1;
if (lower_bound(ALL(vec[2]), pii{ac, 0}) != vec[2].begin() and
(*prev(lower_bound(ALL(vec[2]), pii{ac, 0}))).Y >= ac) {
cnt2L = lower_bound(ALL(vec[2]), pii{ac, 0}) - vec[2].begin() - 1;
}
else {
cnt2L = lower_bound(ALL(vec[2]), pii{ac, 0}) - vec[2].begin();
}
cnt2R = upper_bound(ALL(vec[2]), pii{bc, maxn}) - vec[2].begin() - 1;
// cout << cnt1L << " " << cnt1R << " " << cnt2L << " " << cnt2R << " ";
int ans = (ar == 1 ? cnt1R - cnt1L + 1 : 0LL) + (br == 2 ? cnt2R - cnt2L + 1 : 0LL);
if (ar == 1 and br == 2) {
ans -= cnt1L == 0 and cnt2L == 0;
ans -= cnt1R == vec[1].size() - 1 and cnt2R == vec[2].size() - 1;
}
return ans;
}
Compilation message
rainbow.cpp: In function 'int colour(int, int, int, int)':
rainbow.cpp:72:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
72 | ans -= cnt1R == vec[1].size() - 1 and cnt2R == vec[2].size() - 1;
| ~~~~~~^~~~~~~~~~~~~~~~~~~~
rainbow.cpp:72:53: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
72 | ans -= cnt1R == vec[1].size() - 1 and cnt2R == vec[2].size() - 1;
| ~~~~~~^~~~~~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1 ms |
384 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Correct |
0 ms |
384 KB |
Output is correct |
3 |
Correct |
83 ms |
1672 KB |
Output is correct |
4 |
Correct |
110 ms |
4856 KB |
Output is correct |
5 |
Correct |
120 ms |
4856 KB |
Output is correct |
6 |
Correct |
127 ms |
4596 KB |
Output is correct |
7 |
Correct |
103 ms |
4724 KB |
Output is correct |
8 |
Correct |
72 ms |
3960 KB |
Output is correct |
9 |
Correct |
107 ms |
4728 KB |
Output is correct |
10 |
Correct |
117 ms |
4860 KB |
Output is correct |
11 |
Correct |
120 ms |
4656 KB |
Output is correct |
12 |
Correct |
68 ms |
4344 KB |
Output is correct |
13 |
Correct |
76 ms |
4728 KB |
Output is correct |
14 |
Correct |
83 ms |
4856 KB |
Output is correct |
15 |
Correct |
82 ms |
4548 KB |
Output is correct |
16 |
Correct |
72 ms |
4216 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
384 KB |
Output is correct |
2 |
Runtime error |
2 ms |
768 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1 ms |
384 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1 ms |
384 KB |
Execution killed with signal 11 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |