【JZOJ A组】简单的区间

Description

Input

第一行两个正整数 n 和 k 。
第二行包含 n 个正整数，第 i 个正整数表示 ai。

Output

一行一个正整数，表示答案。

Sample Input

【样例 1 输入】
4 3
1 2 3 4
【样例 2 输入】
4 2
4 4 7 4

Sample Output

【样例 1 输出】
3
【样例 2 输出】
6

Data Constraint

对于 30% 的数据，n ≤ 3000；
对于另外 20% 的数据，数列 a 为随机生成；
对于 100% 的数据，1 ≤ n ≤ 3 × 10^5 , 1 ≤ k ≤ 10^6 , 1 ≤ ai ≤ 10^9。

思路

考虑分治
题解说要找到[l–r]中的最大值K然后分治，结果我二分mid就过了。

我们可以暴力求出二分的右端区间，然后暴力求出sum（mid+1–i的前缀和），mx（mid+1–i的最大值）

然后考虑左区间的合法数量，可以记录一个f[i][0/1]表示在[l,r]区间里i出现过的次数，和有没有越过mid
最后就求出f数组，更新答案

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=3e5+77;
int n,k,cnt,a[maxn],f[1000077][2],pos[maxn],mx[maxn],sum[maxn];
long long ans;
void fin(int l,int r)
{
    if (l==r) return;
    int mid=(l+r)>>1;
    cnt=sum[mid]=mx[0]=0;
    for(int i=mid+1; i<=r; i++)
    {
        if (a[i]>a[mx[cnt]]) mx[++cnt]=i;
        sum[i]=(sum[i-1]+a[i]%k)%k;
        f[(sum[i]-a[mx[cnt]]%k+k)%k][0]++;
        pos[i]=mx[cnt];
    }
    mx[cnt+1]=r+1;
    int p=1,k1=mid+1,mxl=0,suml=0;
    for(int i=mid; i>=l; i--)
    {
        suml=(suml+a[i]%k)%k;
        mxl=max(mxl,a[i]);
        while(p<=cnt&&a[mx[p]]<=mxl) p++;
        while(k1<mx[p]) f[(sum[k1]-a[pos[k1]]%k+k)%k][0]--,f[sum[k1]][1]++,k1++;
        ans+=f[(k+mxl%k-suml)%k][1];
        if(p<=cnt) ans+=f[(k-suml)%k][0];
    }
    for(int i=mid+1;i<k1;i++) f[sum[i]][1]--;
    for(int i=k1;i<=r;i++) f[(sum[i]-a[pos[i]]%k+k)%k][0]--;
    fin(l,mid),fin(mid+1,r);
}
int main()
{
    freopen("interval.in","r",stdin); freopen("interval.out","w",stdout);
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++) scanf("%lld",&a[i]);
    fin(1,n);
    printf("%lld",ans);
}