Gym 101472G/UVALive 6054(树状数组或者更简单的方法?)

这题用树状数组还算是非常清晰明了的,可惜我竟不会直接暴力……

每次从一个点i开始搜到长度为h的一段,然后吧,每一个点和i进行比较,一旦出现一个小于i的点就break,否则就进行单点更新,把i这个点++,和i比较的这个点--(前提是该点值是大于0的,否则跳过)

最后查询一下区间和即可。

代码:

#include<cstdio>
#include<cstring>
#include<string>
#include<string.h>
#include<queue>
#include<algorithm>
#include<iostream>
#include<math.h>
#define maxn 20000
#define ll long long
#define RPG(i,a,b) for(int i=(a);i<(b);i++)
using namespace std;
int C[maxn];
int n,h,b,e;
int a[maxn];

int Lowbit(int x){
  return (x&(-x));
}
int sum(int x){
    int ret=0;
    while(x>0){
        ret+=C[x];x-=Lowbit(x);
    }
    return ret;
}
void add(int x,int d){
    while(x<=n){
        C[x]+=d;
        x+=Lowbit(x);
    }
}

int main(){
    freopen("g.in","r",stdin);
    while(~scanf("%d%d%d%d",&n,&h,&b,&e)){
        memset(C,0,sizeof(C));
        RPG(i,1,n+1){
            scanf("%d",&a[i]);
        }
        RPG(i,1,n+1)
            add(i,1);

        RPG(i,1,n+1){
            RPG(j,i+1,min(i+h,n+1)){
                if(j>n)
                    break;
                if(a[i]<a[j]){
                    if(sum(j)-sum(j-1)>0){
                        add(i,1);
                        add(j,-1);
                    }
                }
                else{
                    break;
                }
            }
        }
        RPG(i,b,e+1){
            if(i!=b)
                printf(" ");
            printf("%d",max(sum(i)-sum(i-1),0));
        }
       printf("\n");
    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/pipitongkw1/article/details/82157008