hdu6601Keen On Everything But Triangle[2019hdu多校第二场,主席树]

//需要依次查询区间第1大,第2大,第3大,..., 第k大,直到构成三角形或者数用完
//O(n*logn*44)
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define N 100005
#define LOG 20
#define mid ((l + r) >> 1)
typedef long long LL;
struct node
{
	int ls,rs,sum;
}tr[N*LOG];
int s,t,n,q,m;
LL a[N], b[N];
LL t1, t2, t3, ans;
int root[N];
int tot;
int build(int l, int r)
{
	int now = ++tot;
	tr[now].sum = 0;
	if (l < r)
	{
		tr[now].ls = build(l, mid);
		tr[now].rs = build(mid+1, r);
	}
	return now;
}
int insert(int pre, int l, int r, int x)
{
	int now = ++tot;
	tr[now] = tr[pre];
	tr[now].sum++;
	if (l < r)
	{
		if (x <= mid) tr[now].ls = insert(tr[pre].ls, l, mid, x);
		else tr[now].rs = insert(tr[pre].rs, mid+1, r, x);
	}
	return now;
}
int query(int s, int t, int l, int r, int k)
{
	if (l == r) return l;
	int sum = tr[tr[t].rs].sum - tr[tr[s].rs].sum;
	if (sum >= k)
		return query(tr[s].rs, tr[t].rs, mid+1, r, k);
	else return query(tr[s].ls, tr[t].ls, l, mid, k - sum);
}
int main()
{
	while (~scanf("%d%d", &n, &q) && n)
	{
		for (int i=1;i<=n;i++)
		{
			scanf("%lld", &a[i]);
			b[i] = a[i];
		}
		sort(b+1,b+n+1);
		m = unique(b+1,b+n+1) - b - 1;
		tot = 0;
		root[0] = build(1,m);
		for (int i=1;i<=n;i++)
		{
			int t = lower_bound(b+1,b+m+1,a[i]) - b;
			root[i] = insert(root[i-1], 1, m, t);
		}
		while (q--)
		{
			scanf("%d%d", &s, &t);
			if (t - s + 1 >= 1) t1 = b[query(root[s-1],root[t],1,m,1)];
			if (t - s + 1 >= 2) t2 = b[query(root[s-1],root[t],1,m,2)];
			ans = -1;
			for (int i=3;i<=t - s + 1;i++)
			{
				t3 = b[query(root[s-1],root[t],1,m,i)];

				//cout<

				if (t3 + t2 > t1)
				{
					ans = t1+t2+t3;
					break;
				}
				t1 = t2;t2 = t3;
			} 
			printf("%lld\n", ans);
		}
	}
}

比赛的时候想莫队,如果更新还暴力的话,就会T,不暴力更新似乎也可以,用set维护,就是在删掉一个点的时候考虑3中情况,更新是一个比较小的常数,但是要写好多if的样子。。看起来就不正解

你可能感兴趣的:(高级数据结构)