# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
950347 | TrieTr | Sum Zero (RMI20_sumzero) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define fastio ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
template<class X, class Y> bool mini(X &a, const Y &b) {return (a > b) ? a = b, true : false;}
template<class X, class Y> bool maxi(X &a, const Y &b) {return (a < b) ? a = b, true : false;}
const int N = 4e5 + 5;
int n, q, res;
int arr[N];
unordered_map<ll, int> pos;
int nxt[N];
int h[N], jump[N];
void make_leaf(int& id, int& p) {
h[id] = h[p] + 1;
if(h[jump[p]] - h[p] == h[jump[jump[p]]] - h[jump[p]]) jump[id] = jump[jump[p]];
else jump[id] = p;
}
int main() {
fastio;
cin >> n;
for(int i = 1; i <= n; i++) cin >> arr[i];
ll sum = 0;
nxt[n + 1] = n + 1;
pos[0] = n + 1;
for(int i = n; i > 0; i--) {
sum += arr[i];
nxt[i] = pos[sum] - 1;
if(nxt[i] == -1) nxt[i] = nxt[i + 1];
else mini(nxt[i], nxt[i + 1]);
pos[sum] = i;
}
for(int i = n; i > 0; i--) {
int u = nxt[i];
if(jump[u] > 0 || u > n) continue;
vis.set(u);
int p = nxt[u + 1];
make_leaf(u, p);
}
cin >> q;
while(q--) {
int l, r; cin >> l >> r;
l = nxt[l];
if(l > r) {
cout << "0\n"; continue;
}
res = 0;
while(l <= r && l != 0) {
if(jump[l] <= r && jump[l] != 0) {
res += h[l] - h[jump[l]];
l = jump[l];
}
else {
res++;
l = nxt[l + 1];
}
}
cout << res << '\n';
}
}