bzoj-1901 Dynamic Rankings

题意:

带修改区间k小值;

n,m<=10000;

a[i]<=10^9;


题解:

听说是道裸题就过来刷刷 (卧槽我最近似乎都是在刷裸题);

写完前缀和的主席树感觉挺厉害,感受了一下树状数组就来写这题;

然后写更新的的时候我就不会了;

前缀和的时候,后一个树从前一个继承一部分结点而来的;

但是树状数组不能这么搞啊= =;

然后发现暴力建就可以了,也是犯二了;

最多n+m次修改,每次修改logn棵线段树,每颗线段树最多增加log(n+m)个结点;

空间也就是O(nlog2n),所以没问题;

修改算是搞过了,查询大概就主席树一下,加上树状数组的复杂度是log2n的;

跑的飞快,328ms  (其实我一开始脑抽写成了O(nlog3n)然而还是500ms+);


代码:


#include<vector>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 11000
using namespace std;
struct node
{
	int l,r,sum;
}a[N<<8];
vector<int>nol,nor;
int dis[N<<1],val[N],root[N];
int L[N],R[N],V[N];
int n,len,cnt;
bool t[N];
char str[10];
int lowbit(int x)
{
	return x&(-x);
}
void Insert(int l,int r,int &now,int val,int delta)
{
	if(!now)	now=++cnt;
	a[now].sum+=delta;
	if(l==r)	return ;
	int mid=l+r>>1;
	if(val<=mid)
		Insert(l,mid,a[now].l,val,delta);
	else
		Insert(mid+1,r,a[now].r,val,delta);
}
void update(int x,int val,int delta)
{
	while(x<=n)
	{
		Insert(1,len,root[x],val,delta);
		x+=lowbit(x);
	}
}
int query(int l,int r,int k)
{
	int temp=0,mid=l+r>>1;
	if(l==r)	return l;
	for(int i=0;i<nor.size();i++)
		temp+=a[a[nor[i]].l].sum;
	for(int i=0;i<nol.size();i++)
		temp-=a[a[nol[i]].l].sum;
	if(k<=temp)
	{
		for(int i=0;i<nor.size();i++)
			nor[i]=a[nor[i]].l;
		for(int i=0;i<nol.size();i++)
			nol[i]=a[nol[i]].l;
		return query(l,mid,k);
	}
	else
	{
		for(int i=0;i<nor.size();i++)
			nor[i]=a[nor[i]].r;
		for(int i=0;i<nol.size();i++)
			nol[i]=a[nol[i]].r;
		return query(mid+1,r,k-temp);
	}
}
int main()
{
	int m,i,j,k,x,l,r;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
	{
		scanf("%d",val+i);
		dis[i]=val[i];
	}
	for(i=1,k=0;i<=m;i++)
	{
		scanf("%s",str);
		if(str[0]=='Q')
			scanf("%d%d%d",L+i,R+i,V+i);
		else
		{
			t[i]=1;
			scanf("%d%d",L+i,V+i);
			dis[n+(++k)]=V[i];
		}
	}
	sort(dis+1,dis+n+k+1);
	len=unique(dis+1,dis+n+k+1)-dis-1;
	for(i=1;i<=n;i++)
	{
		x=lower_bound(dis+1,dis+len+1,val[i])-dis;
		update(i,x,1);
	}
	for(i=1;i<=m;i++)
	{
		if(t[i])
		{
			x=lower_bound(dis+1,dis+len+1,val[L[i]])-dis;
			update(L[i],x,-1);
			x=lower_bound(dis+1,dis+len+1,V[i])-dis;
			update(L[i],x,1);
			val[L[i]]=V[i];
		}
		else
		{
			nol.clear(),nor.clear();
			for(x=R[i];x;x-=lowbit(x))
				nor.push_back(root[x]);
			for(x=L[i]-1;x;x-=lowbit(x))
				nol.push_back(root[x]);
			printf("%d\n",dis[query(1,len,V[i])]);
		}
	}
	return 0;
}




发布了216 篇原创文章 · 获赞 41 · 访问量 23万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览