#include "candies.h"
#include <vector>
struct node{
int tag, num;
} trees[200005 << 2];
inline int ls(int p){return p << 1;}
inline int rs(int p){return p << 1 | 1;}
void push_Down(int l, int r, int p){
int lc = ls(p), rc = rs(p), mid = (l + r) >> 1;
trees[lc].tag += trees[p].tag;
trees[rc].tag += trees[p].tag;
// trees[lc].num += trees[p].tag * (mid - l + 1);
// trees[rc].num += trees[p].tag * (r - mid);
trees[p].tag = 0;
return;
}
void push_Up(int p){
trees[p].num = trees[ls(p)].num + trees[rs(p)].num;
return;
}
// void build(int l, int r, int p, std::vector<int> &a){
// trees[p].tag = 0;
// if(l == r){
// trees[p].num = a[l];
// return;
// }
// int mid = (l + r) >> 1;
// build()
// }
void update(int l, int r, int nl, int nr, int p, int k){
if(nl <= l && r <= nr){
trees[p].tag += k;
// trees[p].num += k * (r - l + 1);
return;
}
push_Down(l, r, p);
int mid = (l + r) >> 1;
if(mid >= nl) update(l, mid, nl, nr, ls(p), k);
if(mid + 1 <= nr) update(mid + 1, r, nl, nr, rs(p), k);
// push_Up(p);
return;
}
int query(int l, int r, int n, int p){
if(l == r && l == n){
return p;
}
push_Down(l, r, p);
int mid = (l + r) >> 1;
if(mid >= n) return query(l, mid, n, ls(p));
else return query(mid + 1, r, n, rs(p));
}
std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
std::vector<int> r, std::vector<int> v) {
int n = c.size(), m = l.size();
c.insert(c.begin(), 0);
l.insert(l.begin(), 0);
r.insert(r.begin(), 0);
v.insert(v.begin(), 0);
std::vector<int> s(n);
for(int i = 1; i <= m; i++){
// printf("%d %d %d\n", l[i] + 1, r[i] + 1, v[i]);
update(1, n, l[i] + 1, r[i] + 1, 1, v[i]);
}
for(int i = 1; i <= n; i++){
int p = query(1, n, i, 1);
s[i - 1] = trees[p].tag >= c[i] ? c[i] : trees[p].tag;
}
return s;
}
Compilation message
candies.cpp: In function 'void push_Down(int, int, int)':
candies.cpp:14:33: warning: unused variable 'mid' [-Wunused-variable]
14 | int lc = ls(p), rc = rs(p), mid = (l + r) >> 1;
| ^~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
143 ms |
17352 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |