Cod sursa(job #2332665)

Utilizator DavidLDavid Lauran DavidL Data 30 ianuarie 2019 23:00:37
Problema Atac Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.55 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fi("atac.in");
ofstream fo("atac.out");

const int NMAX = 35000;
const int LOG = 20;

vector < pair<int, int> > G[NMAX];
int h[NMAX], p[NMAX];
int sus[LOG][NMAX];

void dfs(int nod)
{
    for (auto v: G[nod])
        if (v != p[nod])
        {
            p[v] = nod;
            h[v] = h[nod] + 1;
            dfs(v);
        }
}

void precalcLca()
{
    for (int i = 1; i <= n; i++)
        sus[0][i] = p[i];
    sus[0][1] = -1;

    for (int i = 1; i < LOG; i++)
        for (int j = 1; j <= n; j++)
            sus[i][j] = sus[i - 1][sus[i - 1][j]];
}

int lca(int a, int b)
{
    if (h[a] < h[b])
        swap(a, b);
    for (int i = LOG - 1; i >= 0; i--)
        if (h[a] - (1 << i) >= h[b])
            a = sus[i][a];

    for (int i = LOG - 1; i >= 0; i--)
        if (h[a] - (1 << i) >= 1 && sus[i][a] != sus[i][b])
            a = sus[i][a], b = sus[i][b];
    return p[a];
}

int main()
{
    fi >> n >> m >> p;
    for (int i = 2;  i <= n; i++)
    {
        int v, c;
        fi >> v >> c;
        G[i].push_back({v, c});
        G[v].push_back({i, c});
    }

    h[1] = 1; dfs();
    precalcLca();
    precalcMinim();

    fi >> x >> y >> a >> b >> c >> d;
    for (int i = 1; i <= m; i++)
    {
        int comun = lca(x, y);
        int cost = min(minim(x, comun), minim(y, comun));
        if (i >= m - p + 1)
            fo << cost << "\n";
        x = (x * a + y * b) % n + 1;
        y = (y * c + cost * d) % n + 1;
    }

    return 0;
}