#include <map>
#include <set>
#include <list>
#include <cmath>
#include <ctime>
#include <deque>
#include <queue>
#include <stack>
#include <string>
#include <bitset>
#include <cstdio>
#include <limits>
#include <vector>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <fstream>
#include <numeric>
#include <sstream>
#include <cassert>
#include <iomanip>
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <fstream>
#include <unordered_map>
using namespace std;
typedef long long ll;
typedef long double ld;
//#define int long long
#define all(x) x.begin(),x.end()
#define pb push_back
#define ent "\n"
const int maxn = (int)3e5 + 13;
const ll inf = (long long)1e18 + 20;
const int mod = (int)1e9 + 7;
int n,m;
vector<int>pos[maxn],MID[maxn];
int a[maxn];
int l[maxn],r[maxn],w[maxn];
pair<int,int>seg[maxn];
ll t[maxn * 4],q[maxn * 4];
void push(int v,int tl,int tr){
if(q[v]){
t[v] += q[v] * (tr - tl + 1);
if(tl < tr){
q[v * 2] += q[v];
q[v * 2 + 1] += q[v];
}
q[v] = 0;
}
}
void upd(int v,int tl,int tr,int l,int r,int x){
push(v,tl,tr);
if(l <= tl && tr <= r){
q[v] += x;
push(v,tl,tr);
return;
}
int mid = (tl + tr) >> 1;
if(l > tr || r < tl)return;
upd(v * 2,tl,mid,l,r,x);
upd(v * 2 + 1,mid + 1,tr,l,r,x);
t[v] = t[v * 2] + t[v * 2 + 1];
}
ll get(int v,int tl,int tr,int pos){
push(v,tl,tr);
if(tl == tr){
return t[v];
}
int mid = (tl + tr) >> 1;
if(pos <= mid){
return get(v * 2,tl,mid,pos);
}
else{
return get(v * 2 + 1,mid + 1,tr,pos);
}
}
void build(int v,int tl,int tr){
if(tl == tr){
t[v] = 0;
q[v] = 0;
return;
}
int mid = (tl + tr) >> 1;
build(v * 2,tl,mid);
build(v * 2 + 1,mid + 1,tr);
t[v] = 0,q[v] = 0;
}
ll ans[maxn];
void solve(){
cin >> n >> m;
for(int i = 1 ; i <= m ; i ++){
int o;cin >> o;
pos[o].pb(i);
}
for(int i = 1 ; i <= n ; i ++){
cin >> a[i];
}
int k;cin >> k;
for(int i = 1 ; i <= k ; i++){
cin >> l[i] >> r[i] >> w[i];
}
for(int i = 1 ; i <= n ; i ++){
seg[i] = {1, k};
}
int ok = 1;
while(ok){
build(1,1,m), ok = 0;
for(int i = 1 ; i <= n ; i ++){
if(seg[i].first <= seg[i].second){
int mid = (seg[i].first + seg[i].second) >> 1;
MID[mid].pb(i);
ok = 1;
}
}
for(int i = 1 ; i <= k ; i ++){
if(l[i] <= r[i]){
upd(1,1,m,l[i],r[i],w[i]);
}
else{
upd(1,1,m,l[i],m,w[i]);
upd(1,1,m,1,r[i],w[i]);
}
ll sum = 0;
for(int x : MID[i]){
sum = 0;
for(int y : pos[x]){
sum += get(1,1,m,y);
if(sum >= a[x])break;
}
if(sum >= a[x]){
seg[x].second = i - 1;
ans[x] = i;
}
else{
seg[x].first = i + 1;
}
}
MID[i].clear();
}
}
for(int i = 1 ; i <= n ; i ++){
if (ans[i] == 0) cout << "NIE\n";
else cout << ans[i] << '\n';
}
}
signed main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
int t = 1;
//cin >> t;
while(t --){
solve();
}
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
8 ms |
25948 KB |
Output is correct |
2 |
Correct |
9 ms |
26172 KB |
Output is correct |
3 |
Correct |
9 ms |
25948 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
8 ms |
25948 KB |
Output is correct |
2 |
Correct |
8 ms |
25948 KB |
Output is correct |
3 |
Correct |
11 ms |
26200 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
418 ms |
28824 KB |
Output is correct |
2 |
Correct |
510 ms |
30800 KB |
Output is correct |
3 |
Correct |
491 ms |
30484 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
477 ms |
30032 KB |
Output is correct |
2 |
Correct |
500 ms |
29828 KB |
Output is correct |
3 |
Correct |
489 ms |
32968 KB |
Output is correct |
4 |
Correct |
74 ms |
29780 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
216 ms |
29196 KB |
Output is correct |
2 |
Correct |
378 ms |
33548 KB |
Output is correct |
3 |
Correct |
240 ms |
26428 KB |
Output is correct |
4 |
Correct |
475 ms |
30800 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
470 ms |
28752 KB |
Output is correct |
2 |
Correct |
420 ms |
29776 KB |
Output is correct |
3 |
Correct |
466 ms |
28760 KB |
Output is correct |
4 |
Correct |
512 ms |
34200 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4650 ms |
55212 KB |
Output is correct |
2 |
Correct |
2553 ms |
41812 KB |
Output is correct |
3 |
Correct |
1281 ms |
26648 KB |
Output is correct |
4 |
Runtime error |
4177 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4617 ms |
53916 KB |
Output is correct |
2 |
Correct |
1381 ms |
42020 KB |
Output is correct |
3 |
Correct |
1033 ms |
25948 KB |
Output is correct |
4 |
Runtime error |
3559 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Halted |
0 ms |
0 KB |
- |