【jzoj4866】【禅与园林艺术】【树状数组】

题目大意

有一个长度为 n 的数列{ai}，为一列树木的美观值。现在有m 次询问，每次给出三个数l，r和P，询问对于所有的l <= l’ <= r’ <= r (a[l’] + a[l’ + 1] + … + a[r’]) mod P的最小值。

解题思路

发现区间长度大于模数时答案为零，所以用个树状数组维护一下就可以了。

code

#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
#define LD double
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a>b)?b:a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const inf=2147483647;
int const maxn=5*1e4,maxp=100;
int n,m,a[maxn+10],t[maxp+10];
void change(int pos){
    int p=pos;
    for(;p<=maxp;){
        t[p]=max(t[p],pos);
        p+=p&(-p);
    }
}
int qury(int p){
    int ans=0;
    for(;p>0;){
        ans=max(ans,t[p]);
        p-=p&(-p);
    }
    return ans;
}
int main(){
    //freopen("psy.in","r",stdin);
    //freopen("psy.out","w",stdout);
    freopen("garden.in","r",stdin);
    freopen("garden.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,n)scanf("%d",&a[i]);
    fo(i,1,m){
        int l,r,p;scanf("%d%d%d",&l,&r,&p);
        if(r-l+2>p)printf("0\n");
        else{
            fo(j,0,p)t[j]=0;
            int sum=0,ans=inf;
            change(1);
            fo(j,l,r){
                sum=(sum+a[j])%p;
                if(qury(sum+1))ans=min(ans,sum+1-qury(sum+1));
                change(sum+1);
            }
            printf("%d\n",ans);
        }

    }
    return 0;
}