题目梗概
有n个国家,需要pi个陨石。
在每个时间在某个区间会落下x个陨石。
问每个国家何时收集完陨石。
解题思路
比较明显的整体二分。
二分时间t,表示每个国家何时收集完陨石。
把时间t之前的所有陨石砸下(可以用树状数组维护),判断是否收集完然后继续二分。
#include<cstdio>
#define LL long long
#define lowbit(x) ((x)&-(x))
using namespace std;
inline int _read(){
int num=0;char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
return num;
}
const int maxn=300005;
struct jz{
int L,R,x;
}a[maxn];
int n,m,p[maxn],k,ans[maxn],h[maxn],t[maxn];
int tot,lnk[maxn],nxt[maxn],son[maxn];
LL w[maxn];
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void change(int x,int y){while (x<=m) w[x]+=y,x+=lowbit(x);}
LL query(int x){LL num=0;while(x>0) num+=w[x],x-=lowbit(x);return num;}
void work(int L1,int R1,int L2,int R2){
if (L2>R2) return;
if (L1==R1){for (int i=L2;i<=R2;i++) ans[h[i]]=L1;return;}
int mid=L1+R1>>1;
for (int i=L1;i<=mid;i++)
if (a[i].L<=a[i].R) change(a[i].L,a[i].x),change(a[i].R+1,-a[i].x);else
change(a[i].L,a[i].x),change(1,a[i].x),change(a[i].R+1,-a[i].x);
int i1=L2-1,i2=R2+1;
for (int i=L2;i<=R2;i++){
LL num=0;
for (int j=lnk[h[i]];j;j=nxt[j]){
num+=query(son[j]);
if (num>=p[h[i]]) break;
}
if (num>=p[h[i]]) t[++i1]=h[i];else t[--i2]=h[i],p[h[i]]-=num;
}
for (int i=L2;i<=R2;i++) h[i]=t[i];
for (int i=L1;i<=mid;i++)
if (a[i].L<=a[i].R) change(a[i].L,-a[i].x),change(a[i].R+1,a[i].x);else
change(a[i].L,-a[i].x),change(1,-a[i].x),change(a[i].R+1,a[i].x);
work(L1,mid,L2,i1);work(mid+1,R1,i1+1,R2);
}
int main(){
freopen("exam.in","r",stdin);
freopen("exam.out","w",stdout);
n=_read(),m=_read();
for (int i=1;i<=m;i++){int x=_read();add(x,i);}
for (int i=1;i<=n;i++) p[i]=_read(),h[i]=i;
k=_read();
for (int i=1;i<=k;i++) a[i].L=_read(),a[i].R=_read(),a[i].x=_read();
a[k+1].L=1;a[k+1].R=m;a[k+1].x=1e9;
work(1,k+1,1,n);
for (int i=1;i<=n;i++)
if (ans[i]>k) printf("NIE\n");else printf("%d\n",ans[i]);
return 0;
}