Submission #859685

#	Time	Username	Problem	Language	Result	Execution time	Memory
859685		HoriaHaivas	Sum Zero (RMI20_sumzero)	C++14	61 / 100	194 ms	18260 KiB

sumzero

/*
    "care a facut teste cu Lattice reduction attack e ciudat"
    - 2023 -
*/
#include<bits/stdc++.h>
#define debug(x) cerr << #x << " " << x << "\n"
#define debugs(x) cerr << #x << " " << x << " "
#pragma GCC optimize("Ofast")

using namespace std;

struct secv
{
    int l;
    int r;
    int nodeval;
};

struct cmp
{
    bool operator()(secv a, secv b)
    {
        return a.r>b.r;
    }
};

bool cmp2 (secv a, secv b)
{
    return a.l<b.l;
}

int v[200005];
long long sp[200005];
map<long long,int> last;
secv perechi[300005];
int up[11][200005];
int firstseq[200005];
int pw[11];

int main()
{
    ifstream fin("secvp.in");
    ofstream fout("secvp.out");
    ios_base::sync_with_stdio(false);
    cin.tie();
    cout.tie();
    int n,q,i,j,k,cnt,best,it,mi,a,b,r,pas,node,nxt;
    cin >> n;
    for (j=1; j<=n; j++)
    {
        cin >> v[j];
        sp[j]=sp[j-1]+v[j];
    }
    cnt=0;
    for (j=1; j<=n; j++)
    {
        if (sp[j]==0)
        {
            cnt++;
            perechi[cnt].l=last[sp[j]]+1;
            perechi[cnt].r=j;
        }
        else
        {
            if (last[sp[j]]!=0)
            {
                cnt++;
                perechi[cnt].l=last[sp[j]]+1;
                perechi[cnt].r=j;
            }
        }
        last[sp[j]]=j;
    }
    sort(perechi+1,perechi+1+cnt,cmp2);
    for (j=1; j<=cnt; j++)
        perechi[j].nodeval=j;
    perechi[0].l=n+1;
    perechi[0].r=n+1;
    perechi[0].nodeval=0;
    it=cnt;
    best=0;
    for (j=n; j>=1; j--)
    {
        if (perechi[it].l==j)
        {
            if (perechi[it].r<perechi[best].r)
            {
                best=it;
            }
            it--;
        }
        firstseq[j]=best;
        /*
        debugs(j);
        debug(firstseq[j]);
        */
    }
    for (j=1; j<=cnt; j++)
    {
        up[0][j]=firstseq[perechi[j].r+1];
    }
    pw[0]=1;
    for (j=1; j<=10; j++)
    {
        pw[j]=pw[j-1]*4;
    }
    for (j=1; j<=10; j++)
    {
        for (i=1; i<=cnt; i++)
        {
            up[j][i]=up[j-1][up[j-1][up[j-1][up[j-1][i]]]];
        }
    }
    cin >> q;
    for (j=1; j<=q; j++)
    {
        cin >> a >> b;
        r=0;
        pas=10;
        node=firstseq[a];
        while (pas>=0)
        {
            nxt=up[pas][up[pas][up[pas][node]]];
            if (perechi[nxt].nodeval!=0 && perechi[nxt].r<=b)
            {
                r+=pw[pas]*3;
                node=nxt;
            }
            else
            {
                nxt=up[pas][up[pas][node]];
                if (perechi[nxt].nodeval!=0 && perechi[nxt].r<=b)
                {
                    r+=pw[pas]*2;
                    node=nxt;
                }
                else
                {
                    nxt=up[pas][node];
                    if (perechi[nxt].nodeval!=0 && perechi[nxt].r<=b)
                    {
                        r+=pw[pas];
                        node=nxt;
                    }
                }
            }
            pas--;
        }
        if (perechi[node].nodeval!=0 && perechi[node].r<=b)
            cout << r+1 << "\n";
        else
            cout << 0 << "\n";
    }
    return 0;
}

Compilation message (stderr)

sumzero.cpp: In function 'int main()':
sumzero.cpp:47:17: warning: unused variable 'k' [-Wunused-variable]
   47 |     int n,q,i,j,k,cnt,best,it,mi,a,b,r,pas,node,nxt;
      |                 ^
sumzero.cpp:47:31: warning: unused variable 'mi' [-Wunused-variable]
   47 |     int n,q,i,j,k,cnt,best,it,mi,a,b,r,pas,node,nxt;
      |                               ^~

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