BestCoder Round #8 1002 Reading comprehension[矩阵幂求解递推值]

最新推荐文章于 2019-01-19 17:08:54 发布

creat2012

最新推荐文章于 2019-01-19 17:08:54 发布

阅读量883

点赞数

本文链接：https://blog.csdn.net/creat2012/article/details/39137679

版权

题目链接：点http://acm.hdu.edu.cn/listproblem.php?vol=40

昨天的BestCoder，题目的意思是

ans=0;
for(i=1;i<=n;i++)
{
    if(i&1)ans=(ans*2+1)%m;
    else ans=ans*2%m;
}

需要优化这个程序。1<=n, m <= 1000000000；

一开始的想法，认为是有规律，也还找了点规律。。但是，方向错了。。后来看题解，明白了，用矩阵的快速幂求解递推值。很快的。。

有意学习的戳：http://blog.csdn.net/u011394362/article/details/39136903

对于这个问题，我们应该可以把偶数项化成一个通项。。

对于偶数项：fi = fi-1 * 4 +２；

所以可以求出A矩阵使得，[fn, c] = [fn-1, c] * A；得到了一个A矩阵。

A= {

4, 0,

2, 1,

}

Code：

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define LL long long

const int N = 5;
LL n, m, mod;
struct Matrix
{
    int n, m;
    LL a[N][N];
    Matrix(){
        memset(a, 0, sizeof(a));
    }
}A, B;

Matrix operator * (Matrix a, Matrix b){
    Matrix ans;
    ans.n = a.n; ans.m = b.m;
    for(int i = 1; i <= a.n; i ++){
        for(int j = 1; j <= b.m; j ++){
            for(int k = 1; k <= b.n; k ++)
            ans.a[i][j] = (ans.a[i][j] + a.a[i][k] * b.a[k][j]) % mod;
        }
    }
    return ans;
}

LL Quick_Power(LL n)
{
    Matrix ans = B, k = A;
    while(n){
        if(n & 1) {
            ans = ans * k;
        }
        k = k * k;
        n = n >> 1;
    }
    return ans.a[1][1] % mod;
}
int main()
{
    while(cin >> n >> m){
        if(n == 1){
            cout << 1 % m<< endl;
            continue;
        }
        mod = m;
        A.n = 2; A.m = 2;
        A.a[1][1] = 4; A.a[2][1] = 2; A.a[2][2] = 1;
        B.n = 1; B.m = 2;
        B.a[1][1] = 0; B.a[1][2] = 1;
        LL ans = Quick_Power(n / 2);
//        cout << ans << endl;
        if(n % 2) ans = ans * 2 + 1;
        cout<< ans % mod<< endl;
    }
    return 0;
}