Byteotian Interstellar Union有N个成员国。现在它发现了一颗新的星球,这颗星球的轨道被分为M份(第M份和第1份相邻),第i份上有第Ai个国家的太空站。
这个星球经常会下陨石雨。BIU已经预测了接下来K场陨石雨的情况。
BIU的第i个成员国希望能够收集Pi单位的陨石样本。你的任务是判断对于每个国家,它需要在第几次陨石雨之后,才能收集足够的陨石。
输入:
第一行是两个数N,M。
第二行有M个数,第i个数Oi表示第i段轨道上有第Oi个国家的太空站。
第三行有N个数,第i个数Pi表示第i个国家希望收集的陨石数量。
第四行有一个数K,表示BIU预测了接下来的K场陨石雨。
接下来K行,每行有三个数Li,Ri,Ai,表示第K场陨石雨的发生地点在从Li顺时针到Ri的区间中(如果Li<=Ri,就是Li,Li+1,…,Ri,否则就是Ri,Ri+1,…,m-1,m,1,…,Li),向区间中的每个太空站提供Ai单位的陨石样本。
输出:
N行。第i行的数Wi表示第i个国家在第Wi波陨石雨之后能够收集到足够的陨石样本。如果到第K波结束后仍然收集不到,输出NIE。
数据范围:
1<=n,m,k<=3*10^5
1<=Pi<=10^9
1<=Ai<10^9
Sample Input
3 5
1 3 2 1 3
10 5 7
3
4 2 4
1 3 1
3 5 2
Sample Output
3
NIE
1
题解:
写的第一道整体二分。感觉整体二分和CDQ分治很像,但是CDQ分治是分治所有的操作,整体二分则相当于只对查询的进行二分。不过他们都针对于离线。
如果这个题只有一个国家,那么可以二分答案,统计是否可行。这样对于一些国家一起二分也是可以的,需要用到树状数组。
每次二分流星雨次数mid,可达到的国家答案放在左边,不行的放在右边,做完后再递归解决左右的,于是相当于降了一维复杂度,每次给所有的国家划定一个答案的上(下)界,直到达到准确答案。
有两个小细节:
1.如果所有的地方都是一个国家的,统计答案时会爆long long。因此,只要CntAns>p[i],就break,反正不影响答案。
2.有的国家最后也收集不够,所以在最后下一场所有地区,大小为inf的流星雨,只要统计答案时是inf的,就输出NIE。
代码如下:
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#include<vector>
#define ll long long
#define inf 0x7f7f7f7f
#define N 300005
#define lb(x) (x&(-x))
using namespace std;
ll read()
{
ll x=0;
char c=getchar();
while(c<'0' || c>'9') c=getchar();
while(c<='9' && c>='0'){x=x*10+c-'0';c=getchar();}
return x;
}
int T,n,m,x,p[N],k,l[N],r[N],v[N],id[N];
int ans[N],tmp[N];
bool vis[N];
vector<int> a[N];
ll t[N];
void ins(int x,int v)
{
while(x<=m)
{
t[x]+=v;
x+=lb(x);
}
}
ll ask(int x)
{
ll ret=0;
while(x)
{
ret+=t[x];
x-=lb(x);
}
return ret;
}
void insert(int x,int op)
{
if(l[x]<=r[x])
{
ins(l[x],op*v[x]);
ins(r[x]+1,-op*v[x]);
}
else
{
ins(1,op*v[x]);
ins(r[x]+1,-op*v[x]);
ins(l[x],op*v[x]);
}
}
void solve(int l,int r,int L,int R)
{
if(l>r) return;
if(L==R)
{
for(int i=l;i<=r;i++) ans[id[i]]=L;
return ;
}
int mid=(L+R)>>1;
while(T<=mid) T++,insert(T,1);
while(T>mid) insert(T,-1),T--;
int cnt=0,now;
ll tot;
for(int i=l;i<=r;i++)
{
tot=0;now=id[i];
for(int j=0;j<a[now].size();j++)
{
tot+=ask(a[now][j]);
if(tot>=p[now]) break;
}
if(tot>=p[now]) vis[now]=1,cnt++;
else vis[now]=0;
}
int l1=l,l2=l+cnt;
for(int i=l;i<=r;i++)
{
if(vis[id[i]]) tmp[l1++]=id[i];
else tmp[l2++]=id[i];
}
for(int i=l;i<=r;i++) id[i]=tmp[i];
solve(l,l1-1,L,mid);solve(l1,l2-1,mid+1,R);
}
int main()
{
n=read(),m=read();
for(int i=1;i<=m;i++)
{
x=read();
a[x].push_back(i);
}
for(int i=1;i<=n;i++) p[i]=read();
k=read();
for(int i=1;i<=k;i++) l[i]=read(),r[i]=read(),v[i]=read();
l[++k]=1;r[k]=m;v[k]=inf;
for(int i=1;i<=n;i++) id[i]=i;
solve(1,n,1,k);
for(int i=1;i<=n;i++)
{
if(ans[i]!=k) printf("%d\n",ans[i]);
else puts("NIE");
}
return 0;
}