[CODEVS 1281] Xn数列

描述

给你6个数，m, a, c, x0, n, g
Xn+1 = ( aXn + c ) mod m，求Xn
http://codevs.cn/problem/1281/

分析

比较裸的矩阵乘法题, 好久没做了, 写写思路

假设矩阵 A = { {a1, a2}, {a3, a4} }, B = { {b1, b2}, {b3, b4} }.
根据矩阵乘法的计算方法, 有 :
A×B = { {a1b1+a2b2, a1b2+a2b4}, {a3b1+a4b3, a3b2+a4b4} }.
那么我们想得到的是 a*Xn + c = Xn+1
假设 X 存在a1里, 那么应有 Xn * b1 + a2 * b2 = Xn+1
那么就可以得到 a2 = 1, b1 = a, b2 = c.
(为什么不能 a2 = c, b2 = 1 ? 因为乘完一次 a2 就变了, 下个加的就不是 c 了).

这样矩阵就建好了, 接下来就是按部就班的矩阵乘法了.
不过, 这里数据太大, 不使用特殊技巧用 unsigned long long 都会溢出. 高精度 ? 其实不用, 因为最后都要模一下.

特殊技巧

用类似快速幂的方法写一个快速加, 加一下模一下, 用来避免直接乘的溢出.

代码

20ms 256kB
额… 难道 mod 和 sum 都是保留字 ? 为啥都有高亮 ?

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

LL mod, a, c, x0, n, g;

struct Matrix {
    LL m[2][2];
} base, X0;

// return a * b
LL quickadd(LL a, LL b) {
    LL ans = 0;
    a %= mod; b %= mod;
    while(b > 0) {  
        if(b & 1) ans = (ans + a) % mod;  
        a = (a + a) % mod;  
        b >>= 1;
    }
    return ans;
}

Matrix mul(Matrix a, Matrix b) {
    Matrix ans;
    for(int i = 0; i < 2; i++)
        for(int j = 0; j < 2; j++) {
            LL sum = 0;
            for(int k = 0; k < 2; k++)
                sum = (sum + quickadd(a.m[i][k], b.m[k][j])) % mod;
            ans.m[i][j] = sum;
        }
    return ans;
}

Matrix pow(Matrix a, LL n) {
    Matrix p = {{1, 0, 0, 1}};
    while(n > 0) {
        if(n & 1) p = mul(p, a);
        a = mul(a, a);
        n /= 2;
    }
    return p;
}

int main() {
    scanf("%lld%lld%lld%lld%lld%lld", &mod, &a, &c, &x0, &n, &g);
    base = (Matrix) {{a, 0, 1, 1}};
    X0 = (Matrix){{x0, c, 0, 0}};
    Matrix ans = mul(X0, pow(base, n));
    printf("%lld\n", ans.m[0][0] % g);
    return 0;
}