HDU - 6609 权值线段树

HDU - 6609  

给一个序列 a 对于每个i 询问最少删除几个数ai(i<ai) 能使得i位置的前缀和小于等于m 

表述可能不清楚 大家可以自己看题

我们将题意转换一下 最少需要删几个数  那这几个数肯定是越大越好  并且 这几个数的和   (其中sumi是i位置的前缀和) 那我们先把ai离散化了  然后遍历一次ai数组  每次向权值线段树中插入ai（当然是插入在ai离散化后排名的位置） 分别记录cnt和sum  cnt是每个数出现的次数 sum是求和    然后我们在线段树中搜索 如果右子树之和大于等于s 我们先走右子树 否则  s-=右子树之和  ct+=右子树的个数和   当到达子结点时  ct在加上 s/a[l] 就行了  因为先走右子树 保证了ct最小  

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 2e5+10;
typedef long long ll;
ll sum[N<<2];
int val[N<<2];
int a[N],ans[N],b[N],tot;
#define ls id<<1
#define rs ls|1
#define mid (l+r>>1)
#define lson ls,l,mid
#define rson rs,mid+1,r
int gethas(int x){
	return lower_bound(b+1,b+1+tot,x)-b;
}
void pushup(int id){
	sum[id]=sum[ls]+sum[rs];
	val[id]=val[ls]+val[rs];
}
void build(int id,int l,int r){
	sum[id]=0;val[id]=0;
	if(l==r) return;
	build(lson);build(rson);
}
void update(int id,int l,int r,int pos){
	if(l==r){
		sum[id]+=b[l];
		val[id]++;
		return;
	}
	if(pos<=mid) update(lson,pos);
	else {update(rson,pos);} 
	pushup(id);
}
int query(int id,int l,int r,int ct,ll v){
	if(l==r){
		if(v%b[l]==0) ct+=v/b[l];
		else {ct+=v/b[l]+1;} 
		return ct;
	}
	if(sum[rs]>=v) return query(rson,ct,v);
	else {return query(lson,ct+val[rs],v-sum[rs]);} 
}
int main(){
	int n,t;
	ll m;
	scanf("%d",&t);
	while(t--){
		scanf("%d%lld",&n,&m);
		for(int i = 1; i <= n; i++) scanf("%d",&a[i]),b[i]=a[i];
		sort(b+1,b+1+n);
		tot=unique(b+1,b+1+n)-b-1;
		build(1,1,tot);
		ans[1]=0;
		int now=0;
		ll s=a[1];
		update(1,1,tot,gethas(a[1]));
		for(int i = 2; i <= n; i++){
			s+=a[i];
			if(s>m){
				ans[i]=query(1,1,tot,0,s-m);
			}else {ans[i]=0;} 
			update(1,1,tot,gethas(a[i]));
		}
		for(int i = 1; i <= n; i++) printf("%d ",ans[i]);
		puts(""); 
	}
	return 0;
}