整体二分模板(不带修)

POJ2104

code:

#include<cstdio>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=1e5+5;
int n,m,Ans[MAXN],b[MAXN],cg[MAXN];
struct ren{
	int id,val;
	ren(){};
	ren(int I,int V){id=I,val=V;
	}
};
vector<ren>a;
struct SG{
	int l,r,sum;
}t[MAXN<<2];
struct que{
	int id,l,r,k;
	que(){};
	que(int I,int L,int R,int K){id=I,l=L,r=R,k=K;
	}
};
vector<que>q;
bool cmp(ren x,ren y){return x.val<y.val;
}
void UPD(int p){t[p].sum=t[p<<1].sum+t[p<<1|1].sum;
}
void bui(int p,int l,int r){
	t[p].l=l,t[p].r=r;t[p].sum=0;
	if(l==r) return;
	int mid=(l+r)>>1;
	bui(p<<1,l,mid);bui(p<<1|1,mid+1,r);
}
void upd(int p,int x,int k){
	if(t[p].l==t[p].r){t[p].sum+=k;return;}
	int mid=(t[p].l+t[p].r)>>1;
	if(x<=mid) upd(p<<1,x,k);
	else upd(p<<1|1,x,k);
	UPD(p);
}
int ask(int p,int L,int R){
	if(t[p].l>=L&&t[p].r<=R) return t[p].sum;
	int mid=(t[p].l+t[p].r)>>1,res=0;
	if(L<=mid) res+=ask(p<<1,L,R);
	if(R>mid) res+=ask(p<<1|1,L,R);
	return res;
}
void sol(int L,int R,vector<ren>v1,vector<que>v2){
	if(L==R){
		for(int i=0;i<v2.size();i++) Ans[v2[i].id]=cg[L];
		return;
	}
	int mid=(L+R)>>1;
	vector<ren>a1,a2;vector<que>b1,b2;
	for(int i=0;i<v1.size();i++){
		if(v1[i].val<=mid) a1.push_back(v1[i]),upd(1,v1[i].id,1);
		else a2.push_back(v1[i]);
	}
	for(int i=0;i<v2.size();i++){
		int tt=ask(1,v2[i].l,v2[i].r);
		if(v2[i].k<=tt) b1.push_back(v2[i]);
		else v2[i].k-=tt,b2.push_back(v2[i]);
	}
	for(int i=0;i<v1.size();i++){if(v1[i].val<=mid) upd(1,v1[i].id,-1);
	}
	sol(L,mid,a1,b1);sol(mid+1,R,a2,b2);
}
int main(){
	scanf("%d%d",&n,&m);
	bui(1,1,n);
	for(int i=1;i<=n;i++){
		int x;
		scanf("%d",&x);
		a.push_back((ren){i,x});b[i]=x;
	}
	for(int i=1;i<=m;i++){
		int L,R,K;
		scanf("%d%d%d",&L,&R,&K);
		q.push_back((que){i,L,R,K});
	}
	sort(b+1,b+1+n);
	int c1=unique(b+1,b+1+n)-b-1;
	for(int i=0;i<n;i++){int kk=lower_bound(b+1,b+1+c1,a[i].val)-b;cg[kk]=a[i].val;a[i].val=kk;
	}
	sort(a.begin(),a.end(),cmp);
	sol(1,n,a,q); 
	for(int i=1;i<=m;i++) printf("%d\n",Ans[i]);
	return 0;
}
posted @ 2022-02-16 20:20  StranGePants  阅读(23)  评论(0编辑  收藏  举报