Cod sursa(job #3181904)

Utilizator Cazacu2006RazvanRazvan Cazacu Cazacu2006Razvan Data 8 decembrie 2023 11:11:46
Problema Stramosi Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <fstream>

using namespace std;
ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int n,m,tata[300001],st[21][300001],fr[300001],l[300001],p[300001];
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int x;
        fin>>x;
        tata[i]=x;
        fr[x]=1;
    }
    p[1]=1;
    for(int i=2;i<=n;i++)
    {
        p[i]=p[i/2]+1;
    }
    for(int i=1;i<=n;i++)
    {
        if(fr[i]==0)
        {

            int nr=0;
            int x=i;
            while(x!=0)
            {
                nr++;
                l[nr]=x;
                for(int j=1;j<nr;j=j*2)
                {
                    int put=p[j];
                    st[put][l[nr-j]]=x;
                }
                x=tata[x];
            }

        }
    }
    for(int i=1;i<=m;i++)
    {
        int x,y;
        fin>>x>>y;
        int put=p[y];
        int l=st[put][x];

        int nr=(1<<(put-1));

        while(nr<y)
        {
            l=tata[l];
            nr++;
        }
        fout<<l<<"\n";
    }



    return 0;
}