#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/stdc++.h>
using namespace std;
const int inf = 1e9 + 7;
const int N = 1e6 + 7;
struct SEGT{
int t[N];
int tree_size;
const int identity_element = 0;
int merge(int x , int y){
return min(inf , x + y);
}
void init(int x){
tree_size = x+3;
memset(t , 0 , sizeof(t));
}
void modify(int p, int value) { // set value at position p
p += tree_size;
t[p] = min(inf , t[p] + value);
for (; p > 1; p >>= 1){
t[p>>1] = merge(t[p] , t[p^1]);
}
}
int _query(int l, int r) { // sum on interval [l, r]
int res = identity_element;
for (r+=1 , l += tree_size, r += tree_size; l < r; l >>= 1, r >>= 1) {
if (l&1) res = merge(res , t[l++]);
if (r&1) res = merge(res , t[--r]);
}
return res;
}
void rangeupdate(int l , int r , int val){
// cout << "rangeupdate : " << l << " " << r << " " << val << endl;
modify(l , val);
modify(r+1 , -val);
}
int query(int p){
return _query(0,p);
}
};
void solve(){
int n,m;cin >> n >> m;
vector < int > devlet[n+1];
for(int i = 1;i<=m;i++){
int x;cin >> x;
devlet[x].push_back(i);
}
int need[n+1];
for(int i = 1;i<=n;i++){
cin >> need[i];
}
int q;cin >> q;
vector < array < int , 3 > > query(q+1);
for(int i = 1;i<=q;i++){
cin >> query[i][0] >> query[i][1] >> query[i][2];
}
vector < array < int , 3 > > paralel[q+2],nparalel[q+2];
for(int i = 1;i<=n;i++){
paralel[(1+q+1)/2].push_back({i,1,q+1});
}
SEGT fen;
for(int i = 0;i<=19;i++){
fen.init(m+10);
for(int j = 1;j<=q;j++){
if(query[j][0] <= query[j][1]){
fen.rangeupdate(query[j][0],query[j][1],query[j][2]);
}
else{
fen.rangeupdate(1,query[j][1],query[j][2]);
fen.rangeupdate(query[j][0],m+1,query[j][2]);
}
// cout << "j : ";for(int k = 1;k<=m;k++)cout << fen._query(k,k) << " ";cout << endl;
for(auto itr : paralel[j]){
if(itr[1] == itr[2]){
nparalel[j].push_back(itr);
continue;
}
int cur = 0;
for(auto itr1 : devlet[itr[0]]){
cur = min(inf , cur + fen.query(itr1));
}
if(cur >= need[itr[0]]){//r = mid;
int mid = (itr[1] + j)/2;
nparalel[mid].push_back({itr[0],itr[1],j});
}
else {//l = mid
int mid = (j+1 + itr[2])/2;
nparalel[mid].push_back({itr[0],j+1,itr[2]});
}
}
}
for(int j = 1;j<=q;j++){
paralel[j].swap(nparalel[j]);
nparalel[j].clear();
}
}
int ans[n+1];
memset(ans , -1 , sizeof(ans));
for(int j = 1;j<=q;j++){
for(auto itr : paralel[j]){
ans[itr[0]] = j;
}
}
for(int i = 1;i<=n;i++){
if(ans[i] == -1)cout << "NIE" << '\n';
else cout << ans[i] << '\n';
}
}
signed main(){
ios_base::sync_with_stdio(0);cin.tie(0);
int testcase = 1;//cin >> testcase;
while(testcase--)solve();
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
4444 KB |
Output is correct |
2 |
Correct |
6 ms |
4440 KB |
Output is correct |
3 |
Correct |
7 ms |
4444 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6 ms |
4444 KB |
Output is correct |
2 |
Correct |
5 ms |
4440 KB |
Output is correct |
3 |
Correct |
6 ms |
4444 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
115 ms |
8724 KB |
Output is correct |
2 |
Correct |
167 ms |
13664 KB |
Output is correct |
3 |
Correct |
153 ms |
11912 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
171 ms |
10772 KB |
Output is correct |
2 |
Correct |
151 ms |
10936 KB |
Output is correct |
3 |
Correct |
175 ms |
14360 KB |
Output is correct |
4 |
Correct |
57 ms |
8584 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
110 ms |
9372 KB |
Output is correct |
2 |
Correct |
157 ms |
14500 KB |
Output is correct |
3 |
Correct |
53 ms |
7256 KB |
Output is correct |
4 |
Correct |
150 ms |
13160 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
140 ms |
7512 KB |
Output is correct |
2 |
Correct |
157 ms |
11184 KB |
Output is correct |
3 |
Correct |
114 ms |
8092 KB |
Output is correct |
4 |
Correct |
168 ms |
15948 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1342 ms |
54236 KB |
Output is correct |
2 |
Correct |
1014 ms |
23276 KB |
Output is correct |
3 |
Correct |
257 ms |
19024 KB |
Output is correct |
4 |
Runtime error |
1162 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1394 ms |
51032 KB |
Output is correct |
2 |
Incorrect |
807 ms |
23512 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |