Cod sursa(job #1934204)

Utilizator jurjstyleJurj Andrei jurjstyle Data 21 martie 2017 11:40:17
Problema Stramosi Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <fstream>

using namespace std;

ifstream f("stramosi.in");
ofstream g("stramosi.out");

int n, m;
int dp[20][250002]; //dp[i][j] - al 2^i lea parinte al nodului j
int q, p;

int main()
{
    f >> n >> m;
    for (int i = 1; i <= n; ++i)
        f >> dp[0][i]; //citim tatal nodului respectiv
    for (int p = 1; (1 << p) <= n; ++p)
        for (int i = 1; i <= n; ++i)
            dp[p][i] = dp[p - 1][dp[p - 1][i]] ; //al 2^p lea stramos al lui i este al 2^(p - 1) stramos al 2^(p - 1) tata a lui i
    for (int i = 1; i <= m; ++i)
    {
        f >> q >> p;//citim nodul q si al p-lea stramos pe care il aflam
        for (int k = 0; (1 << k) <= p; ++k) //pt orice putere mai mica ca p
            if (p & (1 << k)) //daca al k-lea bit a lui p este 1
                q = dp[k][q]; //nodul nostru devine al 2^k parinte a lui (sa ne apropiem de al p-lea parinte
        g << q << "\n";
    }
    f.close();
    g.close();
    return 0;
}