题目大意
有一个长度为n的数组a,现在要找一个长度至少为2的子段,求出这一子段的和,然后减去最大值,然后对k取余结果为0。
问这样的子段有多少个。
笛卡尔树
首先有一个想法,就是建立笛卡尔树。
那么对于一个节点,其作为最大值的区间是它的子树。
接下来我们考虑如何计算包含i的有多少个合法区间。
对于包含i的[l,r],sum[r]-sum[l-1]-a[i]是k的倍数。
我们全部模k,就会变成sum[r]=sum[l-1]+a[i]或者sum[l-1]=sum[r]-a[i]
考虑枚举一个l或者r,例如枚举l,然后在右边区间查找对应数值的位置有多少个。
区间查找可以随便做,因为无修改可以考虑对每个权值建vector,然后用lower_bound和upper_bound。
现在考虑枚举的复杂度,对于i的左右子树,枚举size较小的一方,那么复杂度应该和启发式合并一致,枚举的复杂度是n log n。
#include<cstdio>
#include<algorithm>
#include<vector>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=300000+10;
vector<int> b[1000000+10];
int a[maxn],left[maxn],right[maxn],s[maxn],sum[maxn];
int i,j,k,l,t,n,m,top;
ll ans;
int read(){
int x=0;
char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9'){
x=x*10+ch-'0';
ch=getchar();
}
return x;
}
int find(int l,int r,int v){
int fi=upper_bound(b[v].begin(),b[v].end(),r)-b[v].begin()-1;
int se=lower_bound(b[v].begin(),b[v].end(),l)-b[v].begin()-1;
return fi-se;
}
int main(){
n=read();k=read();
fo(i,1,n){
a[i]=read();
sum[i]=(sum[i-1]+a[i])%k;
}
fo(i,1,n){
while (top&&a[i]>=a[s[top]]) top--;
left[i]=s[top];
s[++top]=i;
}
top=0;
s[0]=n+1;
fd(i,n,1){
while (top&&a[i]>a[s[top]]) top--;
right[i]=s[top];
s[++top]=i;
}
fo(i,0,n)
b[sum[i]].push_back(i);
fo(i,1,n)
if (i-left[i]<right[i]-i){
fo(j,left[i]+1,i){
t=(sum[j-1]+a[i]%k)%k;
ans+=(ll)find(i,right[i]-1,t);
}
}
else{
fo(j,i,right[i]-1){
t=((sum[j]-a[i]%k)%k+k)%k;
ans+=(ll)find(left[i],i-1,t);
}
}
printf("%I64d\n",ans-n);
}