#include <bits/stdc++.h>
#include "plants.h"
using namespace std;
typedef long long ll;
vector<ll> lnk;
vector<int> a;
ll find(ll a) {
if (lnk[a] != a) lnk[a] = find(lnk[a]);
return lnk[a];
}
void unite(ll a, ll b) {
a = find(a); b = find(b);
lnk[b] = a;
}
bool same(ll a, ll b) {
return find(a) == find(b);
}
void init(int k, vector<int> r) {
ll n = r.size();
lnk = vector<ll>(n);
for (int i = 0; i < n; i++) lnk[i] = i;
for (int i = n-1; i >= 0; i--) {
int o = (i+1) % n;
if (r[i] == r[o]) unite(i, o);
}
lnk[lnk[0]] = 0;
lnk[0] = 0;
a = vector<int>(n);
for (int i = 0; i < n; i++) a[i] = r[i];
}
int compare_plants(int x, int y) {
ll n = a.size();
ll mul = 1;
if (find(y-1) == 0 && find(x) == 0) { swap(x, y); mul = -1; }
if (same(x, (y-1+n)%n)) {
return (a[x] ? -1 : 1) * mul;
}
if (same((x-1+n)%n, y)) {
return (!a[y] ? -1 : 1) * mul;
}
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
600 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
33 ms |
4184 KB |
Output is correct |
7 |
Incorrect |
45 ms |
5804 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
344 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
344 KB |
Output is correct |
4 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
1 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
600 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
0 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
33 ms |
4184 KB |
Output is correct |
7 |
Incorrect |
45 ms |
5804 KB |
Output isn't correct |
8 |
Halted |
0 ms |
0 KB |
- |