HDU4565-2013年长沙邀请赛A题

转自：http://blog.csdn.net/crazy______/article/details/9021169

Sn=⌈(a+b√)n⌉%m,(a−1)2<b<a2

这个题目也是2008年Google Codejam Round 1A的C题。

做法其实非常简单，记 (a+b√)n 为 An ，配项

Cn=An+Bn=(a+b√)n+(a−b√)n

两项恰好共轭，所以 Cn 是整数。又根据限制条件

(a−1)2<b<a2⇒0<a−b√<1⇒0<(a−b√)n<1⇒Bn<1

也就是说 Cn=⌈An⌉

Sn=(Cn)%m

求 Cn 的方法是递推。 对 Cn 乘以 (a+b√)+(a−b√)

于是

Cn+1=2aCn−(a2−b)Cn−1

把这个递推式写成矩阵形式

[Cn+1Cn]=[2a1−(a2−b)0][CnCn−1]

于是就可以用矩阵快速幂来做了

[Cn+1Cn]=[2a1−(a2−b)0]n[C1C0]

一个需要注意的地方是快速幂的过程中由于取余可能出现负数，要化为正数。

代码：

#include<stdio.h>
__int64 matrix[2][2],a,b,n,m;
void getX(__int64 a[][2],__int64 b[][2])
{
    __int64 temp,matrix1[2][2];
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
        {
            temp=0;
            for(int k=0;k<2;k++)
                temp+=(a[i][k]*b[k][j]);
            matrix1[i][j]=temp%m;
        }
    for(int i=0;i<2;i++)
        for(int j=0;j<2;j++)
            a[i][j]=matrix1[i][j];
}
__int64 getAns()
{
    __int64 ans[2][2]={{1,0},{0,1}};
    matrix[0][0]=(2*a)%m;
    matrix[0][1]=(b-a*a)%m;
    matrix[1][0]=1;
    matrix[1][1]=0;
    while(n>0)
    {
        if(n%2)
            getX(ans,matrix);
        n/=2;
        getX(matrix,matrix);
    }
    __int64 aa=(ans[1][0]*2*a+ans[1][1]*2)%m;
    /*
    aa的结果有可能为负，但是实际情况不可能是负的。并不意味这aa算错，只是在c++中存在负的模
    因为(a+m)%m==(a%m+m%m)%m==a%m,所以算出aa<0是正常的，他有对应的正模(aa+m)%m=aa+m(因为
    -m<aa<0)
    */
    return aa>=0?aa:aa+m;
}
int main()
{
    while(scanf("%I64d%I64d%I64d%I64d",&a,&b,&n,&m)!=EOF)
    {
        printf("%I64d\n",getAns());
    }
    return 0;
}