树状数组
当需要:
- 将第 i 个数加上 k
- 输出 [ i , j ] 内每个数的和
朴素算法
单点修改:O(1)
区间查询:O(n)
树状数组
单点修改:O(logn)
区间查询:O(logn)
lowbit
原理:链接: lowbit()
lowbit(6)=lowbit ( 二进制 [ 0110 ] ) = 二进制 [ 10 ] = 2;
代码实现
int lowbit(int x)
{
return x & -x;
}
此时需要开辟新的数组tr[N]
其中 tr[x] 表示:以x为根节点的子树的叶结点值的和
每个结点的长度为 lowbit(x)
tr[x] 的父节点为tr[lowbit(x)+x]
树的深度为:logn+1
树状数组的两个操作
- add(x,k):将序列第
x
个数加上k
因此只要包括第x个数的所有“前缀和”都需要加上k,树状数组中表示:结点x到其根节点的所有结点都加上k
void add(int x, int k)
{
for (int i = x; i <= n; i += lowbit(i))
tr[i] += k;
}
ask(x)表示求前 x 个数的和
int ask(int x)
{
int res = 0;
for (int i = x; i; i -= lowbit(i))
res += tr[i];
}
241. 楼兰图腾
题目链接:楼兰图腾
题目描述
大致含义为:在一个二维平面,有一些随意放置的点,这些点能组成 V
和∧
分别有多少种
题目分析
- 分别统计
x
左侧比他大和比他小的个数
从左往右扫,统计后将其加入tr[]
分别统计x
右侧比他大和比他小的个数
从右往左扫,统计后将其加入tr[]
- 每个点左边大于他的点的个数 x 右边大于他的点的个数,求
∧
同理求V
CODE
#include <bits/stdc++.h>
using namespace std;
const int N = 2000010;
typedef long long LL;
int n;
int tr[N], a[N];
int lower[N], _greater[N];
int lowbit(int x)
{
return x & -x;
}
void add(int x, int k)
{
for (int i = x; i <= n; i += lowbit(i))tr[i] += k;
}
int ask(int x)
{
int res = 0;
for (int i = x; i; i -= lowbit(i))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];// 统计在i位置之前所有比a[i]大的数的个数、以及比a[i]小的数的个数
lower[i] = ask(y - 1);//[1,y-1]
_greater[i] = ask(n) - ask(y);//[y+1,n]
add(y, 1);
}
memset(tr, 0, sizeof tr);
LL resA = 0, resV = 0;
for (int i = n; i >= 1; i--)
{
int y = a[i];
resA += (LL)lower[i] * ask(y - 1);
resV += (LL)_greater[i] * (ask(n) - ask(y));
add(y, 1);
}
printf("%lld %lld", resV, resA);
return 0;
}
参考:
《算法竞赛进阶指南》
AcWing算法提高课
B站:树状数组
%%% qiaoxinwei
https://www.acwing.com/solution/content/13818/