查询数组中能被k整除且不小于z的连续子序列和的个数

题目：小圆前辈的数组

小圆前辈最近收到了一个长度为$n$数组。她怀疑是不怀好意的魔女给她的陷阱，于是她对数组进行了剖析后发现了两个关键的整数$k$和$z$，而解读此数组只要算出的所有连续子序列中有多少满足：

1，所有数的和为$k$的倍数；
2，且其和至少为$z$；

这个问题难到了小圆前辈，她便把这个问题交给了你，如果你能帮她解决的话，她将奖励你一个Accept。

输入描述:

第一行只有三个整数$n，k，z$。
第一行共n个整数$a[1]...a[n]$。

输出描述:

一个整形数表示答案。

题解：

处理前缀和$pre[i]$，由于是要找所有数和为$k$的倍数的数，我们对于每个$pre[i]\mathrm{m}\mathrm{o}\mathrm{d}k$，当$pre[i]\equiv pre[j](\mathrm{m}\mathrm{o}\mathrm{d}k)$，$i<j$时，有$({\sum }_i^{j}a[i])=0$。及区间$[l,r]$的和为$k$的倍数。这就能找到满足第一个条件的区间。

要满足第二个条件，我们要将$pre[i](\mathrm{m}\mathrm{o}\mathrm{d}k)$相等的放进$vector$里面二分，即枚举$pre[i]$，二分查找，有多少个数小于等于$pre[i]+z$ ，即满足子序列和大于等于$z$。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
const int N=1e5+5;

int n,k,z;
int a[N];

long long pre[N];
vector<ll> mk[N];

void solve()
{
    cin>>n>>k>>z;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        pre[i]=pre[i-1]+a[i];
        mk[pre[i]%k].push_back(pre[i]);
    }
    ll ans=0;
    for(auto x:mk[0])if(x>=z)ans++;
    for(int i=0;i<k;i++)
    {
        int l=mk[i].size(),r;
        for(int j=0;j<l;j++)
        {
            r=lower_bound(mk[i].begin(),mk[i].end(),mk[i][j]+z)-mk[i].begin();
            ans+=l-r;
        }
    }
    cout<<ans<<endl;
}

int main()
{
	ios::sync_with_stdio(0);
    cin.tie(nullptr);
	int t=1;
	// cin>>t;
	while(t--)solve();
	return 0;
}