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);
}