0
点赞
收藏
分享

微信扫一扫

树状数组维护前缀和

陬者 2022-04-02 阅读 66
算法c++

在这里插入图片描述

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define int long long
const int N = 400010;
int s[N];
int tr[N];
int n,m;
vector<int>nums;
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int v)
{
    for(int i=x;i<N;i+=lowbit(i))tr[i]+=v;
}
int query(int x)
{
    int res=0;
    for(int i=x;i>=1;i-=lowbit(i))res+=tr[i];
    return res;
}
int get(int x)
{
    return lower_bound(nums.begin(),nums.end(),x)-nums.begin()+1;
}
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>s[i],s[i]+=s[i-1];
    for(int i=0;i<=n;i++)
    {
        nums.push_back(s[i]);
        nums.push_back(s[i]-m);
    }
    sort(nums.begin(),nums.end());
    nums.erase(unique(nums.begin(),nums.end()),nums.end());
    int res=0;
    for(int i=0;i<=n;i++)
    {
        //cout<<"res +  "<<query(N-1)-query(get(s[i]-m))<<endl;
        //cout<<s[i]<<endl;
        
        res+=query(N-1)-query(get(s[i]-m));
        add(get(s[i]),1);
    }
    cout<<res<<'\n';
}

 

举报

相关推荐

0 条评论