hdu 5667 Sequence【费马小定理+矩阵快速幂】

最新推荐文章于 2020-03-08 19:50:52 发布

lhfl911

最新推荐文章于 2020-03-08 19:50:52 发布

阅读量658

点赞数

分类专栏：数论文章标签：费马小定理矩阵快速幂

本文链接：https://blog.csdn.net/lhfl911/article/details/51173389

版权

数论专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题意：

 Lcomyn 是个很厉害的选手,除了喜欢写17kb+的代码题,偶尔还会写数学题.他找到了一个数列:
fn=⎧⎩⎨⎪⎪1,ab,abfcn−1fn−2,n=1n=2otherwise

他给了你几个数:n,a,b,c，你需要告诉他 $f_{n}$ 模p后的数值。

输入描述

第一行一个数T,为测试数据组数.

每组数据一行,一行五个正整数,按顺序为n,a,b,c,p.

$1≤T≤10，1≤n≤10^{18}，1≤a,b,c≤10^9$ ,p是质数且 $p≤10^9+7.$

输出描述

对每组数据输出一行一个数,输出 $f_{n}$ 对p取模后的数值.

思路：

设 $g_{n}$ = $log_{a}f_{n}$ ， $g_{n}$ 满足 $g_{n}=b+c*g_{n-1}+g_{n-2}$ 。
可以转化为矩阵相乘：

⎧ ⎩ ⎨ ⎪ ⎪ g n g n - 1 b ⎫ ⎭ ⎬ ⎪ ⎪ = ⎧ ⎩ ⎨ ⎪ ⎪ c 10 100101 ⎫ ⎭ ⎬ ⎪ ⎪ * ⎧ ⎩ ⎨ ⎪ ⎪ g n - 1 g n - 2 b ⎫ ⎭ ⎬ ⎪ ⎪

$\left\{ \begin{matrix}g_{n}\\g_{n-1}\\b\end{matrix}\right\}=\left\{ \begin{matrix}c&1&1\\1&0&0\\0&0&1\end{matrix}\right\}*\left\{ \begin{matrix}g_{n-1}\\g_{n-2}\\b\end{matrix}\right\}$
由此得：

⎧ ⎩ ⎨ ⎪ ⎪ g n g n - 1 b ⎫ ⎭ ⎬ ⎪ ⎪ = ⎧ ⎩ ⎨ ⎪ ⎪ c 10 100101 ⎫ ⎭ ⎬ ⎪ ⎪ n - 2 * ⎧ ⎩ ⎨ ⎪ ⎪ b 0 b ⎫ ⎭ ⎬ ⎪ ⎪

$\left\{ \begin{matrix}g_{n}\\g_{n-1}\\b\end{matrix}\right\}=\left\{ \begin{matrix}c&1&1\\1&0&0\\0&0&1\end{matrix}\right\}^{n-2}*\left\{ \begin{matrix}b\\0\\b\end{matrix}\right\}$

用矩阵快速幂和费马小定理求出 $g_n$
快速幂求出 $a^{g_n}$ 。

费马小定理：
假如p是质数，且gcd(a,p)=1，那么 $a^{p-1}$ ≡ 1(mod p)。

注意a%p = 0的情况。（bc时就错在这了）

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

typedef long long LL;

LL p;
void mul(LL a[3][3], LL b[3][3])
{
    LL c[3][3];
    for(int i = 0; i < 3; i++)
        for(int j = 0; j < 3; j++)
        {
            c[i][j] = 0;
            for(int k = 0; k < 3; k++)
            {
                c[i][j] += a[i][k] * b[k][j];
                c[i][j] %= p-1;                         //费马小定理 p是素数，则a^(p-1)%p = 1;
            }
        }
    for(int i = 0; i < 3; i++)
        for(int j = 0; j < 3; j++)
            a[i][j] = c[i][j];
}
void matpow(LL a[3][3], LL n)
{
    LL ans[3][3];
    memset(ans, 0, sizeof(ans));
    for(int i = 0; i < 3; i++)
        ans[i][i] = 1;
    while(n)
    {
        if(n&1)
            mul(ans, a);
        mul(a, a);
        n >>= 1;
    }
    for(int i = 0; i < 3; i++)
        for(int j = 0; j < 3; j++)
            a[i][j] = ans[i][j];
}
LL fastpow(LL a, LL b)
{
    LL k = a;
    LL ans = 1;
    while(b)
    {
        if(b&1) 
        {
            ans *= k;
            ans %= p;
        }
        k *= k;
        k %= p;
        b >>= 1;
    }
    return ans;
}
int main(int argc, char const *argv[])
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        LL n, a, b, c;
        cin >> n >> a >> b >> c >> p;
        if(n == 1)
            cout << "1" << endl;
        else if(a % p == 0)
            cout << "0" << endl;            //!!!!
        else
        {
            LL mat[3][3] = {c, 1, 1, 
                            1, 0, 0,
                            0, 0, 1};
            matpow(mat, n-2);
            LL ans[3][3] = {b, 0, 0,
                                0, 0, 0,
                                b, 0, 0};
            mul(mat, ans);
            cout << fastpow(a, mat[0][0]) << endl;
        }


    }
    return 0;
}