Cod sursa(job #1303598)

Utilizator horiainfoTurcuman Horia horiainfo Data 28 decembrie 2014 10:01:07
Problema Stramosi Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.98 kb
#include <fstream>
#include <cstdio>
#include <vector>
using namespace std;
ofstream fout("stramosi.out");
int p,q,m,n,s[250001][20];
int a[250001],co[250001];
void merge(int poz)
{
    int p1=s[poz][0];
    for(int i=0;s[p1][i]!=0;i++)
        s[poz][i+1]=s[p1][i],p1=s[p1][i];
    a[poz]=0;
}
void DFS(int poz)
{
    int k=0;
    co[k]=poz;
    while(a[co[k]]!=0)
        co[k+1]=s[co[k]][0]=a[co[k]],k++;
    for(int i=k-1;i>=0;i--)
        merge(co[i]);
}
int main()
{
    long long aux;
    freopen("stramosi.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        if(a[i]!=0)
            DFS(i);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&q,&p);
        while(p)
        {
            aux=1;
            while(1<<aux<p)
                aux++;
            aux--;
            q=s[q][aux];
            p=p-(1<<aux);
        }
        fout<<q<<'\n';
    }
    return 0;
}