#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef tuple<int,int,int> t3;
const int N=3e5+5;
struct fenwick{
ll tree[N]={};
void init(){
for(int i=0;i<N;++i)tree[i]=0;
}
void add(int i,ll v){
while(i<N){
tree[i]+=v;
i+=i&-i;
}
}
ll read(int i){
ll sum=0;
while(i>0){
sum+=tree[i];
i-=i&-i;
}
return sum;
}
}f;
int n,m,k;
int o[N],l[N],r[N],ans[N];
ll p[N];
bool can[N];
ll sum[N];
vector<t3> v;
vector<int> res[N];
int main(){
cin.tie(nullptr)->sync_with_stdio(false);
cin >> n >> m;
for(int i=1;i<=m;++i)cin >> o[i];
for(int i=1;i<=n;++i)cin >> p[i];
cin >> k;
v.emplace_back(0,0,0);
for(int i=1;i<=k;++i){
int a,b,c;
cin >> a >> b >> c;
v.emplace_back(a,b,c);
}
for(int i=1;i<=n;++i)l[i]=0,r[i]=k+1;
while(true){
for(int i=1;i<=k;++i)res[i].clear();
int cnt=0;
for(int i=1;i<=n;++i){
if(l[i]+1>=r[i])continue;
sum[i]=0;
res[(l[i]+r[i])/2].emplace_back(i);
++cnt;
}
if(cnt==0)break;
for(int i=1;i<=k;++i){
if(res[i].empty())continue;
f.init();
for(int j=1;j<=i;++j){
auto [a,b,c]=v[j];
f.add(a,c);
f.add(b+1,-c);
if(b<a){
f.add(1,c);
f.add(m+1,-c);
}
}
for(int j=1;j<=m;++j)sum[o[j]]+=f.read(j);
for(auto j:res[i]){
if(sum[j]>=p[j])r[j]=i;
else l[j]=i;
if(l[j]+1>=r[j])ans[j]=r[j];
}
}
}
for(int i=1;i<=n;++i){
if(ans[i]<=k)cout << ans[i] << '\n';
else cout << "NIE" << '\n';
}
return 0;
}
Compilation message
met.cpp: In function 'int main()':
met.cpp:66:22: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
66 | auto [a,b,c]=v[j];
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
16 ms |
9792 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
8 ms |
9824 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
124 ms |
11300 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
303 ms |
11908 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5098 ms |
11392 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
102 ms |
10944 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1016 ms |
26020 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1425 ms |
24960 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |