题目-楼兰图腾-树状数组

在完成了分配任务之后,西部314来到了楼兰古城的西部。相传很久以前这片土地上(比楼兰古城还早)生活着两个部落,一个部落崇拜尖刀(‘V’),一个部落崇拜铁锹(‘∧’),他们分别用V和∧的形状来代表各自部落的图腾。西部314在楼兰古城的下面发现了一幅巨大的壁画,壁画上被标记出了N个点,经测量发现这N个点的水平位置和竖直位置是两两不同的。西部314认为这幅壁画所包含的信息与这N个点的相对位置有关,因此不妨设坐标分别为(1,y1),(2,y2),…,(n,yn)(1,y1),(2,y2),…,(n,yn)

,其中y1y1

~ynyn

是1到n的一个排列。西部314打算研究这幅壁画中包含着多少个图腾。如果三个点(i,yi),(j,yj),(k,yk)(i,yi),(j,yj),(k,yk)

满足1≤i<j<k≤n且yi>yj,yj<yk1≤i<j<k≤n且yi>yj,yj<yk

,则称这三个点构成V图腾;如果三个点(i,yi),(j,yj),(k,yk)(i,yi),(j,yj),(k,yk)

满足1≤i<j<k≤n且yi<yj,yj>yk1≤i<j<k≤n且yi<yj,yj>yk

,则称这三个点构成∧图腾;西部314想知道,这n个点中两个部落图腾的数目。因此,你需要编写一个程序来求出V的个数和∧的个数。输入格式第一行一个数n。第二行是n个数,分别代表y1,y2,…,yny1,y2,…,yn

。输出格式两个数,中间用空格隔开,依次为V的个数和∧的个数。数据范围对于所有数据,n≤200000n≤200000

,且输出答案不会超过int64。输入样例:5
1 5 3 2 4
输出样例:3 4

树状数组的分析:树状数组的用法很简单,但是想要理解却不太容易,我这里简单讲一下树状数组:他的关键还是二进制数,把一个区间的和分为很多个区间的和,它主要执行俩个功能,第一个功能是在x处插入c代码如下:

void add(int x,int c){
 for(int i=x;i<=n;i+=lowbit(i))    tr[i]+=c;
}

如何理解这个函数:tr[I]数组表示的是以i结尾长度为lowbit(i)的一个区间内所有数的和,当在I处插入一个数的·时候,把一个很大的区间分成几个小区间每一个区间的长度都是当前的lowbit(i),当在I处插入c,它上面的tr数组也要增加c,而当这个数的最后一个1加1的话,他就会向上进一位数,所以这样就能增加一个数。
第二个函数是求前x个数的和,代码如下

int sum(int x){
 int res=0;
 for(int i=x;i;i-=lowbit(x))       res+=tr[i];
 return res;
}

这个代码相较于上个代码就很好理解了,就是把x分成的每个tr区间的和加起来就可以了。
接下来就是这道题的思路:以V图腾为例,分为任何一个点为v的底下的那个点,他左面的比他大的点的数量乘上右面的比它大的点的数量就可以了,而求他左面的比他大的点,就从左向右枚举,把他左面的点都加进树状数组里面,然后求出差即可,同理,右面的点也是。
ac的代码如下:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
typedef long long LL;
int n;
const int N=200010;
int a[N];
int tr[N];
int gao[N],di[N];
using namespace std;
int lowbit(int x){
 return x&-x;
}
void add(int x,int c){
 for(int i=x;i<=n;i+=lowbit(i))    tr[i]+=c;
}
int sum(int x){
 int res=0;
 for(int i=x;i;i-=lowbit(x))       res+=tr[i];
 return res;
}
int main(){
 scanf("%d",&n);
    for(int i=1;i<=n;i++)    scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
     int y=a[i];
     gao[i]=sum(n)-sum(y);
     di[i]=sum(y-1);
     add(y,1);
 }
 memset(tr,0,sizeof tr);
 LL res1=0,res2=0;
 for(int i=n;i;i--){
  int y=a[i];
     res1+=gao[i]*(LL)sum(n)-sum(y);
     res2+=di[i]*(LL)sum(y-1);
     add(y,1);
 }
 cout<<res1<<" "<<res2<<endl;
 return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值