2021-06-29 连续非空子序列

连续非空子序列

给定一个数 n, 再给出长度为 n 的数列 ai​,求有多少连续非空子序列使得序列中的数之和大于0。
题目链接： 子序列

暴力求解的想法是，对于每一个 i，查询满足 j < i && sum[ i ] - sum[ j ] > 0，如果满足则ans++，但是这样暴力查找时间复杂度明显是O（N2）。

sum[ i ] - sum[ j ] > 0，即sum[ i ] > sum[ j ]，也就是想要知道每个 i 左边有多少个数小于它。于是问题转化为前缀和数组的“顺序对”有多少对。
参考“逆序对”的做法之一是树状数组，这题也可以利用树状数组来求解。
需要注意的点：
1.前缀和可能很大，但是只需要知道这个数组里元素的大小关系就好了，不需要知道确切的值是多少，所以要对前缀和数组进行离散化。将数组元素转化为“第几小”。
2.前缀和中有些值可能会相等，这时候不能将他们统计为“顺序对”，解决的方法一是排序的时候相等的元素的“第几小”依次递减，这样相当于将相等的数变为“逆序对”，就不会被统计到“顺序对”里去了。方法二则是直接用lowerbound函数找最小的下标。

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn = 1e6 + 5;
struct node {
    int val, th;
    bool operator < (const node &k) const {
        if (val != k.val) return val < k.val;
        else return th > k.th;
    }
} a[maxn];
int c[maxn];
int n;
int lowbit(int x) { return x & -x; }
void add(int i, int x)
{
    while (i <= n) {
        c[i] += x;
        i += lowbit(i);
    }
}
int que(int i)
{
    int ret = 0;
    while (i > 0) {
        ret += c[i];
        i -= lowbit(i);
    }
    return ret;
}
signed main()
{
    cin >> n;
    int ans = 0;
    for (int i = 1; i <= n; i++) {
        scanf("%lld", &a[i].val);
        a[i].val += a[i - 1].val;
        a[i].th = i;
        if (a[i].val > 0) ans++;
    }
    sort(a + 1, a + 1 + n);
    for (int i = 1; i <= n; i++) {
        a[a[i].th].val = i;
    }
    for (int i = 1; i <= n; i++) {
        ans += que(a[i].val - 1);
        add(a[i].val, 1);
    }
    cout << ans;
    return 0;
}