2025年寒假算法天梯赛 05

C. 蚂蚁工厂

• 二分+线段树查询和直接在线段树上查询的做法虽然都是\(O(nlog^2n)\)，但是由于后者跑不满(1+2+...+log(n))，在3s的时间限制下可以通过500000的数据

#include <bits/stdc++.h>
using namespace std;
int a[500005];
struct t1
{
	int l,r,v;
	long long sum;
}t[500000*21+10];
int root[500005],tot;
int insert(int q,int l,int r,int x)
{
	int p=++tot;
	t[p]=t[q];
	if(l==r)
	{
		t[p].v++;
		t[p].sum+=x;
		return p;
	}
	int mid=(l+r)>>1;
	if(x<=mid)
	{
		t[p].l=insert(t[q].l,l,mid,x);
	}
	else
	{
		t[p].r=insert(t[q].r,mid+1,r,x);
	}
	t[p].v=t[t[p].l].v+t[t[p].r].v;
	t[p].sum=t[t[p].l].sum+t[t[p].r].sum;
	return p;
}
int ask0(int p,int q,int l,int r)
{
	if(l==r)
	{
		return l;
	}
	int mid=(l+r)>>1;
	if(t[t[q].r].v-t[t[p].r].v==0)
	{
		return ask0(t[p].l,t[q].l,l,mid);
	}
	else
	{
		return ask0(t[p].r,t[q].r,mid+1,r);
	}
}
int ask1(int p,int q,int l,int r,int x)
{
	if(l==r)
	{
		return 1;
	}
	int mid=(l+r)>>1;
	if(!(t[t[q].l].v-t[t[p].l].v)||ask0(t[p].l,t[q].l,l,mid)-(t[t[q].l].v-t[t[p].l].v)<x-1)
	{
		return (t[t[q].l].v-t[t[p].l].v)+ask1(t[p].r,t[q].r,mid+1,r,x+(t[t[q].l].v-t[t[p].l].v));
	}
	else
	{
		return ask1(t[p].l,t[q].l,l,mid,x);
	}
}
long long ask2(int p,int q,int l,int r,int x)
{
	if(x==0)
	{
		return 0;
	}
	if(x==t[q].v-t[p].v)
	{
		return t[q].sum-t[p].sum;
	}
	int mid=(l+r)>>1;
	if(x<=(t[t[q].l].v-t[t[p].l].v))
	{
		return ask2(t[p].l,t[q].l,l,mid,x);
	}
	else
	{
		return t[t[q].l].sum-t[t[p].l].sum+ask2(t[p].r,t[q].r,mid+1,r,x-(t[t[q].l].v-t[t[p].l].v));
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n,m;
	cin>>n>>m;
	t[0].l=t[0].r=t[0].v=t[0].sum=0;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		root[i]=insert(root[i-1],1,1000000,a[i]);
	}
	for(int i=1;i<=m;i++)
	{
		long long u,v,l,r;
		cin>>u>>v>>l;
		r=l+v-u;
		int p=ask1(root[u-1],root[v],1,1000000,l)-1,q=v-u+1-p;
		long long res=ask2(root[u-1],root[v],1,1000000,p);
		cout<<(l+l+p-1)*p/2-res+(t[root[v]].sum-t[root[u-1]].sum-res)-(r+r-q+1)*q/2<<"\n";
	}
	return 0;
}