#include "plants.h"
//Bismillahirrahmanirrahim
#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx")
#include <bits/stdc++.h>
using namespace std;
typedef long long lo;
typedef pair< lo,lo > PII;
#define fi first
#define se second
#define mp make_pair
#define endl "\n"
#define pb push_back
#define fio() ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
#define FOR for(int i=0;i<n;i++)
#define mid ((start+end)/2)
#define ort ((bas+son)/2)
const lo inf = 1000000000000000000;
const lo KOK = 100000;
const lo LOG = 30;
const lo li = 500005;
const lo mod = 1000000007;
int m,b[li],a[li],flag,t,siz[li],say,vis[li],dep[li],kk;
int cev;
string s;
vector<int> v[li],vv;
inline void dfs(int node,int der){
if(vis[node])return;
vis[node]=say;
dep[node]=der;
for(int i=0;i<(int)v[node].size();i++){
int go=v[node][i];
dfs(go,der+1);
}
}
void init(int k, std::vector<int> r) {
for(int i=0;i<(int)r.size();i++){
if(r[i]==1){
v[(i+1)%(int)r.size()].pb(i);
siz[i]++;
}
}
//~ queue<int>
int n=(int)r.size();
//~ FOR{
//~ if(siz[i]==0){
//~ say++;
//~ dfs(i,0);
//~ }
//~ }
vv=r;
kk=k;
for(int i=0;i<(int)vv.size();i++){
int ind=-1,ind1=-1;
for(int j=0;j<(int)vv.size();j++){
if(vis[j])continue;
if(vv[j]==0 && ind==-1)ind=j;
else if(vv[j]==0 && ind1==-1)ind1=j;
}
if(ind1==-1){
vis[ind]=i+1;
for(int j=0;j<(int)vv.size();j++){
if(((j+kk-1)%(int)vv.size()>=ind && j<ind) || ((j+kk-1)%(int)vv.size()>=ind && j+kk-1>=(int)vv.size()))vv[j]--;
}
continue;
}
if((ind+kk-1)<ind1){
vis[ind]=i+1;
for(int j=0;j<(int)vv.size();j++){
if(((j+kk-1)%(int)vv.size()>=ind && j<ind) || ((j+kk-1)%(int)vv.size()>=ind && j+kk-1>=(int)vv.size()))vv[j]--;
}
}
else{
vis[ind1]=i+1;
for(int j=0;j<(int)vv.size();j++){
if(((j+kk-1)%(int)vv.size()>=ind1 && j<ind1) || ((j+kk-1)%(int)vv.size()>=ind1 && j+kk-1>=(int)vv.size()))vv[j]--;
}
}
}
}
int compare_plants(int x, int y) {
if(vis[x]<vis[y])return 1;
if(vis[x]>vis[y])return -1;
if(vis[x]==vis[y]){
//~ cout<<x<<" : : "<<y<<" : ;"<<vis[x]<<" : : "<<vis[y]<<endl;
return 0;
}
if(vis[x]!=vis[y])return 0;
if(dep[x]>dep[y])return -1;
return 1;
}
Compilation message
plants.cpp: In function 'void init(int, std::vector<int>)':
plants.cpp:53:6: warning: unused variable 'n' [-Wunused-variable]
53 | int n=(int)r.size();
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
9 ms |
12032 KB |
Output is correct |
2 |
Correct |
9 ms |
12032 KB |
Output is correct |
3 |
Correct |
9 ms |
12032 KB |
Output is correct |
4 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
12032 KB |
Output is correct |
2 |
Correct |
8 ms |
12032 KB |
Output is correct |
3 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
12032 KB |
Output is correct |
2 |
Correct |
8 ms |
12032 KB |
Output is correct |
3 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
9 ms |
12032 KB |
Output is correct |
2 |
Correct |
9 ms |
12032 KB |
Output is correct |
3 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
10 ms |
12032 KB |
Output is correct |
2 |
Correct |
9 ms |
12032 KB |
Output is correct |
3 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
9 ms |
12032 KB |
Output is correct |
2 |
Correct |
9 ms |
12032 KB |
Output is correct |
3 |
Correct |
9 ms |
12032 KB |
Output is correct |
4 |
Incorrect |
9 ms |
12032 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |