阴晴不定的大橘学长

题目链接：阴晴不定的大橘学长

思路：离散化+树状数组

首先求出他们的前缀和，然后再开一个数组也保存前缀和，对该数组排序，然后一个个枚举前缀和，查找小于等于pre[i]-x 有多少数，最后加上该位置。别忘了本身如果大于等于x也要加上。

代码：

#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f;
const int N = 2e5+5;
int n,k;
int a[N],b[N];
int tr[N*4];

int lowbit(int x){
    return x&(-x);
}

void add(int x){
    for(int i=x;i<N;i+=lowbit(i)){
        tr[i] ++;
    }
    return;
}

int ask(int x){
    int res = 0;
    for(int i=x;i>=1;i-=lowbit(i)){
        res += tr[i];
    }
    return res;
}

void solve(){
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        a[i]+=a[i-1];
        b[i]=a[i];
    }

    sort(b+1,b+1+n);
    int ans = 0;

    for(int i=1;i<=n;i++){
        if(a[i]>=k)ans++;
        int q = upper_bound(b+1,b+1+n,a[i]-k) -b-1;
        ans += ask(q);
        q = upper_bound(b+1,b+1+n,a[i])-b-1;
        add(q);
    }

    cout<<ans<<'\n';

}

signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t=1;
    while(t--){
        solve();
    }

    return 0;
}