[2021 icpc昆明] M.Stone Games 主席树

题目链接：M.Stone Games

题解

题目大意是求区间里任意数加和形成的集合中的mex值。
mex值是集合中第一个未出现的整数。

假设当前区间为$a_l,a_{l+1},..,a_r$，可以合成范围为[1,x] 。显然x=${\sum }_{i=l}^{i=r}{a}_i$
对于剩余未参与合成的数范围为[1,x+1]的一些数，总和为sum，若参与合成，则合成范围会扩大为[1,x+sum]。

我们可以用主席树维护一个区间中不大于某个数的和，每次加上未参与合成的≤x+1中的数之和，扩大合成范围，当发现范围不再变化，就得到了最终合成范围，很显然答案就是边界+1。(合成范围下界一定是1，所以说的边界值就是上界)

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<bitset>
#include<cassert>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<deque>
#include<iomanip>
#include<list>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<unordered_set>
#include<unordered_map>
using namespace std;
//extern "C"{void *__dso_handle=0;}
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define pii pair<int,int>
#define lowbit(x) x&-x

const double PI=acos(-1.0);
const double eps=1e-6;
const ll mod=1e9+7;
const int inf=0x3f3f3f3f;
const int maxn=1e6+10;
const int maxm=100+10;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

struct Tree
{
	int ls,rs;
	ll sum;
}tree[maxn*31];
int a[maxn],rt[maxn],sz;

void add(int l,int r,int p,int &q,int v)
{
	q=++sz;
	tree[q]=tree[p];
	tree[q].sum+=v;
	if(l==r) return ;
	int mid = (l+r) >> 1;
	if(v<=mid) add(l, mid, tree[p].ls, tree[q].ls, v);
	else add(mid+1, r, tree[p].rs, tree[q].rs, v);
}

ll query(int l,int r,int p,int q,ll v)
{
	if(l==r) return tree[q].sum-tree[p].sum;
	int mid=(l+r) >> 1;
	if(v<=mid)
		return query(l, mid, tree[p].ls, tree[q].ls, v);
	else
		return query(mid+1, r, tree[p].rs, tree[q].rs, v)+tree[tree[q].ls].sum-tree[tree[p].ls].sum;
}

int main()
{
	int n,q; scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++) 
	{
		scanf("%d",&a[i]);
		add(1, 1000000000, rt[i-1], rt[i], a[i]);
	}
	ll ans=0;
	int l,r;
	while (q--) {
		scanf("%d%d",&l,&r);
		l=(l+ans)%n+1;
		r=(r+ans)%n+1;
		if(l>r) swap(l, r);
		ans=0;
		while (true) {
			ll new_ans = query(1, 1000000000, rt[l - 1], rt[r],ans+1);
			if(new_ans==ans)
				break;
			ans=new_ans;
		}
		printf("%lld\n",++ans);
	}
}