HDU4565 So Easy!（数学+矩阵快速幂）

最新推荐文章于 2019-05-16 16:17:13 发布

Tailless

最新推荐文章于 2019-05-16 16:17:13 发布

阅读量386

点赞数

分类专栏： Algorithm 文章标签： HDU 代码

本文链接：https://blog.csdn.net/wust_xhj/article/details/47377319

版权

Algorithm 专栏收录该内容

33 篇文章 0 订阅

订阅专栏

So Easy!

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Problem Description

　　A sequence S _n is defined as:

Where a, b, n, m are positive integers.┌x┐is the ceil of x. For example, ┌3.14┐=4. You are to calculate S _n.
　　You, a top coder, say: So easy!

Input

　　There are several test cases, each test case in one line contains four positive integers: a, b, n, m. Where 0< a, m < 2 ¹⁵, (a-1) ²< b < a ², 0 < b, n < 2 ³¹.The input will finish with the end of file.

Output

　　For each the case, output an integer S _n.

Sample Input

Sample Output

这题有个神奇的条件：(a-1)^2< b < a^2 => a-1 < sqrt(b) < a
这就是说（a + sqrt(b)）^n的共轭式（a - sqrt(b)）^n < 1
设Cn = An + Bn = （a + sqrt(b)）^n + （a - sqrt(b)）^n = [An] = [（a + sqrt(b)）^n]
也就是说Sn = Cn
Cn是可以递推的：Cn两边同乘（a + sqrt(b)） + (a - sqrt(b)) :
Cn[ （a + sqrt(b)） + (a - sqrt(b)) ] = [（a + sqrt(b)）^n + （a - sqrt(b)）^n] * [ （a + sqrt(b)） + (a - sqrt(b)) ]
= Cn+1 + (a^2 - b)Cn-1

Cn+1 = 2aCn + (b - a^2)Cn-1 ←喜闻乐见的矩阵了。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#define LL long long
using namespace std;
LL a,b,n,mod;
struct Mvt
{
    LL v[3][3];
    Mvt operator * (Mvt &rhs)
    {
        Mvt ret;
        for(int i=0;i<2;i++)
        {
            for(int j=0;j<2;j++)
            {
                ret.v[i][j]=0;
                for(int k=0;k<2;k++)
                {
                    ret.v[i][j]+=v[i][k]*rhs.v[k][j]%mod;
                }
                ret.v[i][j]%=mod;
            }
        }
        return ret;
    }
};
Mvt mul(Mvt x,LL p)
{
    Mvt ans;
    ans.v[0][0]=ans.v[1][1]=1;
    ans.v[1][0]=ans.v[0][1]=0;
    while(p>0)
    {
        if(p&1)
            ans=ans*x;
        x=x*x;
        p>>=1;
    }
    return ans;
}

int main()
{
    while(scanf("%lld%lld%lld%lld",&a,&b,&n,&mod)!=EOF)
    {
        if(n==1)
        {
            printf("%lld\n",a*2%mod);
            continue;
        }
        Mvt tmp;
        tmp.v[0][0]=2*a%mod;
        tmp.v[0][1]=(b-a*a)%mod;
        tmp.v[1][0] = 1;
        tmp.v[1][1] = 0;
        tmp=mul(tmp,n);
        LL ans = ((2 * a * tmp.v[1][0] + 2 * tmp.v[1][1])%mod+mod)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}