poj 1990 两颗树状数组

最新推荐文章于 2021-02-16 03:44:45 发布

原创最新推荐文章于 2021-02-16 03:44:45 发布 · 267 阅读

0 ·

CC 4.0 BY-SA版权

POJ 同时被 2 个专栏收录

67 篇文章

订阅专栏

算法

27 篇文章

订阅专栏

本文介绍了一种解决区间查询与更新问题的有效方法。通过使用离线处理和树状数组（Binary Indexed Tree, BIT）的数据结构技巧，该算法可以在O(N log N)的时间复杂度内解决涉及多个元素的区间查询与更新操作。文章详细展示了如何定义结构体来存储数据，如何进行排序以及如何实现树状数组的插入和查询操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<stdio.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn=20005;
long long t1[maxn],t2[maxn],ans,N;
int M;
struct Cow
{
int x,v;
}c[20005];
int cmp(Cow c1,Cow c2)
{
return c1.v<c2.v;
}
int lowbit(int x)
{
return x & (-x);
}
void insert(int x,int detal,long long t[])
{
for(int i=x;i<=M;i+=lowbit(i))
t[i]+=detal;
}
long long getsum(int x,long long t[])
{
long long ret=0;
for(int i=x;i>0;i-=lowbit(i))
ret+=t[i];
return ret;
}

int main()
{
// freopen("test.txt","r",stdin);
scanf("%d",&N);
for(int i=0;i<N;i++)
{
  scanf("%d%d",&c[i].v,&c[i].x);
  M=max(c[i].x,M);
}
sort(c,c+N,cmp);
memset(t1,0,sizeof(t1));
memset(t2,0,sizeof(t2));
long long tsum=0;
for(int i=0;i<N;i++)
{
  int x=c[i].x,v=c[i].v;
  long long cnt=getsum(x,t1),sum=getsum(x,t2);
  ans+=(long long)v*((cnt*x-sum)+(tsum-sum-(i-cnt)*x));
  insert(x,1,t1);
  insert(x,x,t2);
  tsum+=c[i].x;
}
printf("%lld\n",ans);
return 0;

}