#include <bits/stdc++.h>
#ifdef LOCAL
#include "/home/trcmai/code/tools.h"
#define debug(x...) cerr << "\e[91m"<<__func__<<":"<<__LINE__<<" [" << #x << "] = ["; _print(x); cerr << "\e[39m" << endl;
#else
#define debug(x...)
#endif
using namespace std;
#define all(a) a.begin(), a.end()
#define ll long long
#define endl '\n'
const int N = 1e6 + 6, LOG = 27, MOD = 1e9 + 7;
const ll INF = 1e18;
int n,q;
namespace segtree{
int n;
vector<ll>tr,lz;
void init(int sz){
n = sz;
tr.resize(4*n + 5);
lz.resize(4*n + 5);
}
void push(int i,int l,int r){
if(lz[i] == 0) return;
tr[i] += lz[i];
if(l != r){
lz[2*i] += lz[i];
lz[2*i + 1] += lz[i];
}
lz[i] = 0;
}
void update(int i,int l,int r,int ql,int qr,int val){
push(i,l,r);
if(l > qr || r < ql) return;
if(ql <= l && r <= qr){
lz[i] += val;
push(i,l,r);
return;
}
int m = (r+l) >> 1;
update(2*i,l,m,ql,qr,val);
update(2*i+1,m+1,r,ql,qr,val);
tr[i] = max(tr[2*i],tr[2*i + 1]);
}
int walk(ll val){
if(tr[1] < val) return n + 1;
int i = 1,l = 1,r = n;
while(l < r){
push(i,l,r);
int m = (r+l) >> 1;
if(tr[2*i] >= val) i = 2*i,r = m;
else i = 2*i + 1, l = m + 1;
}
return l;
}
ll fval(int pos){
int i = 1,l = 1,r = n;
while(l < r){
push(i,l,r);
int m = (r+l) >> 1;
if(pos <= m) i = 2*i,r = m;
else i = 2*i + 1,l = m + 1;
}
return tr[i];
}
void update(int l,int r,ll val){ update(1,1,n,l,r,val);}
}
signed main() {
cin.tie(0)->sync_with_stdio(0);
auto solver=[&](){
cin>>n>>q;
segtree::init(n);
vector<int>a(n + 1);
for(int i = 1;i <= n;++i) cin>>a[i];
sort(a.begin() + 1,a.end());
for(int i = 1;i <= n;++i){
segtree::update(i,i,a[i]);
}
while(q--){
char t;
cin>>t;
if(t == 'F'){
int tot; ll val; cin>>tot>>val;
int l = segtree::walk(val);
if(l > n) continue;
tot = min(tot,n - l + 1);
ll last_val = segtree::fval(l + tot - 1);
//update first range
int r = segtree::walk(last_val) - 1;
segtree::update(l,r,1);
//update second range
int r2 = segtree::walk(last_val + 1) - 1;
int l2 = r2 - (tot - (r - l + 1)) + 1;
segtree::update(l2,r2,1);
}else{
ll u,v; cin>>u>>v;
int l = segtree::walk(u);
int r = segtree::walk(v + 1);
cout<< r - l << endl;
}
}
};
int t = 1; // cin>>t;
while (t--) solver();
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
38 ms |
7308 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 |
16 ms |
1884 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
24 ms |
2140 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
49 ms |
5800 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
40 ms |
6448 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
31 ms |
6748 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
93 ms |
7736 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
42 ms |
7508 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
66 ms |
9264 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |