L-阴晴不定的大橘学长

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 4e5+10;
const int mod = 998244353;
const int P=131;
const double PI=acos(-1);
typedef pair<int, int> PII;
typedef pair<int,string> PIS;
#define x first
#define y second
typedef long long LL;
typedef unsigned long long ULL;
#define pb push_back
int dx[] = {-1, 1, 0, 0};
int dy[] = {0, 0, -1, 1};
int tr[N];
int sum[N];
int n;
int lowbit(int x)
{
   return x&(-x);
}
void add(int x,int k)
{
   for(int i=x;i<N;i+=lowbit(i))
   {
       tr[i]+=k;
   }
}
int ask(int x)
{
   int res=0;
   for(int i=x;i>=1;i-=lowbit(i))
   {
      res+=tr[i];
   }
   return res;
}
void solve()
{
   int x;
   cin>>n>>x;
   vector<int>a(n+1);
   for(int i=1;i<=n;i++)cin>>a[i];
   vector<int>find;
   find.push_back(0);
   for(int i=1;i<=n;i++)
   {
      sum[i]=sum[i-1]+a[i];
      find.push_back(sum[i]);
      find.push_back(sum[i]-x);
   }
   sort(find.begin(),find.end());
   find.erase(unique(find.begin(),find.end()),find.end());
   int res=0;
   int pos_0=lower_bound(find.begin(),find.end(),0)-find.begin()+1;
   add(pos_0,1);
   for(int i=1;i<=n;i++)
   {
      int tar=lower_bound(find.begin(),find.end(),sum[i]-x)-find.begin()+1; //找一下sum[i]-x的位置
      int pos=lower_bound(find.begin(),find.end(),sum[i])-find.begin()+1;
      res+=ask(tar);
      add(pos,1);
   }
   cout<<res<<"\n";
   
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int T = 1;
    //cin>>T;
   for(int i=1;i<=T;i++) 
   {
      solve();
   }
    return 0;
}