【JZOJ4866】禅与园林艺术

Description

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

Solution

我们发现 P 很小（P≤100）

对于一个区间，前缀和（模 P ）至多有P种数，根据抽屉原理，那么区间长度大于 P+1 答案必定为0。

还有一个优化，当前答案如果为0就退出。

如果要达到 log 级别的，就用数据结构查询。

Code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define N 50010
#define P 101
#define ll long long
using namespace std;
int a[N];
ll s[N];
int main()
{
    freopen("garden.in","r",stdin);
    freopen("garden.out","w",stdout);
    int n,m;
    cin>>n>>m;
    fo(i,1,n) scanf("%d",&a[i]),s[i]=s[i-1]+a[i];
    while(m--)
    {
        int l,r,p;
        scanf("%d %d %d",&l,&r,&p);
        if(r-l>p)
        {
            printf("0\n");
            continue;
        } 
        ll ans=p;
        fo(i,l,r)
        {
            fo(j,i,r)
            {
                ans=min(ans,(s[j]-s[i-1])%p);
                if(!ans) break;
            }
            if(!ans) break;
        }
        printf("%lld\n",ans);
    }
}