关闭

BZOJ 2527 [Poi2011]Meteors [整体二分+线段树]

标签: BZOJpoi整体二分线段树
174人阅读 评论(0) 收藏 举报
分类:

题意:给你n个国家,m个观测点组成的环,每个观测点所属的国家,和每个国家所需要的陨石数,总共k波陨石向范围为(L,R)中的观测点提供a个陨石。问每个国家在第几波陨石降落的时候第一次获得所需要的陨石数量。

题解:整体二分法,类似于二分答案,每次将陨石波数二分为(L,mid)(mid+1,R),用线段树维护从第1波陨石到第mid波陨石降落时观测点陨石数量,(区间更新,单点查询),暴力判断当前mid波的陨石降落时是否能使这些国家集合满足需求,能满足要求的国家分到一组,不能满足需求则分到另一组,再分别继续进行二分陨石区间。

有一个坑点就是一个国家最多会得到(3*10^5)*(3*10^5)*10^9个陨石 爆long long = =(陨石波数*国家拥有观测点数*单次落下陨石数)

AC代码:

#include<stdio.h>
#include<vector>
#define N 300005
using namespace std;
typedef long long ll;
vector<ll>vt[N];
ll a[N];
ll need[N];
ll add[N*8];
ll ql[N],qr[N],qw[N];
ll ans[N];
ll q,n,m;
ll S[N],newS[N],pos;
void update(ll l,ll r,ll L,ll R,ll root,ll k)
{
	if(l<=L&&R<=r)
	{
		add[root]+=k;
		return ;
	}
	ll mid=L+R>>1;
	if(mid>=r)update(l,r,L,mid,root<<1,k);
	else if(mid<l)update(l,r,mid+1,R,root<<1|1,k);
	else 
	{
		update(l,mid,L,mid,root<<1,k);
		update(mid+1,r,mid+1,R,root<<1|1,k);
	}
}
ll query(ll x,ll L,ll R,ll root,ll ad)
{
	if(L==R)
		return ad+add[root];
	ll mid=L+R>>1;
	if(x<=mid)return query(x,L,mid,root<<1,ad+add[root]);
	else return query(x,mid+1,R,root<<1|1,ad+add[root]);
}
void solve(ll l,ll r,ll L,ll R)
{
	ll mid=l+r>>1;
	while(pos<=mid)
	{
		update(ql[pos],qr[pos],0,2*m-1,1,qw[pos]);
		pos++;
	}
	while(pos>=mid+2)
	{
		pos--;
		update(ql[pos],qr[pos],0,2*m-1,1,-qw[pos]);
	}
	if(l==r)
	{
		for(ll i=L;i<=R;i++)
		{
			ll sum=0;
			ll flag=1;
			for(ll j=0;j<vt[S[i]].size();j++)
			{
				ll to=vt[S[i]][j];
				sum+=query(to,0,2*m-1,1,0);
				sum+=query(to+m,0,2*m-1,1,0);
				if(sum>=need[S[i]])
				{
					flag=0;
					break;
				}
			}
			if(!flag)
				ans[S[i]]=l;
			else ans[S[i]]=-1;
		}
		return ;
	}
	ll l1=L-1,l2=R+1;
	for(ll i=L;i<=R;i++)
	{
		ll sum=0;
		ll flag=1;
		for(ll j=0;j<vt[S[i]].size();j++)
		{
			ll to=vt[S[i]][j];
			sum+=query(to,0,2*m-1,1,0);
			sum+=query(to+m,0,2*m-1,1,0);
			if(sum>=need[S[i]])
			{
				flag=0;
				break;
			}
		}
		if(!flag)
			newS[++l1]=S[i];
		else newS[--l2]=S[i];
	}
	for(ll i=L;i<=R;i++)
		S[i]=newS[i];
	if(l1!=L-1)solve(l,mid,L,l1);
	if(l2!=R+1)solve(mid+1,r,l2,R);
}
int main()
{
	scanf("%lld%lld",&n,&m);
	for(ll i=0;i<m;i++)
	{
		scanf("%lld",&a[i]);
		vt[a[i]-1].push_back(i);
	}
	for(ll i=0;i<n;i++)
		scanf("%lld",&need[i]);
	scanf("%lld",&q);
	for(ll i=0;i<q;i++)
	{
		scanf("%lld%lld%lld",&ql[i],&qr[i],&qw[i]);
		ql[i]--;qr[i]--;
		if(qr[i]<ql[i])qr[i]+=m;
	}
	pos=0;
	for(ll i=0;i<n;i++)S[i]=i;
	solve(0,q-1,0,n-1);
	for(ll i=0;i<n;i++)
	{
		if(ans[i]==-1)printf("NIE\n");
		else printf("%lld\n",ans[i]+1);
	}
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:27488次
    • 积分:1504
    • 等级:
    • 排名:千里之外
    • 原创:127篇
    • 转载:2篇
    • 译文:0篇
    • 评论:28条
    最新评论