제출 #597661

#제출 시각아이디문제언어결과실행 시간메모리
597661Hacv16Sum Zero (RMI20_sumzero)C++17
22 / 100
93 ms21724 KiB
#include<bits/stdc++.h>
using namespace std;

#pragma GCC optimize("O3")
#pragma GCC target("avx2")

typedef long long ll;
typedef pair<int, int> pii;

const int MAX = 100001;
const int LOG = 20;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;

#define pb push_back
#define sz(x) (int) x.size()
#define fr first
#define sc second
#define mp make_pair
#define all(x) x.begin(), x.end()
#define dbg(x) cout << #x << ": " << "[ " << x << " ]\n"

ll n, v[MAX], jump[MAX][LOG], p[MAX], m[MAX], q;

ll query(int l, int r){
    ll ans = 0;

    for(int j = LOG - 1; j >= 0; j--){
        if(jump[r][j] >= l - 1){
            ans += (1 << j);
            r = jump[r][j];
        }
    }

    return ans;
}

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); 

    cin >> n;

    for(int i = 1; i <= n; i++)
        cin >> v[i];

    unordered_map<ll, ll> sums; 
    sums[0] = 0;

    for(ll i = 1, s = 0; i <= n; i++){
        s += v[i];

        if(sums.find(s) != sums.end()) 
            p[i] = sums[s];
        else p[i] = -1;

        sums[s] = i;
    }

    for(int i = 0; i < LOG; i++)
        jump[0][i] = -1;

    for(int i = 1; i <= n; i++){
        if(p[i] < jump[i - 1][0]) jump[i][0] = jump[i - 1][0];
        else jump[i][0] = p[i];
    }

    for(int j = 1; j < LOG; j++){
        for(int i = 1; i <= n; i++){
            if(jump[i][j - 1] != -1) jump[i][j] = jump[jump[i][j - 1]][j - 1];
            else jump[i][j] = -1;
        }
    }    

    cin >> q;

    while(q--){
        int l, r; cin >> l >> r;
        cout << query(l, r) << '\n';
    }

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...