【BZOJ2527】MET-Meteors（整体二分）

【BZOJ2527】MET-Meteors（整体二分）

题面

BZOJ权限题，良心洛谷链接

题解

其实我也不会做

看了zsy博客才会做。。。

这题如果直接爆算做显然行不通

如果只有单次询问，我们就可以二分答案

但是询问太多。。

不会二分。。

怎么办？

我们来想想瓶颈在哪里

如果每次都进行一次单次二分

我们就需要不停的计算前缀和

但是其实再进行别的二分的时候我们已经算过了

这里就算重了

怎么解决？

那我们就不让他算重：

我们把所有询问一起二分

这样每次计算完之后

所有的询问就被分成了两部分：

一部分是当前二分值有解

另一部分是无解

这样就可以解决重复计算的问题了

因为要反复计算前缀和

所以使用树状数组来计算就行了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 320000
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Rain{int l,r;ll w;}r[MAX];
ll c[MAX];
int n,m,ans[MAX],K;
int lowbit(int x){return x&(-x);}
void Add(int x,int w){while(x<=m)c[x]+=w,x+=lowbit(x);}
ll getsum(int x){ll ret=0;while(x)ret+=c[x],x-=lowbit(x);return ret;}
vector<int> a[MAX];
struct Query{int id;ll p;}p[MAX],p1[MAX],p2[MAX];
void Fall(int t,int opt)
{
	if(r[t].l>r[t].r)Add(1,opt*r[t].w);
	Add(r[t].l,opt*r[t].w),Add(r[t].r+1,-opt*r[t].w);
}
void Work(int l,int r,int L,int R)
{
	if(L>R)return;
	if(l==r)
	{
		for(int i=L;i<=R;++i)ans[p[i].id]=l;
		return;
	}
	int mid=(l+r)>>1;
	ll tot;
	int t1=0,t2=0;
	for(int i=l;i<=mid;++i)Fall(i,1);
	for(int i=L;i<=R;++i)
	{
		tot=0;
		for(int j=0,l=a[p[i].id].size();j<l;++j)tot+=getsum(a[p[i].id][j]);
		if(tot>=p[i].p)p1[++t1]=p[i];
		else p[i].p-=tot,p2[++t2]=p[i];
	}
	for(int i=l;i<=mid;++i)Fall(i,-1);
	for(int i=1;i<=t1;++i)p[i+L-1]=p1[i];
	for(int i=1;i<=t2;++i)p[i+L+t1-1]=p2[i];
	Work(l,mid,L,L+t1-1);
	Work(mid+1,r,L+t1,R);
}
int main()
{
	n=read();m=read();
	for(int i=1;i<=m;++i)a[read()].push_back(i);
	for(int i=1;i<=n;++i)p[i].p=read(),p[i].id=i;
	K=read();
	for(int i=1;i<=K;++i)r[i].l=read(),r[i].r=read(),r[i].w=read();
	++K;r[K]=(Rain){1,m,1e9};
	Work(1,K,1,n);
	for(int i=1;i<=n;++i)ans[i]==K?puts("NIE"):printf("%d\n",ans[i]);
	return 0;
}