【NOI2014】随机数生成器

代码

#include<cstdio>
using namespace std;
typedef long long LL;

const int N = 5000;
int n , m , q , tot;
LL a , b , c , d;
int x[N * N + 5] , T[N * N + 5] , l[N + 5] , r[N + 5] , ans[N + N + 5];

inline void swap(int &x , int &y)
{
    register int t = x;
    x = y , y = t;
}

int main()
{
    scanf("%d%lld%lld%lld%lld%d%d%d" , &x[0] , &a , &b , &c , &d , &n , &m , &q);
    for(register int i = 1; i <= n * m; i++) 
        x[i] = (a * x[i - 1] * x[i - 1] + b * x[i - 1] + c) % d , T[i] = i;
    register int t , u , v;
    for(register int i = 1; i <= n * m; i++) swap(T[i] , T[x[i] % i + 1]);
    for(register int i = 1; i <= q; i++)
    {
        scanf("%d%d" , &u , &v);
        swap(T[u] , T[v]);
    }
    for(register int i = 1; i <= n * m; i++) x[T[i]] = i;
    for(register int i = 1; i <= n; i++) r[i] = 2e9;
    for(register int i = 1; i <= n * m; i++)
    {
        u = x[i] % m ? x[i] / m + 1 : x[i] / m;
        v = (x[i] - 1) % m + 1;
        if (v >= l[u] && v <= r[u])
        {
            ans[++tot] = i;
            if (tot == n + m - 1) break;
            for(register int j = u - 1; j >= 1; j--)
            {
                if (v < r[j]) r[j] = v;
                else break;
            }
            for(register int j = u + 1; j <= n; j++)
            {
                if (v > l[j]) l[j] = v;
                else break;
            }
        }
    }
    for(register int i = 1; i <= tot; i++) printf("%d " , ans[i]);
}