C++树状数组入门模板+例题逆序对

树状数组,是一个用于在近似 O(logn)时间内动态修改以及查询前缀和的数据结构
C++树状数组入门模板+例题逆序对_第1张图片
该模板有区间修改和区间求和

 #include
using namespace std;
const int MAX=50005;
int a[MAX],tree[MAX],n;
int lowbit(int x)  //找最低位的1 
{
	return x&-x;
}
void add(int i,int x)//修改数据在i加x 
{
    while(i<=n)
    {
        tree[i]+=x;
        i+=lowbit(i);
    }
}

int sum(int i)
{
    int s=0;
    while(i>0)
    {
        s+=tree[i];
        i-=lowbit(i);
    }
    return s;
}
void range_add(int l, int r, int a) //区间修改
    {
        add(l, a);
        add(r + 1, -a);
    }
ll range_sum(int l, int r)  //区间求和
    {
        return sum(r) - sum(l - 1);
    }

int main()
{
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	
	
	
 } 

例题
树状数组入门应用求 逆序对 (求数组右边比它小的个数和)
因为卡时间,离散化,用了二分查找下标差,
用tree数组记录每个数组每次出现的次数。
力扣 315. 计算右侧小于当前元素的个数
codeforces 180. Inversions

#include
using namespace std;
const int MAX=100010;
const int mod=1e9+7;
#define ll long long 
#define _for(i,j,k) for(int i=j;i
#define endl  '\n'
#define inf 1<<29
int a[MAX],n,Max;
int tree[MAX];

int lowbit(int x)  //找最低位的1 
{
	return x&-x;
}
void add(int i,int x)//修改数据在i加x 
{
    while(i<=n)
    {
        tree[i]+=x;
        i+=lowbit(i);
    }
}

int sum(int x)//求1至i的和,若需求a至b的和,只需sum(b)-sum(a)+aks(a); 
{
    int sum = 0;
    while (x > 0) {
        sum += tree[x];
        x -= lowbit(x);
    }
    return sum;
}

int main()
{
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n;
	vector<int>ans(n);
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		ans[i]=a[i];
	}
	sort(ans.begin(),ans.end());
		ll cnt=0; 
	for(int i=n-1;i>=0;i--)  //如果换成for(int i=0;i
	{
		int id=lower_bound(ans.begin(),ans.end(),a[i])-ans.begin()+1;
		cnt+=sum(id-1);
		add(id,1);
	}
	cout<<cnt;
}

你可能感兴趣的:(数据结构与算法)