2019icpc南京网络赛_F_Greedy Sequence

题意

题意不明,队友告诉我对于每个\(i\),所在下标\(p[i]\),在\([p[i]-k,p[i]+k]\)中找到小于\(i\)的最大数\(x\),然后\(ans[i]=ans[x]+1\)即可。

分析

  • 第一种方法无脑主席树,求区间小于某个值的最大数。
  • 第二种方法是线段树,因为对于每个数\(i\),只有比他小的数才有用,所以从小到大枚举,在线段树中(此时所有值都小于\(i\)),查询区间最大值即可。

代码

code1

#include 
using namespace std;
const int N=1e5+50;
struct CT{
#define mid (l+r)/2
    int tot,sum[N*30],ls[N*30],rs[N*30];
    void init(){
        tot=0;
    }
    int build(int l,int r){
        int rt=++tot;
        sum[rt]=ls[rt]=rs[rt]=0;
        if(l=r){
            if(l

code2

#include 
using namespace std;
#define ls i<<1
#define rs i<<1|1
#define mid (l+r)/2
const int N=1e5+50;
int ans[N];
int T,n,k,a[N],p[N];
int mx[N*4];
void pushup(int i){
    mx[i]=max(mx[ls],mx[rs]);
}
void build(int i,int l,int r){
    mx[i]=0;
    if(l==r){
        return;
    }
    build(ls,l,mid);
    build(rs,mid+1,r);
    pushup(i);
}
int query(int i,int l,int r,int ql,int qr){
    if(ql<=l && qr>=r){
        return mx[i];
    }
    int ans=0;
    if(ql<=mid){
        ans=max(ans,query(ls,l,mid,ql,qr));
    }
    if(qr>mid){
        ans=max(ans,query(rs,mid+1,r,ql,qr));
    }
    return ans;
}
void update(int i,int l,int r,int p,int v){
    if(l==r){
        mx[i]=v;
        return;
    }
    if(p<=mid){
        update(ls,l,mid,p,v);
    }else{
        update(rs,mid+1,r,p,v);
    }
    pushup(i);
}
int main(){
//    freopen("in.txt","r",stdin);
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            p[a[i]]=i;
        }
        build(1,1,n);
        for(int i=1;i<=n;i++){
            int L=max(1,p[i]-k);
            int R=min(n,p[i]+k);
            int x=query(1,1,n,L,R);
            update(1,1,n,p[i],i);
            ans[i]=ans[x]+1;
        }
        for(int i=1;i<=n;i++){
            printf("%d%c",ans[i],i==n?'\n':' ');
        }
    }
    return 0;
}

你可能感兴趣的:(2019icpc南京网络赛_F_Greedy Sequence)