欢迎访问我的pat顶级题解目录哦
题目描述
算法设计
可以利用树状数组来解决这个问题。
由于输入序列的每个元素的值都不会超过
1
0
5
10^5
105,因此我们可以开辟一个长
1
0
5
+
5
10^5+5
105+5的树状数组c
。设计getSum(x)函数表示1到x这些数字在序列中出现次数之和。设计update函数用于更新数字出现次数。
我们要对整个序列A
进行两次遍历,第一次从前向后遍历,针对遍历到的数字A[i]
,求其左侧比A[i]
大的数字出现次数之和,并把结果存储到数组left
中。第二次从后向前遍历,针对遍历到的数字A[i]
,求其右侧比A[i]
小的数字出现次数之和,假设为k,计算
A
[
i
]
∗
k
A[i]*k
A[i]∗k,即为以A[i]
为中间的数的triple inversion
的个数。累加所有这样的
A
[
i
]
∗
k
A[i]*k
A[i]∗k,即为最终结果。
C++代码
#include <bits/stdc++.h>
using namespace std;
auto lowbit = [](int x) { return x & (-x); };
vector<int> c(100005);
void update(int x, int v) {
for (int i = x; i < c.size(); i += lowbit(i))
c[i] += v;
}
int getSum(int x) {
int sum = 0;
for (int i = x; i > 0; i -= lowbit(i))
sum += c[i];
return sum;
}
int main() {
int n;
long long ans = 0;
cin >> n;
vector<int> A(n), left(n);
for (int i = 0; i < A.size(); ++i) {
cin >> A[i];
update(A[i], 1);
left[i] = i + 1 - getSum(A[i]);
}
fill(c.begin(), c.end(), 0);
for (int i = A.size() - 1; i >= 0; --i) {
update(A[i], 1);
ans += left[i] * 1LL * getSum(A[i] - 1);
}
cout << ans;
return 0;
}