Cod sursa(job #1817153)

Utilizator TibixbAndrei Tiberiu Tibixb Data 27 noiembrie 2016 13:57:01
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.84 kb
#include<fstream>
#define NMAX 250005
#define LMAX 20
using namespace std;
int n, m, x, p, q, d[LMAX][NMAX];
ifstream cin("stramosi.in");
ofstream cout("stramosi.out");
int main()
{
    cin >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        cin >> x;
        d[0][i] = x;
    }
    for(int i = 1; (1 << i) <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            // ATENTIE unele noduri nu pot avea 2 ^ i stramosi
            // merge caci d[i][0] e 0
            int p = d[i - 1][j];
            d[i][j] = d[i - 1][p];
        }
    }
    for(int i = 1; i <= m; i++)
    {
        cin >> q >> p;
        for(int i = 0; (1 << i) <= p; i++)
        {
            if((p & (1 << i)) > 0)
            {
                q = d[i][q];
            }
        }
        cout << q << "\n";
    }
    return 0;
}