杭电2019多校第三场 HDU-6609 Find the answer（离散化+权值线段树）

最新推荐文章于 2021-03-09 20:35:06 发布

原创最新推荐文章于 2021-03-09 20:35:06 发布 · 168 阅读

0 ·

CC 4.0 BY-SA版权

树专栏收录该内容

2 篇文章

订阅专栏

本文深入探讨了离散化与权值线段树算法的应用，通过一组样例阐述了如何利用这两种算法解决特定问题，即在一系列数值中加入新元素后，判断是否需要移除部分元素以保持总和不超过预设值，同时提供了详细的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：t组样例，每个样例一个n和m，接着有n个数字，每次询问放入ai后如果总和大于m，至少要删除几个a[i]之前的数才可使和小于m
离散化+权值线段树，线段树存r-l的个数和总和
代码

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+100;
#define ll long long

struct st{
	int no,v;
}stt[N];

int num[N];

struct node{
	int r,l;
	int v;
	ll s;
}tree[N<<2];

int cmp(const st&a,const st&b){
	if(a.v>b.v) return 1;
	else return 0;
}

void build(int l,int r,int rt){
	tree[rt].l=l;
	tree[rt].r=r;
	if(l==r) {
		tree[rt].v=0;
		tree[rt].s=tree[rt].v*stt[rt].v;
		return ;
	}
	int mid=(l+r)>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	tree[rt].v=0;
	tree[rt].s=0;
}

void update(int rt,int t){
	if(tree[rt].l==tree[rt].r){
		tree[rt].v++;
		tree[rt].s=tree[rt].v*stt[t].v;
		return ; 
	} 
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(t<=mid) update(rt<<1,t);
	else if(t>mid) update(rt<<1|1,t);
	tree[rt].v=tree[rt<<1].v+tree[rt<<1|1].v;
	tree[rt].s=tree[rt<<1].s+tree[rt<<1|1].s;
}

int query(int rt,ll k){
	if(tree[rt].l==tree[rt].r){
		if(tree[rt].v==0) return 0;
		int sum=tree[rt].s/tree[rt].v;
		if(sum==0) return 0;
		if(k%sum==0) return k/sum;
		else return k/sum+1;
	}
	ll ans=0;
	if(k>=tree[rt<<1].s) ans=tree[rt<<1].v+query(rt<<1|1,k-tree[rt<<1].s);
	else ans=query(rt<<1,k);
	return ans;
}

int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		int n,m,i;
		scanf("%d %d",&n,&m);
		ll sum=0;
		for(i=1;i<=n;i++){
			 scanf("%d",&stt[i].v);
			 stt[i].no=i;
		}
		sort (stt+1,stt+n+1,cmp);
		
		for(i=1;i<=n;i++){
			num[stt[i].no]=i;
		}
		
		build(1,n,1);
		for(i=1;i<=n;i++){
			
			sum+=stt[num[i]].v;
			ll ans=0;
			if(sum>m){
				ans=query(1,sum-m);
			}
			update(1,num[i]);

			printf("%d ",ans);
		}
		printf("\n");
	}	
}