题目链接
题目大意:让你在一个n个数的序列中,找总共有多少连续的子序列,子序列中的数和大于0。
题解:这个题就是让我们找sum[j]-sum[i-1]>0,(1<=j<=n,1<=i<=j),我们可以把他转化为sum[j]>sum[i-1],这样这个题目就简单了,我们可以用树状数组来解决这个问题,但是这个题的数据太大了,我们需要离散化一下,然后这个题就解决了。
启发:这个题我在区间上折腾了半天,一直都没有好的解决方法,以后有时候要换一个思路,转化一下,或许这个题就很好写了。
#include<bits/stdc++.h>
#define ll long long
#define pr pair<ll,ll>
#define ios ios::sync_with_stdio(false)
#define CRL(a) memset(a,0,sizeof a)
#define endl "\n"
using namespace std;
const int maxn = 1e6 + 5;
ll hx[maxn];
int n;
ll sum[maxn];
ll a[maxn];
int lowbit(int a)
{
return a & -a;
}
void add(int a,int b)
{
for (int i = a; i <= n;i+=lowbit(i))
{
hx[i] += b;
}
}
int solve(int a)
{
int ans = 0;
for (int i = a; i;i-=lowbit(i))
{
ans += hx[i];
}
return ans;
}
int main()
{
cin >> n;
vector<ll> v;
for (int i = 1; i <= n;i++)
{
int a;
cin>>a;
sum[i]=sum[i-1]+a;
v.push_back(sum[i]);
}
sort(v.begin(), v.end());
v.erase(unique(v.begin(), v.end()), v.end());
for (int i = 1; i <= n;i++)
{
a[i] = lower_bound(v.begin(), v.end(), sum[i]) - v.begin() + 1;
// cout << a[i] << endl;
}
ll ans = 0;
for (int i = 1; i <= n;i++)
{
add(a[i], 1);
if(sum[i]>0)
ans++;
ans += solve(a[i] - 1);
}
cout << ans;
}