제출 #1132148

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1132148		StefanSebez	식물 비교 (IOI20_plants)	C++20	14 / 100	4094 ms	11212 KiB

plants

#include <bits/stdc++.h>
#include "plants.h"
using namespace std;
#define fi first
#define se second
#define pb push_back
#define ll long long
#define ld long double
const int N=2e5+50,inf=1e8;
int k,n;
int pref[2*N],r[2*N],a[N];
int Plus(int i,int x){i+=x;while(i<1)i+=n;while(i>n)i-=n;return i;}
int suma(int l,int r){
    if(l<=r) return pref[r]-pref[l-1];
    return pref[r+n]-pref[l-1];
}
int dist(int l,int r){
    if(l<=r) return r-l+1;
    return n+(r-l+1);
}
struct SegTree{
    int root,nc,lc[2*N],rc[2*N],mn[2*N];
    SegTree(){mn[0]=inf;}
    void Update(int &c,int ss,int se,int qind,int qval){
        if(!c)c=++nc;
        if(ss==se){mn[c]+=qval;return;}
        int mid=ss+se>>1;
        if(qind<=mid)Update(lc[c],ss,mid,qind,qval);
        else Update(rc[c],mid+1,se,qind,qval);
        mn[c]=min(mn[lc[c]],mn[rc[c]]);
    }
    int Get(int c,int ss,int se,int qs,int qe){
        if(qs>qe||se<qs||qe<ss)return inf;
        if(qs<=ss&&se<=qe)return mn[c];
        int mid=ss+se>>1;return min(Get(lc[c],ss,mid,qs,qe),Get(rc[c],mid+1,se,qs,qe));
    }
}ST;
void init(int k1, std::vector<int> r1){
    k=k1;n=r1.size();
    for(int i=1;i<=n;i++) r[i]=r1[i-1];
    for(int i=n+1;i<=2*n;i++) r[i]=r[i-n];
    bool was[n+10]={false};
    for(int i=1;i<=n;i++) ST.Update(ST.root,1,n,i,r[i]);
    vector<int>nule;for(int i=1;i<=n;i++) if(r[i]==0) nule.pb(i);
    for(int c=n;c>=1;c--){
        int ind=-1,maks=0,idx=0;
        for(int I=0;I<nule.size();I++){
            int i=nule[I];
            if(was[i]||r[i]!=0) continue;
            int nesto=inf;
            /*for(int j=Plus(i,-1),ct=k-1;ct--;j=Plus(j,-1)){
                if(was[j]) continue;
                nesto=min(nesto,r[j]);
            }*/
            int levo=Plus(i,-k+1),desno=Plus(i,-1);
            if(levo<=desno) nesto=ST.Get(ST.root,1,n,levo,desno);
            else nesto=min(ST.Get(ST.root,1,n,1,desno),ST.Get(ST.root,1,n,levo,n));
            //printf("*%i\n",nesto);
            if(nesto>maks){
                maks=nesto,ind=i,idx=I;;
                break;
            }
        }
        //printf("%i\n",ind);
        swap(nule[idx],nule.back());
        nule.pop_back();
        a[ind]=c;
        was[ind]=true;
        for(int j=Plus(ind,-1),ct=k-1;ct--;j=Plus(j,-1)){
            if(was[j]) continue;
            r[j]--;if(r[j]==0) nule.pb(j);
            ST.Update(ST.root,1,n,j,-1);
        }
        ST.Update(ST.root,1,n,ind,inf);
        //for(int i=1;i<=n;i++) printf("%i ",r[i]);printf("\n");
    }
    //for(int i=1;i<=n;i++) printf("%i ",a[i]);printf("\n");
}
int compare_plants(int x, int y){
    x++,y++;
    if(a[x]>a[y]) return 1;
    if(a[x]<a[y]) return -1;
    return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 14 / 14
• Subtask 3: 0 / 13
• Subtask 4: 0 / 17
• Subtask 5: 0 / 11
• Subtask 6: 0 / 15
• Subtask 7: 0 / 25