Work[线段树][二分答案]

Work[线段树][二分答案]_第1张图片

分析

Work[线段树][二分答案]_第2张图片

开始也想到二分,但是无法判断哪些天数比它大,比它小

排序啊,人按浪费的时间排序,题按时间排序,下标加入线段树

#include
#define N 200005
#define LL long long 
using namespace std;
int m,n,ans[N],Mid,cnt[N*4];//时间总和  个数总和 
LL tot[N*4];
struct Node1{int w;int position;}t[N];
struct Node2{int d;int r;int position;}v[N];
bool cmp1(Node1 x,Node1 y){return x.w>y.w;}
bool cmp2(Node2 x,Node2 y){return x.d>y.d;}
int read(){
	int cnt=0;char ch=0;
	while(!isdigit(ch)) ch=getchar();
	while(isdigit(ch)) cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt;
}
void add(int root,int l,int r,int position,int w){
	tot[root]+=w,cnt[root]++;
	if(l==r) return;
	int mid=(l+r)>>1;
	if(position<=mid) add(root<<1,l,mid,position,w);
	else add(root<<1|1,mid+1,r,position,w);
}
void quary(int root,int l,int r,int w,int cur){
    if(l==r){Mid=l;return;}
    int mid=(l+r)>>1;
    if(tot[root<<1]-(LL)cnt[root<<1]*w>=cur)//根据cur的值来找到第几天
      quary(root<<1,l,mid,w,cur);
    else quary(root<<1|1,mid+1,r,w,cur-(tot[root<<1]-(LL)cnt[root<<1]*w));
}
	 
int main()
{
	//freopen("1.in","r",stdin);
	n=read(),m=read();
	for(int i=1;i<=m;i++) t[i].w=read(),t[i].position=i;
	sort(t+1,t+m+1,cmp1);//时间降序排列 
	for(int i=1;i<=n;i++) v[i].d=read(),v[i].r=read(),v[i].position=i;
	sort(v+1,v+n+1,cmp2);//d降序排列 
	int left=1;
	for(int i=1;i<=n;i++){
	  for(;left<=m&&t[left].w>=v[i].d;left++){//t[i]-d[i]>=0 进入线段树 
	    add(1,1,m,t[left].position,t[left].w);
	  }
	  if(tot[1]-(LL)(v[i].d*cnt[1])当前的人 
	  else{//二分答案 
	    Mid=0;
	    quary(1,1,m,v[i].d,v[i].r);
	    ans[v[i].position]=Mid;
	  }
    }
	for(int i=1;i<=n;i++) cout<

你可能感兴趣的:(线段树)