hdoj2604 Queuing(矩阵快速幂)

此题如果直接利用递推关系,处理不好会超内存的。

首先找出递推关系式,先给出递推关系式:( L )=( L - 1 ) + ( L - 3 ) + ( L - 4 );可以先尝试推导一下,推不出来再看下面的解释。

PS.以前做过一个类似的递推关系的题。

考虑当L=n时的情况,有两种情况:

①.如果最后一个字符为m :此时,只要前面长度为n-1的串符合要求,则当前长度为n的串必然符合要求。

②.如果最后一个字符为f:此时,无法确定,因为可能存在不符合要求的串,继续分情况讨论

          (1).最后倒数二个字符为f,仍然可能存在不符合要求的串,继续分情况讨论

                  1.倒数第三个字符为f,因为存在fff,所以该种情况必然不符合要求,舍去

                  2.倒数第三个字符为m,仍然有可能不符合要求,再分

                       a.最后第四个字符为f,存在fmf,所以该种情况必然不符合要求,舍去

                       b.最后第四个字符为m,只要前面长度为n-4的串符合要求,则当前长度为n的串必然也符合要求

           (2).最后第二个字符为m,存在可能不符合要求的情况,分

                  1.最后第三个字符为f,存在fmf,此时必然不符合要求舍去

                  2.最后第三个字符为m,只要前面长度为n-3的串的情况符合要求,则当前长度为n的串必然符合要求。

所以讲符合要求的情况相加就得到:( L )=( L - 1 ) + ( L - 3 ) + ( L - 4 ); 

前面已经讲过如果只是用普通递归方法会超内存,所以这里要考虑优化。

怎么优化?先看下面的矩阵相乘的结果:

x矩阵是多少会得到后面的矩阵?我们只需考虑后面矩阵的第一行,因为其他元素为0.

第1行第1列的元素我们需要得到f ( n ),因为f(n)=f(n-1)+f(n-3)+f(n-4); 所以我们必须保留f(n-1),f(n-3),f(n-4) 所以与之相乘的数必须为1.

所以第1列元素可以确定,为1 0 1 1,注意,是第一列而不是第一行。

根据第一行第二列元素,我们可以确定x矩阵第二列元素:1 0 0 0.

根据第一行第三列元素,我们可以确定x矩阵第三列元素:0 1 0 0.

根据第一行第四列元素,我们可以确定x矩阵第四列元素:0 0 1 0.

所以x矩阵已经确定,所以我们可以得到下面的矩阵乘式:

所以,反复乘以x矩阵就可以得到想要的f(n);

所以可以先求出x矩阵的L-4(不是L)次方,到这就转化为了矩阵快速幂问题。然后在用  f(4)   f(3)  f(2)   f(1)   乘以求次方后的矩阵的第一列元素  ,相加就得到f(n)=res[0][0]*f[4]+res[1][0]*f[3]+res[2][0]*f[2]+res[3[0]*f[1]。


#include<iostream>
#include<cstring>
#include<string>
#define maxn 5
using namespace std;
struct mat{
    int a[maxn][maxn];
};
mat mat_mul(mat x,mat y,int Mod){
    mat ans;
    memset(ans.a,0,sizeof(ans.a));
    for (int i=0;i<4;i++)
        for (int j=0;j<4;j++)
        for (int k=0;k<4;k++){
            ans.a[i][j]+=x.a[i][k]*y.a[k][j];
            ans.a[i][j]%=Mod;
        }
    return ans;
}
void mat_pow(mat &res,int k,int Mod){   //res的k次方 
    mat c=res;
    k--;
    while (k){
        if (k&1) res=mat_mul(res,c,Mod);
        k>>=1;
        c=mat_mul(c,c,Mod);
    }
}
int main(){
    int l,m;
    while (cin >> l >> m){
        int f[10]={0};
        f[1]=2;f[2]=4;f[3]=6;f[4]=9;
        mat res;
        memset(res.a,0,sizeof(res.a));
        res.a[0][0]=res.a[0][1]=res.a[1][2]=res.a[2][0]=res.a[2][3]=res.a[3][0]=1;
        if (l<=4){
            cout << f[l]%m << endl;
            continue;
        }
        else mat_pow(res,l-4,m);
        int ans=0;
        for (int i=0;i<4;i++){
            ans+=res.a[i][0]*f[4-i]%m;
        }
        cout << ans%m << endl;
    }
}
原文地址:https://www.cnblogs.com/changer-qyz/p/8442829.html