HDU 3292 pell方程

最新推荐文章于 2020-02-04 15:44:02 发布

韩小侠

最新推荐文章于 2020-02-04 15:44:02 发布

阅读量1k

点赞数

分类专栏：数论

本文链接：https://blog.csdn.net/prime7/article/details/9625715

版权

数论专栏收录该内容

69 篇文章 1 订阅

订阅专栏

题意：求x^2+n*y^2=1 按x排序第k大的解。

题目就是求佩尔方程，用矩阵连乘的方法，当n为完全平方数的时候无解。

如果我们求出Pell方程的最小正整数解后，就可以根据递推式求出所有的解。

则根据上式我们可以构造矩阵，然后就可以快速幂了。

这样就可以求出第k大的解。

HDU3292题就要用到上面的矩阵方法求第k大的解。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define M 8191
const int MAX=2;
typedef struct
{
    long long m[MAX][MAX];
} Matrix;
Matrix I= {1,0,
           0,1
          };
Matrix P;
Matrix matrixmul(Matrix a,Matrix b) //矩阵乘法
{
    int i,j,k;
    Matrix c;
    for (i = 0 ; i < MAX; i++)
        for (j = 0; j < MAX; j++)
        {
            c.m[i][j] = 0;
            for (k=0; k<MAX; k++)
                c.m[i][j]+=((a.m[i][k]%(M))*(b.m[k][j]%(M)))%(M);
            c.m[i][j] %=M;
        }
    return c;
}
Matrix quickpow(long long n)
{
    Matrix m = P, b = I;
    while (n >= 1)
    {
        if (n & 1)
            b = matrixmul(b,m);
        n = n >> 1;
        m = matrixmul(m,m);
    }
    return b;
}
int main()
{
    long long n,k,x,y;
    while(~scanf("%I64d%I64d",&n,&k))
    {
        if(n==4||n==9||n==16||n==25)
        {
            puts("No answers can meet such conditions");
            continue;
        }
        for(y=1;; y++)
        {
            x=(long long)sqrt(double(y*y*n+1));
            if(x*x==y*y*n+1)
                break;
        }
        P.m[0][0]=P.m[1][1]=x%M;
        P.m[0][1]=(n*y)%M,P.m[1][0]=y%M;
        Matrix ans=quickpow(k-1);
        printf("%I64d\n",((ans.m[0][0]*x)%M+(ans.m[0][1]*y)%M)%M);
    }
    return 0;
}