题目大意:有K次操作,每次操作给一段(或者两段)区间加一个数,然后在这段长度为m的区间内,每个点都有一个国家的接收器,每个国家都有一个值,求对于每个国家,他们的所有接收器的总和大于等于这个值的时间
整体二分的练手题...(之前好像写了挺长时间?)
二分答案,但是要对所有询问一起处理,二分完了答案之后我们可以把当前每个城市的陨石数量暴力算出来,然后把询问归成两类,第一类是已经满足条件了的,第二类是还没满足的,然后把第一类的答案在[l,mid]里二分,第二类的在[mid+1,r]里二分就可以了
至于如何动态维护每个点有多少陨石,可以用一个类似扫描线的东西(其实是整体二分一部分)和树状数组一起搞
#include<iostream>
#include<cstdio>
#include<vector>
#define N 300010
using namespace std;
vector<long long>a[N];
long long b[N];
long long c[N],d[N],w[N];
long long ans[N],fan[N];
long long C[N],P;
long long a1[N],a2[N];
long long n,m;
long long lowbit(long long x){return x&(-x);}
void change(long long x,long long v)
{
for(;x<=m;x+=lowbit(x))
C[x]+=v;
}
long long check(long long x)
{
long long ans=0;
for(;x;x-=lowbit(x))
ans+=C[x];
return ans;
}
void changeit(long long i,long long v)
{
change(c[i],v*w[i]);change(d[i]+1,-v*w[i]);
if(c[i]>d[i]) change(1,v*w[i]);
}
bool checkit(long long i)
{
long long j,ans=0;
for(j=0;j<a[i].size();j++)
{
ans+=check(a[i][j]);
if(ans>=b[i]) return true;
}
return false;
}
void solve(long long l,long long r,long long L,long long R)
{
long long i,j;
if(l>r||L==R)
{
for(i=l;i<=r;i++)
ans[fan[i]]=L;
return;
}
long long mid=(L+R)>>1;
while(P<mid) P++,changeit(P,1);
while(P>mid) changeit(P,-1),P--;
long long cnt1=0,cnt2=0;
for(i=l;i<=r;i++)
{
if(checkit(fan[i])) cnt1++,a1[cnt1]=fan[i];
else cnt2++,a2[cnt2]=fan[i];
}
long long k=l+cnt1;
for(i=1;i<=cnt1;i++) fan[i+l-1]=a1[i];
for(i=1;i<=cnt2;i++) fan[i+l+cnt1-1]=a2[i];
solve(l,k-1,L,mid);solve(k,r,mid+1,R);
}
int main()
{
scanf("%lld%lld",&n,&m);
long long i,j,x,y;
for(i=1;i<=m;i++)
{
scanf("%lld",&x);
a[x].push_back(i);
}
for(i=1;i<=n;i++)
{
scanf("%lld",&b[i]);
fan[i]=i;
}
long long q;
scanf("%lld",&q);
for(i=1;i<=q;i++)
scanf("%lld%lld%lld",&c[i],&d[i],&w[i]);
solve(1,n,1,q+1);
for(i=1;i<=n;i++)
{
if(ans[i]!=q+1&&ans[i]) printf("%lld\n",ans[i]);
else puts("NIE");
}
}