hdu6464 免费送气球(广东工业大学校赛-权值线段树+k小值思想)

题目

q(q<=1e5)次操作，分两种

①1 x y 加入x个y值(0<=x,y<1e9)

②2 x y 查询从小到大排名为x到排名为y的数的和(x,y不大于当前插入的数的数量)

思路来源

https://blog.csdn.net/lzc504603913/article/details/88605039

https://www.cnblogs.com/zquzjx/p/10560355.html

题解

权值线段树，先将操作1的插入的值离散化，排序去重，

在线处理，遇1操作就插入；遇2操作就用前缀和作差，[l,r]的询问等价于[1,r]-[1,l-1]

每个节点维护两个值，num维护的是出现个数，sum维护的是这段区间完整的答案值

注意sum要取模，而num不能取模，不然会影响询问

 

引入了查询k小值的思想，

k小值是说查询第k小值，

如果左子个数num[p<<1]>k就去查左子的第k大，

否则去查右子的第k-num[p<<1]大，

当l==r时，返回l，就是要查的值

 

而这里，查询[1,k]的和的值，是类似的，

如果左子个数num[p<<1]>k就去左子里查询这个和，

否则答案=左子完整区间sum[p<<1]+右子的前k-num[p<<1]个值的和

当l==r时，返回当前剩余k个*当前位置值a[l]的和

心得

心态崩了，自己的代码两个月了还是不知道为什么错

一次耽误三个小时，一次耽误两个小时，对拍也对不出来

算了，补上就完事了，不死抠不死抠

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int mod=1e9+7;
const int maxn=1e5+10;
typedef long long ll;
struct node
{
	int op;
	ll x,y;
}e[maxn];
ll num[maxn*5],sum[maxn*5];
ll a[maxn]; 
int q,n;
void pushup(int p)
{
	num[p]=num[p<<1]+num[p<<1|1];
	sum[p]=(sum[p<<1]+sum[p<<1|1])%mod;
}
void update(int p,int l,int r,int pos,ll v)
{
	if(l==r)
	{
		num[p]+=v;
		(sum[p]+=v*a[l]%mod)%mod;
		return;
	}
	int mid=(l+r)/2;
	if(pos<=mid)update(p<<1,l,mid,pos,v);
	else update(p<<1|1,mid+1,r,pos,v);
	pushup(p);
}
ll ask(int p,int l,int r,ll cnt)
{
	if(cnt==0)return 0;
	if(l==r)return cnt*a[l]%mod;
	int mid=(l+r)/2;
	if(num[p<<1]>cnt)return ask(p<<1,l,mid,cnt)%mod;
	else return (sum[p<<1]+ask(p<<1|1,mid+1,r,cnt-num[p<<1]))%mod;
} 
int main()
{
	scanf("%d",&q);
	for(int i=1;i<=q;++i)
	{
		scanf("%d%lld%lld",&e[i].op,&e[i].x,&e[i].y);
		if(e[i].op==1)a[++n]=e[i].y;
	}
	sort(a+1,a+n+1);
	n=unique(a+1,a+n+1)-(a+1);
	for(int i=1;i<=q;++i)
	{
		if(e[i].op==1)
		{
			int pos=lower_bound(a+1,a+n+1,e[i].y)-a;
			update(1,1,n,pos,e[i].x);
		}
		else
		{
			ll L=ask(1,1,n,e[i].x-1);
			ll R=ask(1,1,n,e[i].y);
			//printf("R:%lld L:%lld\n",R,L);
			printf("%lld\n",(R-L+mod)%mod);
		}
	}
	return 0;
}