Leetcode 891. 子序列宽度之和

本题官方题解是排序寻找递推规律，这里写一种不排序的更普遍性做法。
求子序列宽度之和可以拆开，先求每个元素在答案中作为序列最大值贡献的次数及和，记为最大值和；再求每个元素在答案中作为序列最小值贡献的次数及和，记为最小值和。两个问题是对偶问题，易进行转换。
单独看最大值和问题，可转换为求数组中每个元素左、右两侧小于（小于等于）该元素的元素数量。
由于本题范围在[1,100000] ，求左边比当前元素小的元素数量可以用树状数组进行计算。树状数组统计当前的前缀和，因此从左向右遍历，统计当前的sum(x)，该值代表个数（从1累计加到x的个数，即为小于等于x的个数）, 然后更新树状数组，对当前元素值加1。

typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<string> vs;
typedef unsigned int ui;

const int N = 1e5 + 5;
const int mod = 1e9 + 7;

class Solution {
	int c[N];
	ll pow2[N];

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

	void add(int x, int v) {
		while (x < N) {
			c[x] += v;
			x += lowbit(x);
		}
	}

	int query(int x) {
		int ret = 0;
		while (x > 0) {
			ret += c[x];
			x -= lowbit(x);
		}
		return ret;
	}
	int maxItems(const vi& cmp, const vi &nums) {
		int n = cmp.size();
		vi left(n), right(n);
		memset(c, 0, sizeof(c));
		for (int i = 0; i < n; ++i) {
			if (i > 0) {
				left[i] = query(cmp[i]);
			}
			add(cmp[i], 1);
			
		}
		memset(c, 0, sizeof(c));
		for (int i = n - 1; i >= 0; --i) {
			if (i < n - 1) {
				right[i] = query(cmp[i] - 1);
			}
			add(cmp[i], 1);
			
		}
		ll ret = 0;
		for (int i = 0; i < n; ++i) {
			ret += pow2[left[i] + right[i]] * nums[i];
			ret %= mod;
		}
		return int(ret);
	}

public:
	int sumSubseqWidths(vector<int>& nums) {
		pow2[0] = 1;
		for (int i = 1; i < N; ++i) {
			pow2[i] = (pow2[i - 1] * 2) % mod;
		}
		int m0 = maxItems(nums, nums);
		int n = nums.size();
		vi comp(n);
		int maxn = 0;
		for (int i = 0; i < n; ++i) {
			maxn = max(maxn, nums[i]);
		}
		maxn++;
		for (int i = 0; i < n; ++i) {
			comp[i] = maxn - nums[i];
		}
		int m1 = maxItems(comp, nums);
		int ret = (m0 - m1) % mod;
		if (ret < 0)
			ret = (ret + mod) % mod;
		return ret;
	}
};