#include <bits/stdc++.h>
#define ios ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define file(s) if (fopen(s".in", "r")) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
#define all(a) a.begin() , a.end()
#define F first
#define S second
using namespace std;
using ll = long long;
const int N = 3e5+5 , inf = 2e9 + 7;
const ll INF = 1e18 , mod = 1e9+7 , P = 6547;
int p[N] , n , m , k , L[N] , R[N] , res[N];
ll t[N*4];
int l[N] , r[N] , c[N];
vector<int> g[N] , q[N] , q1[N];
void build(int v , int tl , int tr){
t[v] = 0;
if(tl == tr) return;
int tm = (tl+tr) >> 1;
build(v*2,tl,tm);
build(v*2+1,tm+1,tr);
}
void upd(int v, int tl , int tr , int l , int r , int val){
if(l <= tl && tr <= r){
t[v] += val;
return;
}
if(tl > r || tr < l) return;
int tm = (tl+tr) >> 1;
upd(v*2,tl,tm,l,r,val);
upd(v*2+1,tm+1,tr,l,r,val);
}
ll get(int v, int tl , int tr , int pos){
if(tl == tr) return t[v];
int tm = (tl+tr) >> 1;
if(tm >= pos) return get(v*2,tl,tm,pos)+t[v];
else return get(v*2+1,tm+1,tr,pos)+t[v];
}
void solve(){
cin >> n >> m;
for(int i = 1; i <= m; i++) {
int a;
cin >> a;
g[a].push_back(i);
}
for(int i = 1; i <= n; i++) cin >> p[i];
cin >> k;
for(int i = 1; i <= n; i++) {
L[i] = 1;
R[i] = k;
q[(L[i]+R[i])/2].push_back(i);
}
for(int i = 1; i <= k; i++) cin >> l[i] >> r[i] >> c[i];
int ok = 1;
while(ok){
ok = 0;
build(1,1,m);
for(int i = 1; i <= k; i++){
if(l[i] > r[i]){
upd(1,1,m,l[i],m,c[i]);
upd(1,1,m,1,r[i],c[i]);
} else {
upd(1,1,m,l[i],r[i],c[i]);
}
for(int x : q[i]){
ll sum = p[x];
for(int y : g[x]) sum -= get(1,1,m,y);
if(sum <= 0){
res[x] = i;
R[x] = i-1;
} else {
L[x] = i+1;
}
if(L[x] <= R[x]){
q1[(L[x]+R[x])/2].push_back(x);
ok++;
}
}
q[i].clear();
}
swap(q,q1);
}
for(int i = 1; i <= n; i++) {
if(res[i] == 0){
cout << "NIE\n";
} else {
cout << res[i] << "\n";
}
}
}
/*
*/
signed main(){
ios;
solve();
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
18 ms |
21460 KB |
Output is correct |
2 |
Correct |
18 ms |
21592 KB |
Output is correct |
3 |
Correct |
19 ms |
21460 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
19 ms |
21560 KB |
Output is correct |
2 |
Correct |
20 ms |
21560 KB |
Output is correct |
3 |
Correct |
19 ms |
21620 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
275 ms |
23796 KB |
Output is correct |
2 |
Correct |
327 ms |
25924 KB |
Output is correct |
3 |
Correct |
318 ms |
25548 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
291 ms |
24908 KB |
Output is correct |
2 |
Correct |
296 ms |
24992 KB |
Output is correct |
3 |
Correct |
326 ms |
26060 KB |
Output is correct |
4 |
Correct |
80 ms |
24428 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
159 ms |
24388 KB |
Output is correct |
2 |
Correct |
221 ms |
26608 KB |
Output is correct |
3 |
Correct |
142 ms |
22184 KB |
Output is correct |
4 |
Correct |
332 ms |
25904 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
301 ms |
23508 KB |
Output is correct |
2 |
Correct |
276 ms |
26152 KB |
Output is correct |
3 |
Correct |
272 ms |
24796 KB |
Output is correct |
4 |
Correct |
319 ms |
28504 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2631 ms |
46928 KB |
Output is correct |
2 |
Incorrect |
1377 ms |
34528 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2636 ms |
45800 KB |
Output is correct |
2 |
Incorrect |
1047 ms |
34516 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |