1009 Triple Inversions (35 分)

最新推荐文章于 2023-01-12 20:07:36 发布

SYaoJun

最新推荐文章于 2023-01-12 20:07:36 发布

阅读量218

点赞数

分类专栏： PAT顶级真题题解文章标签：算法数据结构 PTA

本文链接：https://blog.csdn.net/SYaoJun/article/details/123160362

版权

PAT顶级真题题解专栏收录该内容

3 篇文章 1 订阅

订阅专栏

题意

找出数组中有多少个三元组，满足单调递减。

思路

求出数组中每个位置上，左边比它大的数量，右边比它小的数量，根据乘法原理求得结果。
将每个数值映射到下标上，从左往右添加元素，判断已经添加的元素，在当前下标右边的位置有多少个数。可以使用前缀和求得，树状数组就是用来求带修改的前缀和。因为数据范围确定，所以求比A[i]大的数量，就是求1~n的前缀和减去1~A[i]的前缀和。

代码

#include <iostream>
#include <string>
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <utility>
#include <unordered_set>
#include <queue>
#include <deque>
using namespace std;


int f[1010];
char g[1010][1010];
int dx[4]={1,-1,0,0};
int dy[4]={0,0,1,-1};
struct Node{
    int x, y, d;
    Node(int x_, int y_, int d_): x(x_), y(y_), d(d_){}
};
const int N = 1e5+10;
int tr[N];
int n;
int A[N];
int r[N];
int l[N];
inline int lowbit(int x){
    return x & -x;
}
void add(int idx, int v){
    for(int i = idx; i <= n; i += lowbit(i)) {
        tr[i] += v;
    }
}
int query(int idx) {
    int res = 0;
    for(int i = idx; i; i -= lowbit(i)) res += tr[i];
    return res;
}
using i64 = long long;
int main() {
    int x;
    cin >> n;
    for(int i = 1; i <= n; i ++) {
        cin >> A[i];
    }


    // 统计右边比当前值小
    memset(tr, 0, sizeof tr);
    for(int i = n; i > 0; i--) {
        r[i] = query(A[i]-1); // 求1~A[i]-1的前缀和
        add(A[i], 1);
    }
    // 统计左边比当前值大
//    cout<<"yyy"<<endl;
    memset(tr, 0, sizeof tr);
    for(int i = 1; i <= n; i++) {
        l[i] = query(n) - query(A[i]); // 查询1到k的前缀和
        add(A[i], 1);
    }
    i64 res = 0;
    for(int i = 1; i <= n; i ++) {
//        cout<<"l = "<<l[i]<<" r = "<<r[i]<<endl;
        res += l[i]*(i64)r[i];
    }
    cout<<res<<endl;
    return 0;
}