Cod sursa(job #365259)

Utilizator popoiu.georgeGeorge Popoiu popoiu.george Data 18 noiembrie 2009 11:53:15
Problema Stramosi Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.53 kb
#include<stdio.h>
//#include<fstream>
using namespace std;

//fstream f("stramosi.in",ios::in),g("stramosi.out",ios::out);

int n,m;
//int v[251000],mat[300001][3];
int v[251000],mat[300001][3];
int st[4][1000000],k=0;

void cauta(int q,int p,int i)
{
/*if(i==p)
	{
	//printf("%d\n",q);
	g<<q<<"\n";
	return;
	}
else
	{

	if(q!=0)cauta(v[q],p,i+1);
	else {g<<"0\n";return;}
	}*/
k=1;
st[1][k]=q;
st[2][k]=p;
st[3][k]=i;
while(k>0)
    {
    if(st[3][k]==st[2][k])
        {
        //g<<st[1][k]<<"\n";
        printf("%d\n",st[1][k]);
        return;
        }
    else
        {
        if(st[1][k]!=0){k++;st[1][k]=v[st[1][k-1]];st[2][k]=st[2][k-1];st[3][k]=1+st[3][k-1];}
        else
            {
            //g<<"0\n";
            printf("0\n");
            return;
            }
        }
    }
}

int main()
{
int i;
freopen("stramosi.in","r",stdin);
freopen("stramosi.out","w",stdout);
scanf("%d%d",&n,&m);
//f>>n>>m;
for(i=1;i<=n;i++)scanf("%d",&v[i]);//f>>v[i];
for(i=1;i<=m;i++)
	{
	scanf("%d",&mat[i][1]);
	scanf("%d",&mat[i][2]);
	//f>>mat[i][1]>>mat[i][2];
	//f>>mat[1][i]>>mat[2][i];
	/*if(mat[i][2]==1)printf("%lld\n",v[mat[i][1]]);
	else cauta(v[v[mat[i][1]]],mat[i][2],2);*/
	}
for(i=1;i<=m;i++)
	{
	if(mat[i][2]==1)printf("%d\n",v[mat[i][1]]);//g<<v[mat[i][1]]<<"\n";
	else cauta(v[v[mat[i][1]]],mat[i][2],2);
	//if(mat[2][i]==1)printf("%d\n",v[mat[1][i]]);//g<<v[mat[1][i]]<<"\n";
	//else cauta(v[v[mat[1][i]]],mat[2][i],2);
	}
//f.close();
//g.close();
return 0;
}