其他分享
首页 > 其他分享> > 做题集——(矩阵快速幂+推公式)Covering

做题集——(矩阵快速幂+推公式)Covering

作者:互联网

题目:

http://acm.hdu.edu.cn/showproblem.php?pid=6185

题面:

 

 Sample Input

1
2

Output

1
5

题意:

  有地毯为2 * 1。有操场4 * n,地毯尽可以横放及竖放,求地毯不重叠的铺满整个地板有多少种方法。

思路:

  多少种方法其实可以联想到dp,每一个n也刚好可以对应一种状态,但是考虑n的数据范围有点大,dp时间复杂度过高,并不靠谱。实际上通过n的数据范围可以考虑这道题应该是求递推公式的类型。并且稍加思考也确实不难发现规模n与规模n - 1与规模n - 2……确实有着某种关系在里边。因此得出该题应该先求出递推式,然后通过矩阵快速幂的方法得到所需的解。

  经过漫长的递推式求解过程,可以得到如下关系式f(x) = 5 * sum(x - 2) + f(x - 3),于是将代码敲出。

Code:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<iostream>
#define ll long long
using namespace std;
const ll MOD = 1e9 + 7;

struct Mat{
    ll m[4][4];
}unit;
void Unit(){
    for(ll i = 0;i < 4;i++)
        unit.m[i][i] = 1;
    return;
}
Mat Mult(Mat a,Mat b){
    Mat ret;
    ll x;
    for(ll i = 0;i < 4;i++){
        for(ll j = 0;j < 4;j++){
            x = 0;
            for(ll k = 0;k < 4;k++)
                x = (x + a.m[i][k] * b.m[k][j] % MOD) % MOD;
            ret.m[i][j] = x;
        }
    }
    return ret;
}
Mat QPow(Mat a,ll n){
    Mat ret = unit;
    while(n){
        if(n & 1) ret = Mult(ret, a);
        a = Mult(a, a);
        n >>= 1;
    }
    return ret;
}
int main(){
    ll n;
    Unit();
    while(~scanf("%lld", &n)) {
        if(n == 1) printf("1\n");
        else if(n == 2) printf("5\n");
        else if(n == 3) printf("11\n");
        else{
            Mat b;
            b.m[0][0]=0,b.m[0][1]=1,b.m[0][2]=1,b.m[0][3]=0;
            b.m[1][0]=5,b.m[1][1]=1,b.m[1][2]=0,b.m[1][3]=0;
            b.m[2][0]=0,b.m[2][1]=0,b.m[2][2]=0,b.m[2][3]=1;
            b.m[3][0]=1,b.m[3][1]=0,b.m[3][2]=0,b.m[3][3]=0;
            b = QPow(b,n-3);
            printf("%lld\n",(11 * b.m[0][0] + 7 * b.m[1][0] + 5 * b.m[2][0] + b.m[3][0]) % MOD);
            //11 7 5 1 表示i - 1, sum(i - 2), i - 2, i - 3;
        }
    }
    return 0;
}

 

标签:return,Mat,Covering,++,ll,矩阵,ret,做题,include
来源: https://www.cnblogs.com/TanJI-life/p/14742966.html