#include<bits/stdc++.h>
#define N 300001
#define rep(i,j,k) for(I i=j;i<=k;i++)
using namespace std;
#define I long long
I bit[N],ans[N],L[N],R[N],a[N];
struct c{I g, id;vector<I>st;};
void u(I x, I y) {
for(;x<N;x+=x&-x)
bit[x]+=y;
}
void upd(I l, I r, I x) {
u(l,x),u(r+1,-x);
if(l>r)u(1,x);
}
I q(I x) {
I r = 0;
while(x)r+=bit[x],x-=x&-x;
return r;
}
c X[N];
void solve(I l, I r, vector<int> cs){
if(l==r) {
for(auto i: cs) ans[X[i].id] = l;
l++;
}
if(cs.empty()||l>r) return;
I m=l+r>>1;
rep(i,l,m)
upd(L[i],R[i],a[i]);
vector<int>c1,c2;
for(auto i: cs) {
I x=0;
for(auto j:X[i].st)x+=q(j);
if(x<X[i].g) X[i].g-=x,c2.push_back(i);
else c1.push_back(i);
}
cs.clear();
rep(i,l,m) upd(L[i],R[i],-a[i]);
solve(l,m,c1);
solve(m+1,r,c2);
}
int main() {
I n,m,k;
cin>>n>>m;
vector<int> v(n);
iota(v.begin(),v.end(),0);
rep(i,1,m){
I x;
cin>>x;
X[x-1].st.push_back(i);
}
rep(i,0,n-1)cin>>X[i].g,X[i].id=i+1;
cin>>k;
rep(i,1,k)cin>>L[i]>>R[i]>>a[i];
L[k+1]=1,R[k+1]=m,a[k+1]=1e18;
solve(1,k+1,v);
rep(i,1,n)
if(ans[i]<=k) cout << ans[i] <<'\n';
else cout << "NIE\n";
}
Compilation message
met.cpp: In function 'void solve(long long int, long long int, std::vector<int>)':
met.cpp:28:10: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
28 | I m=l+r>>1;
| ~^~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
12116 KB |
Output is correct |
2 |
Correct |
7 ms |
12116 KB |
Output is correct |
3 |
Correct |
11 ms |
12140 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
8 ms |
12064 KB |
Output is correct |
2 |
Correct |
8 ms |
12120 KB |
Output is correct |
3 |
Correct |
8 ms |
12116 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
63 ms |
14924 KB |
Output is correct |
2 |
Correct |
140 ms |
17800 KB |
Output is correct |
3 |
Correct |
126 ms |
14936 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
144 ms |
14660 KB |
Output is correct |
2 |
Correct |
131 ms |
14704 KB |
Output is correct |
3 |
Correct |
104 ms |
16588 KB |
Output is correct |
4 |
Correct |
42 ms |
14592 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
129 ms |
14692 KB |
Output is correct |
2 |
Correct |
185 ms |
16948 KB |
Output is correct |
3 |
Correct |
63 ms |
13276 KB |
Output is correct |
4 |
Correct |
135 ms |
15132 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
84 ms |
14152 KB |
Output is correct |
2 |
Correct |
148 ms |
14792 KB |
Output is correct |
3 |
Correct |
86 ms |
14368 KB |
Output is correct |
4 |
Correct |
169 ms |
16012 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
803 ms |
37652 KB |
Output is correct |
2 |
Incorrect |
501 ms |
23928 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
670 ms |
34376 KB |
Output is correct |
2 |
Incorrect |
471 ms |
23604 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |