Cod sursa(job #2449629)

Utilizator eutu33eu tu el ea eutu33 Data 20 august 2019 12:02:10
Problema Stramosi Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.45 kb
/// https://www.youtube.com/watch?v=dQw4w9WgXcQ ;)
#include <iostream>
#include <fstream>
#include <bitset>
#include <math.h>
#include <vector>

#define NMAX 250005
#define LOGN 20
using namespace std;

int mat [NMAX][LOGN];
int currentRow;
vector<int> vec;

void MakeRow(int newRow, int index);
int GetResult(int row, int index);

int main()
{
	ifstream fin("stramosi.in");
	ofstream fout("stramosi.out");
	int n, m;
	fin >> n >> m;

	for (int i = 1; i <= n; i++)
	{
	    int a;
	    currentRow = i;
		fin >> a;
        MakeRow(a,0);
	}

	for (int i = 1; i <= m; i++)
    {
        int q, p;
        fin >> q >> p;
        bitset<20> bin(p);
        vec.clear();
        for(int j = 0; j < 20; j++)
        {
            if(bin[j] == 1)
            {
                vec.push_back(j);
                //cout<<j<<" ";
            }
        }
        //cout<<"\n";
        fout << GetResult(q, 0) << "\n" ;
    }
    /*
	for (int i = 0; i <= n; i++)
    {
        for (int j = 0; j < LOGN; j++)
        {
            cout << mat[i][j] << " ";
        }
        cout << endl;
    }
    */

}

int GetResult(int row, int index)
{
    int newRow = mat[row][vec[index]];
    if(index == vec.size() - 1)
        return newRow;
    return GetResult(newRow,index+1);
}

void MakeRow(int newRow, int index)
{
    mat[currentRow][index] = newRow;
    if(newRow!=0)
        MakeRow(mat[newRow][index], index+1);
}