Submission #1080406

#	Time	Username	Problem	Language	Result	Execution time	Memory
1080406		batsukh2006	Sum Zero (RMI20_sumzero)	C++17	61 / 100	651 ms	33512 KiB

sumzero

#include<iostream>
#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<deque>
#include<stack>
#include<string>
#include<math.h>
#include<vector>
#include<stdio.h>
#include<utility>
#include<iomanip>
#include<string.h>
#include<limits.h>
#include<algorithm>
#include<functional>
#include<unordered_map> 
using namespace std;
#pragma GCC target("popcnt")
#define MOD 1000000007
// #define int long long
#define ss second
#define ff first
#define endl '\n'
void solve(){
    int n; cin>>n;
    vector<int> a(n+1);
    for(int i=1; i<=n; i++) cin>>a[i];
    vector<vector<int> > p(n+3,vector<int>(5));
    int sum=0;
    map<int,int> mp;
    mp[0]=n+1;
    p[n+1][0]=n+2;
    p[n+2][0]=n+2;
    for(int i=n; i>=1; i--){
        sum+=a[i];
        if(i==n){
            if(mp[sum]==0) p[i][0]=n+2;
            else p[i][0]=mp[sum];
        }else{
            if(mp[sum]==0) p[i][0]=p[i+1][0];
            else p[i][0]=min(mp[sum],p[i+1][0]);
        }
        mp[sum]=i;
    }
    for(int i=1; i<5; i++){
        for(int j=1; j<=n+2; j++){
            p[j][i]=j;
            for(int k=1; k<=10; k++){
                p[j][i]=p[p[j][i]][i-1];
            }
        }
    }
    vector<int> dp(5);
    dp[0]=1;
    for(int i=1; i<5; i++) dp[i]=dp[i-1]*10;
    int q; cin>>q;
    while(q--){
        int l,r; cin>>l>>r;
        int ans=0;
        for(int i=4; i>=0; i--){
            while(p[l][i]-1<=r){
                l=p[l][i];
                ans+=dp[i];
            }
        }
        cout<<ans<<endl;
    }
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	int T=1;
    // cin>>T;
    while(T--){
        solve();
        cout<<endl;
    }
	return 0;
}

• Subtask 1: 22 / 22
• Subtask 2: 39 / 39
• Subtask 3: 0 / 39