bzoj4299 Codechef FRBSUM【主席树】

题目分析

当我们考虑如果能够 组成 [0,x]的所有的数，那么这些数都能扩展区间，因此，我们考虑每次统计 小于 x的所有数的和，和就是当前能够扩展的区间，然后再统计小于和的所有值的和，不断倍增就可以了。

所以用一个主席树维护一下
复杂度是 $O(nlognlogn)$

代码详解

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5;
const int M = 6e6;
int rt[N],lc[M],rc[M],sz;
long long s[M];
inline int read()
{
	int x=0,f=1; char ch = getchar();
	while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch =getchar();}
	while(ch>='0'&&ch<='9'){x= x*10+ch-'0';ch =getchar();}
	return x*f;
}
void ins(int &np,int pre,int l,int r,int v)
{
	np = ++sz; s[np] = s[pre]+(long long)v;
	if(l==r) return;
	int mid = (l+r)>>1;
	if(v<=mid) ins(lc[np],lc[pre],l,mid,v),rc[np]=rc[pre];
	else ins(rc[np],rc[pre],mid+1,r,v),lc[np]= lc[pre];
}
long long que(int ls,int rs,int l,int r,int v)
{
	if(l>r) return 0; 
	long long tmp = s[rs]-s[ls];
	if(v==r) return tmp;
	int mid = (l+r)>>1; 
//	cout<<l<<" "<<r<<" "<<mid<<endl;
	if(v<=mid) return que(lc[ls],lc[rs],l,mid,v);
	else return que(rc[ls],rc[rs],mid+1,r,v)+s[lc[rs]]-s[lc[ls]];
}
int main()
{
	int n= read();
	for(int i=1;i<=n;i++) ins(rt[i],rt[i-1],1,n,read());
	int m = read();
	while(m--)
	{
		int l = read(),r=read();
		long long pr = -1,cr=0;
		while(cr!=pr)
		{
			pr = cr;
			cr = que(rt[l-1],rt[r],1,n,min(cr+1,(long long)N));
		}
		printf("%lld\n",cr+1);
	}
	return 0;
}