首先我们可以很轻松的想到把做一个前缀和,枚举左端点,然后查询在合法区间内有多少个满足条件的右端点,因为数据结构太菜,比赛时先想了一下分块,但复杂度不太对,大概只能 O ( l o g n ) O(logn) O(logn)的查询,感觉不太可能,以为是什么NB的我不会的数据结构,然后就自闭了。
题解
首先我们把合法的前缀和离散化之后插到树状数组里面,实际上我们要维护的是一个前缀和的后缀和,在枚举的过程中,当左端点加一时 ,对应右端点的可能位置区间向右移了一位,我们加上那个后面的前缀和,同理,减去已经不再合法的前缀和即可(保证每次查询时我们的查询区间都是合法的)。直接枚举统计答案即可。
#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef long long ll;
typedef pair<int,int> PII;
const ll mod=1000000007;
ll powmod(ll a,ll b) {
ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){
if