#include "plants.h"
#include<bits/stdc++.h>
using namespace std;
const int maxn = 200005;
#define REP(i,n) for(int i=0;i<(n);i++)
#define eb emplace_back
#define pb push_back
#define SZ(a) ((int)a.size())
#define ALL(a) a.begin(),a.end()
const int INF = 0x3f3f3f3f;
int n,k;
bool adj[maxn][maxn];
set<int> zero;
set<int> st;
vector<int> a;
vector<int> _a;
vector<int> b;
int seg[maxn*4]; // store the minimum
int tg[maxn*4];
void push(int i){
seg[i*2]+=tg[i];
tg[i*2]+=tg[i];
seg[i*2+1]+=tg[i];
tg[i*2+1]+=tg[i];
tg[i]=0;
}
void mdf(int i,int l,int r,int ql,int qr, int delta){
if(l>=qr||ql>=r) return;
if(ql<=l&&r<=qr){
seg[i]+=delta;
tg[i]+=delta;
return;
}
int mid=(l+r)/2;
push(i);
mdf(i*2,l,mid,ql,qr,delta);
mdf(i*2+1,mid,r,ql,qr,delta);
seg[i] = min(seg[i*2],seg[i*2+1]);
}
void reset(int i,int l,int r){
seg[i]=tg[i]=0;
if(l!=r-1){
int mid=(l+r)/2;
reset(i*2,l,mid);
reset(i*2+1,mid,r);
}
}
int get_idx(int i,int l,int r){
if(l==r-1) return l;
int mid=(l+r)/2;
push(i);
if(seg[i*2]==seg[i]) return get_idx(i*2,l,mid);
else return get_idx(i*2+1,mid,r);
}
void add(int i){ // adding i to zero while maintaining st
if(zero.count(i))return;
// cout<<"adding: "<<i<<endl;
auto it = zero.lower_bound(i);
if(SZ(zero)==0){
zero.insert(i);
}
else if(it==zero.end()){
int pre = *prev(zero.end());
if(i-pre>=k) st.insert(i);
}
else if(it==zero.begin()){
if((*it) - i >=k){
st.insert((*zero.begin()));
}
}
else{
int pre = *prev(it);
int nxt = *it;
if(nxt-pre>=k){
st.erase(nxt);
}
if(i-pre>=k) st.insert(i);
if(nxt - i >=k){
st.insert(nxt);
}
}
zero.insert(i);
}
void rm(int i){ // removing i to zero while maintaining st
if(!zero.count(i)) return;
// cout<<"removing: "<<i<<endl;
zero.erase(i);
auto it = zero.lower_bound(i);
if(it==zero.end()){
st.erase(i);
}
else if(it==zero.begin()){
st.erase(*it);
}
else{
int pre = *prev(it);
int nxt = *it;
if(i-pre>=k) st.erase(i);
if(nxt - i >=k){
st.erase(nxt);
}
if(nxt-pre>=k){
st.insert(nxt);
}
}
}
void init(int _k, std::vector<int> _r) {
k=_k;
n = _r.size();
{
zero.clear();
st.clear();
vector<int> r = _r;
REP(j,n){
if(r[j]==0) zero.insert(j),zero.insert(j+n);
}
{
int pre = -1;
for(int i:zero){
if(pre!=-1&&i-pre>=k){
st.insert(i);
}
pre = i;
}
}
REP(i,n){
mdf(1,0,n,i,i+1,(r[i])?r[i]:INF);
}
REP(i,n){
int tmp = *st.begin();
tmp%=n;
// cout<<"found: "<<tmp<<endl;
a.pb(tmp);
st.erase(tmp);
st.erase(tmp+n);
rm(tmp);
rm(tmp+n);
int l = (tmp-(k-1)+n)%n, r = tmp;
if(l<r){
mdf(1,0,n,l,r,-1);
}
else{
mdf(1,0,n,l,n,-1);
mdf(1,0,n,0,r,-1);
}
while(seg[1] == 0){
int cur = get_idx(1,0,n);
// cout<<"new: "<<cur<<endl;
add(cur);
add(cur+n);
mdf(1,0,n,cur,cur+1,INF);
}
// cout<<"zero: ";
// for(int j:zero) cout<<j<<' ';
// cout<<'\n';
// cout<<"st: ";
// for(int j:st) cout<<j<<' ';
// cout<<'\n';
}
// REP(i,n) cout<<a[i]<<" \n"[i==n-1];
// cout<<endl;
}
{
reset(1,0,n);
zero.clear();
st.clear();
vector<int> r = _r;
REP(i,n) r[i] = (k-1-r[i]);
// REP(i,n) cout<<r[i]<<" \n"[i==n-1];
REP(j,n){
if(r[j]==0) zero.insert(j),zero.insert(j+n);
}
{
int pre = -1;
for(int i:zero){
if(pre!=-1&&i-pre>=k){
st.insert(i);
}
pre = i;
}
}
REP(i,n){
mdf(1,0,n,i,i+1,(r[i])?r[i]:INF);
}
REP(i,n){
int tmp = *st.begin();
tmp%=n;
// cout<<"found: "<<tmp<<endl;
b.pb(tmp);
st.erase(tmp);
st.erase(tmp+n);
rm(tmp);
rm(tmp+n);
int l = (tmp-(k-1)+n)%n, r = tmp;
if(l<r){
mdf(1,0,n,l,r,-1);
}
else{
mdf(1,0,n,l,n,-1);
mdf(1,0,n,0,r,-1);
}
while(seg[1] == 0){
int cur = get_idx(1,0,n);
// cout<<"new: "<<cur<<endl;
add(cur);
add(cur+n);
mdf(1,0,n,cur,cur+1,INF);
}
// cout<<"zero: ";
// for(int j:zero) cout<<j<<' ';
// cout<<'\n';
// cout<<"st: ";
// for(int j:st) cout<<j<<' ';
// cout<<'\n';
}
// REP(i,n) cout<<b[i]<<" \n"[i==n-1];
}
// {
// bool vis[maxn]={0};
// for(int i:a){
// for(int j=i+1;j<i+k;j++){
// // cout<<i<<' '<<j<<endl;
// int t = j%n;
// if(!vis[t]){
// adj[i][t]=1;
// }
// }
// vis[i]=1;
// }
// }
// {
// bool vis[maxn]={0};
// for(int i:b){
// for(int j=i+1;j<i+k;j++){
// int t = j%n;
// if(!vis[t]){
// adj[t][i]=1;
// }
// }
// vis[i]=1;
// }
// }
_a.resize(n);
REP(i,n){
_a[a[i]]=i;
}
// REP(i,n) REP(j,n) cout<<adj[i][j]<<" \n"[j==n-1];
// REP(j,n) REP(i,n) REP(k,n){
// if(adj[i][j]&&adj[j][k]) adj[i][k] = 1;
// }
return;
}
int compare_plants(int x, int y) {
if(_a[x]<_a[y]) return 1;
else if(_a[y]<_a[x]) return -1;
else return 0;
}
Compilation message
/tmp/cc80Z0eP.o: In function `add(int)':
plants.cpp:(.text+0x3a5): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x46d): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x4ae): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x4ce): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
/tmp/cc80Z0eP.o: In function `rm(int)':
plants.cpp:(.text+0x5b4): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x63d): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x661): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
/tmp/cc80Z0eP.o: In function `init(int, std::vector<int, std::allocator<int> >)':
plants.cpp:(.text+0x6b8): relocation truncated to fit: R_X86_64_PC32 against symbol `k' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x6ce): relocation truncated to fit: R_X86_64_PC32 against symbol `n' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x794): relocation truncated to fit: R_X86_64_PC32 against symbol `n' defined in .bss section in /tmp/cc80Z0eP.o
plants.cpp:(.text+0x7b8): additional relocation overflows omitted from the output
collect2: error: ld returned 1 exit status
