Cod sursa(job #3134248)

Utilizator tudor.pistolPistol Tudor tudor.pistol Data 28 mai 2023 19:57:35
Problema Stramosi Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <fstream>
#include <cmath>
using namespace std;

ifstream fin("stramosi.in");
ofstream fout("stramosi.out");
int n, m, p, q, i, j;

int main()
{
    fin >> n >> m;
    int** stramosi = new int*[(int)log2(n)+1];
    stramosi[0] = new int[n+1];
    
    for(int i = 1; i <= n; i++)
    {
        stramosi[i] = new int[n+1];
        fin >> stramosi[1][i];
    }

    for (int j = 2; (1 << (j - 1)) <= n; j++)
           for (int i = 1; i <= n; i++)
               stramosi[j][i] = stramosi[j - 1][stramosi[j - 1][i]];

    for(int i = 0; i < m; i++)
    {
        fin >> q >> p;
        for (j = 1; j <= (int)log2(n); j++)
                if (p & (1 << (j - 1)))
                    q = stramosi[j][q];
        fout << q << endl;
    }
    
    fin.close();
    fout.close();
    return 0;
}