楼兰图腾(树状数组)

题目传送门

思路

以后但凡你能看到一道题可以:
修改某个数,或者求前缀和
那么都可以考虑使用树状数组

这道题的思路感觉总体上和求逆序对的感觉上差不多

总体上你可以在坐标轴上画一画,数出来构成V字形和倒V字形的3个数据的个数。
那么每一次我去找每一个数的左边比他大的乘上右边比它大的,然后求出所有的这样的和,就是我们要找到的V字形的组数

反之同理,那就找小的。

那么如何找就是一个问题了:
可以顺序遍历一遍,把在左边比它小的/大的分别用数组记录下来
然后再倒叙遍历一遍,把在右边比它小的/大的,和原数组相乘起来,乘积相加便是结果了。

#include<iostream>
#include<cstring>
using namespace std;

const int N=2e5+7;

typedef long long ll;

int tree[N];
int up[N];
int down[N];

int lowbit(int x){
    return x&(-x);
}
void add(int p,int x){
    for(int i=p;i<=N;i+=lowbit(i)){
        tree[i]+=x;
    }
    return;
}
int query(int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i)){
        res+=tree[i];
    }
    return res;
}
int query(int x,int y){
    return query(y)-query(x-1);
}
int a[N];
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
        up[i]=query(a[i],n);
        down[i]=query(a[i]);
        add(a[i],1);
    }
    ll ans1=0,ans2=0;
    memset(tree,0,sizeof tree);
    for(int i=n;i;i--){
        ans1+=up[i]*(ll)query(a[i],n);
        ans2+=down[i]*(ll)query(a[i]);
        add(a[i],1);
    }
    cout<<ans1<<' '<<ans2<<endl;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值