「树状数组」楼兰图腾

天涯浪子 提交于 2019-11-30 10:25:50

楼兰图腾

原题链接:楼兰图腾

题目大意

给你\(n\)个数,再给你两种限制条件,问你在符合这两种限制条件的情况,每种限制最多有多少个数

题目题解

这个题可以拓展到一些需要用到离散化的题

这个题很简单,两个限制条件,可以这么认为

  • 限制条件1:枚举所有的数,然后找左边比它大的,找右边比它大的
  • 限制条件2:枚举所有的书,然后处理方式与上面相反

当然方式可以更简单一点,只用计算一种即可,另外一种肯定是左右剩下的数组成(不懂?不懂可以看看代码),那么剩下怎么办呢?很简单,直接用 \(l_i \times r_i\) 就可以了(想想,为什么?答案很显然)

答案会爆int 这里要开long long 代码如下

//#define fre yes

#include <cstdio>

const int N = 200005;
int arr[N], r[N], l[N], c[N];

namespace BIT {
    int c[N];
    int lowbit(int x) {
        return x & (-x);
    }
    
    inline void init(int n) {
        for (int i = 1; i <= n; i++) {
            c[i] = 0;
        }
    }
    
    int ask(int x) {
        int res = 0;
        while(x) {
            res += c[x];
            x -= lowbit(x);
        } return res;
    }
    
    inline void add(int x, int y, int n) {
        while(x <= n) {
            c[x] += y;
            x +=   lowbit(x);
        }
    }
}

int main() {
    static int n;
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &arr[i]);
    }
    
    for (int i = n; i >= 1; i--) {
        r[i] = BIT::ask(arr[i] - 1);
        BIT::add(arr[i], 1, n);
    }
    
    BIT::init(n);
    for (int i = 1; i <= n; i++) {
        l[i] = BIT::ask(arr[i] - 1);
        BIT::add(arr[i], 1, n);
    }
    
    long long ans = 0;
    for (int i = 1; i <= n; i++) {
        ans += (long long)(n - i - r[i]) * (i - 1 - l[i]);
    } printf("%lld ", ans);
    
    ans = 0;
    for (int i = 1; i <= n; i++) {
        ans += (long long)r[i] * l[i];
    } printf("%lld\n", ans);
    return 0;
}
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!