Cod sursa(job #1713596)

Utilizator akaprosAna Kapros akapros Data 6 iunie 2016 00:11:07
Problema Robotei Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.11 kb
#include <bits/stdc++.h>
#define maxM 1000002
#define maxN 1002
#define ll long long
using namespace std;
int n, m, X, Y, modx, mody, offx, offy, len;
ll ans[maxM], ox[maxN], oy[maxN];
int vis[maxN][maxN];
bool a[maxN][maxN];
void read()
{
    freopen("robotei.in", "r", stdin);
    scanf("%d %d %d %d %d %d %d %d", &n, &m, &X, &Y, &modx, &mody, &offx, &offy);
    vis[X][Y] = 1;
}
void solve()
{
    int i, j;
    int x = X, y = Y;
    while (!a[x][y])
    {
        ++ len;
        a[x][y] = 1LL;
        x = (1LL * x * x + 1LL * offx) % modx;
        y = (1LL * y * y + 1LL * offy) % mody;
    }
    if (x != X || y != Y)
    {
        ans[0] = 1LL * n * n - 1;
        ans[1] = 1;
        return ;
    }
    for (i = 0; i < n; ++ i)
    {
        ++ ox[(1LL * i * i + 1LL * offx) % modx];
        ++ oy[(1LL * i * i + 1LL * offy) % mody];
    }

    for (i = 0; i < modx; ++ i)
        for (j = 0; j < mody; ++ j)
        {
            x = i;
            y = j;
            int l = 0;
            while (!vis[x][y])
            {
                ++ l;
                vis[x][y] = -1;
                x = (1LL * x * x + 1LL * offx) % modx;
                y = (1LL * y * y + 1LL * offy) % mody;
            }
            if (vis[x][y] != -1)
            {
                l += vis[x][y];
                if (l <= m)
                    ans[1 + (m - l) / len] += 1LL * ox[i] * oy[j];

                x = i;
                y = j;
                while (!vis[x][y] || vis[x][y] == -1)
                {
                    vis[x][y] = l --;
                    x = (1LL * x * x + 1LL * offx) % modx;
                    y = (1LL * y * y + 1LL * offy) % mody;
                }
            }
        }
    X = (1LL * X * X + 1LL * offx) % modx;
    Y = (1LL * Y * Y + 1LL * offy) % mody;
    -- ans[vis[X][Y]];
    ++ ans[vis[X][Y] + 1];
}
void write()
{
    int i;
    freopen("robotei.out", "w", stdout);

    for (i = 0; i <= m; ++ i)
        if (ans[i] != 0)
            printf("%d %lld\n", i, ans[i]);
}
int main()
{
    read();
    solve();
    write();
    return 0;
}