E. Tufurama

最新推荐文章于 2024-09-12 19:02:20 发布

混子本混

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量153

点赞数

分类专栏：主席树文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_52410477/article/details/120566502

版权

主席树专栏收录该内容

3 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用主席树（也称为区间树）解决动态区间求和的问题。文章首先定义了主席树的数据结构，并展示了如何构建和更新节点。接着，通过一个实例解释了如何在O(log n)的时间复杂度内查询和插入元素。最后，代码实现了一个完整的解决方案，用于计算数组中每个元素对区间和的影响。

摘要由CSDN通过智能技术生成

E. Tufurama

链接

（主席树）

#include <bits/stdc++.h>
using namespace std;

const int N = 200010;

struct node 
{
	int l , r ; 
	int sum ; 
}tr[N * 40 ] ;

int n  , a[N] ;
int idx ,root[N]; 
long long ans ; 
int build(int l , int r ) 
{
	int p = ++idx ;
	if(l == r ) return p;
	
	int mid = l + r >> 1 ;
	tr[p].l = build(l , mid ) , tr[p].r = build(mid + 1 , r ) ;
	return p ; 
}

int query(int p , int l , int r , int L , int R) 
{
	if(L <= l && r <= R ) return tr[p].sum ;
	int mid = l +r >> 1 , sum = 0 ;
	if(L <= mid ) sum += query(tr[p].l , l , mid , L , R) ;
	if(R > mid ) sum += query(tr[p].r , mid + 1 ,r , L,R) ;
	return sum; 
}

int insert(int p , int l , int r , int k )
{
	int q = ++idx ;
	tr[q] = tr[p] ;
	if(l == r ) 
	{
		tr[q].sum ++ ;
		return q ;
	}
	int mid = l + r >> 1 ;
	if(k <= mid ) tr[q].l = insert(tr[q].l , l , mid , k ) ;
	else  tr[q].r = insert(tr[q].r, mid + 1 , r , k ) ;
	tr[q].sum = tr[tr[q].l].sum + tr[tr[q].r ].sum ;
	return q ; 
}

int main()
{
	int n ;
	cin>>n;
	root[0] = build(1 , n ) ; 
	for(int i = 1 ; i <= n ; i ++ ) 
	{
		cin>>a[i] ;
		a[i] = min(a[i] , n ) ;
		ans += (long long)query(root[min(i - 1 , a[i])] , 1 , n , i  , n ) ;
		root[i] = insert(root[i - 1 ] , 1 , n , a[i ]) ; 
	}
	cout<<ans ;
	return 0 ; 
}