Submission #646889

#	Time	Username	Problem	Language	Result	Execution time	Memory
646889		DobromirAngelov	Sum Zero (RMI20_sumzero)	C++14	61 / 100	290 ms	21876 KiB

sumzero

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;

const int MAXN=4e5+5;
const int INF=1e9+5;

int n,q;
//int a[MAXN];
long long a_pref_prev[MAXN];
//int nxt[MAXN];
//int previous[MAXN];
int prv[MAXN];
int nxt_l[MAXN], r[MAXN];
int ans[MAXN];
map<long long,int> m;

int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);

cin>>n;
for(int i=1;i<=n;i++) cin>>a_pref_prev[i]; // pref[i]=a[i]

for(int i=1;i<=n;i++)
{
    a_pref_prev[i]=a_pref_prev[i-1]+a_pref_prev[i];//cout<<pref[i]<<" ";
}//cout<<endl;
for(int i=n;i>=0;i--)
{
    if(!m[a_pref_prev[i]]) nxt_l[i]=INF; ///
    else nxt_l[i]=m[a_pref_prev[i]];
    m[a_pref_prev[i]]=i;
    //cout<<nxt[i]<<" ";
}//cout<<endl;
int ptr=n;
///memset(prv,-1,sizeof(prv));
for(int i=n;i>=0;i--)
{
    while(ptr>=nxt_l[i])
    {
        a_pref_prev[ptr]=i+1;
        ptr--;
    }
}
//for(;ptr>=0;ptr--) prv[0][ptr]=-1;
//for(int i=1;i<=n;i++) cout<<prv[0][i]<<" "; cout<<endl;
cin>>q;
for(int i=1;i<=q;i++)
{
    cin>>nxt_l[i]>>r[i];
}

for(int lg=18;lg>=0;lg--)
{
    for(int i=1;i<=n;i++) prv[i]=a_pref_prev[i];

    for(int j=1;j<=lg;j++)
    {
        for(int i=n;i>0;i--)
        {
            if(prv[i]>1) prv[i]=prv[prv[i]-1];
            else prv[i]=0;
        }
    }

    for(int i=1;i<=q;i++)
    {
        if(nxt_l[i]<=r[i] && prv[r[i]]>=nxt_l[i])
        {
            ans[i]+=(1<<lg);
            r[i]=prv[r[i]]-1;
        }
    }
}

for(int i=1;i<=q;i++) cout<<ans[i]<<endl;

return 0;
}

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