题目
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。
输入样例
3 5
1 3 2 1 3
10 5 7
3
4 2 4
1 3 1
3 5 2
输出样例
3
NIE
1
提示
1<=n,m,k<=3*10^5
1<=Pi<=10^9
1<=Ai<10^9
题解
显然我们需要二分每个国家最早到达需求的时刻,但我们不可能每次都从头统计一遍
那就整体二分就搞定了√
具体地,每次对一个区间前半的操作进行操作,然后查看当前区间的询问是否满足,不满足就丢到左区间继续查询,否则就减去当前值后丢到右区间继续查询
#include<iostream>
#include<cstdio>
#include<cmath>
#include<vector>
#include<cstring>
#include<algorithm>
#define lbt(x) (x & -x)
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");
using namespace std;
const int maxn = 300005,maxm = 10000005,INF = 1000000000;
inline int read(){
int out = 0,flag = 1; char c = getchar();
while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
return out * flag;
}
int n,m,K,q[maxn],p[maxn],t[maxn],ans[maxn];
LL S[maxn];
vector<int> at[maxn];
struct node{int l,r,v;}e[maxn];
void add(int u,LL v){while (u <= m) S[u] += v,u += lbt(u);}
LL query(int u){LL ans = 0; while (u) ans += S[u],u -= lbt(u); return ans;}
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[q[i]] = l;
return;
}
int mid = l + r >> 1;
for (int i = l; i <= mid && i <= K; i++){
if (e[i].l <= e[i].r){
add(e[i].l,e[i].v);
add(e[i].r + 1,-e[i].v);
}
else {
add(1,e[i].v);
add(e[i].r + 1,-e[i].v);
add(e[i].l,e[i].v);
}
}
int li = L,ri = R;
LL tmp;
for (int i = L; i <= R; i++){
int u = q[i];
tmp = 0;
for (unsigned int j = 0; j < at[u].size(); j++)
tmp += query(at[u][j]);
if (tmp < p[u]) p[u] -= tmp,t[ri--] = u;
else t[li++] = u;
}
for (int i = L; i <= R; i++) q[i] = t[i];
for (int i = l; i <= mid && i <= K; i++){
if (e[i].l <= e[i].r){
add(e[i].l,-e[i].v);
add(e[i].r + 1,e[i].v);
}
else {
add(1,-e[i].v);
add(e[i].r + 1,e[i].v);
add(e[i].l,-e[i].v);
}
}
solve(l,mid,L,li - 1); solve(mid + 1,r,ri + 1,R);
}
int main(){
n = read(); m = read();
for (int i = 1; i <= m; i++){
int u = read();
at[u].push_back(i);
}
for (int i = 1; i <= n; i++) p[i] = read();
K = read();
for (int i = 1; i <= K; i++){
e[i].l = read();
e[i].r = read();
e[i].v = read();
}
for (int i = 1; i <= n; i++) q[i] = i;
solve(1,K + 1,1,n);
for (int i = 1; i <= n; i++){
if (ans[i] == K + 1) puts("NIE");
else printf("%d
",ans[i]);
}
return 0;
}