矩阵乘法(四):分析问题,确定递推式,采用矩阵快速幂求解

应用矩阵快速幂运算可以解决递推问题。在实际应用中,有时候题目并没有直接给出递推式,需要认真分析问题,找出递推式,然后再利用矩阵快速幂运算加快问题的求解。

【例1】程序阅读理解。

有如下的C语言程序:

#include <stdio.h>
int main()
{
     int n,m,f,i;
     while(scanf("%d%d",&n,&m)!=EOF)
     {
           f=0;
           for(i=1;i<=n;i++)
           {
                if (i&1)f=(f*2+1)%m;
                else f=f*2%m;
           }
           printf("%d\n",f);
     }
     return 0;
}

阅读上面的程序,根据输入的n和m,写出程序运行的结果。例如,输入 3  10,输出应为5。

但由于给定输入的n和m的数据范围为1<=n, m <= 1000000000,且测试集中数据量较大,因此如果直接将给定的程序提交会超时的。请你编写一个程序,能根据输入的n和m快速完成问题的求解,以实现给定程序的功能。

(1)编程思路。

给定程序段实际是通过迭代的方式求f(n)%m的值。先不考虑求余,找到f(n)的求法。

分析给定程序知,f(0)=0, 当 n为奇数时,f(n)=2*f(n-1)+1;当n为偶数时,f(n)=2*f(n-1)。

下面进一步分析,找到不考虑n的奇偶性的一个统一的递推式。

当 n为奇数时,f(n)=2*f(n-1)+1,n-1一定为偶数,f(n-1)=2*f(n-2)。因此,

f(n)=f(n-1)+f(n-1)+1=2*f(n-2)+f(n-1)+1。

当 n为偶数时,f(n)=2*f(n-1),n-1一定为奇数,f(n-1)=2*f(n-2)+1。因此,

f(n)=f(n-1)+f(n-1)=2*f(n-2)+f(n-1)+1。

由此,得到统一的递推式: f(0)=0,f(1)=1,  f(n)=2*f(n-2)+f(n-1)+1  (n>=3)。

确定了递推式后,可以构造矩阵P,进行快速幂运算求解。

(2)源程序。

#include <stdio.h>
#include <string.h>
struct Matrix
{
      __int64 mat[4][4]; // 存储矩阵中各元素
};
Matrix matMul(Matrix a ,Matrix b,int n,int m)
{
      Matrix c;
      memset(c.mat,0,sizeof(c.mat));
      int i,j,k;
      for (k = 1; k<=n ; k++)
          for (i=1 ;i<=n ; i++)
              if (a.mat[i][k]!=0)
                  for (j = 1 ;j<=n ;j++)
                      c.mat[i][j] = (c.mat[i][j] + a.mat[i][k] * b.mat[k][j]) % m;
      return c;
}
Matrix quickMatPow(Matrix a ,int n,int b,int m) // n阶矩阵a快速b次幂
{
      Matrix c;
      memset(c.mat ,0 ,sizeof(c.mat));
      int i;
      for (i = 1 ;i <= n ;i++)
           c.mat[i][i] = 1;
      while (b!=0)
      {
           if (b & 1)
               c = matMul(c ,a ,n,m); // c=c*a;
           a = matMul(a ,a ,n,m); // a=a*a
           b /= 2;
      }
      return c;
}
int main()
{
      int n,m;
      __int64 ans;
      Matrix p;
      while(scanf("%d%d" ,&n,&m)!=EOF)
      {
            memset(p.mat,0,sizeof(p.mat));
            p.mat[2][1]=2;
            p.mat[1][2]=p.mat[2][2]=1;
            p.mat[2][3]=p.mat[3][3]=1;
            if (n<3)
                 printf("%d\n",n%m);
           else
           {
                 p = quickMatPow(p,3,n-2,m);
                 ans=p.mat[2][1]% m;
                 ans=(ans+p.mat[2][2]*2)% m;
                 ans=(ans+p.mat[2][3])% m;
                 printf("%I64d\n" ,ans);
          }
      }
      return 0;
}

【例2】将灯全熄灭。

有n个灯排成一行,初始时是全亮的,第一个灯可以按(按下之后改变状态)。然后如果前k个灯全熄灭且第k+1个灯亮,则第k+2个灯可以按。问至少要多少步灭掉所有灯?

例如,n=2时,需要2歩。第1歩灭掉2号灯,第2歩灭掉1号灯。n=3时,需要5歩。第1歩灭掉1号灯,第2歩灭掉3号灯,第3歩点亮1号灯(注意1号灯不点亮,不能直接灭2号灯),第4歩灭掉2号灯,第5歩灭掉1号灯。

(1)编程思路。

设f[n]代表n个全亮的灯变成全熄灭所需的最少步数,也可以代表n个全熄灭的灯变成全点亮所需的最少步数。 
      1)要想灭掉最后一个灯,得先灭掉前n-2个灯(第n-1个灯留亮),需要步数 f[n-2]+1。 
      2)要想灭掉第n-1个灯,得先让第n-2个灯变回亮,要第n-2个灯变回亮,得先让第n-3个灯变回亮...即要把前n-2个灯都变回亮,需要步数 f[n-2]。 
      3)把前n-2个灯变回亮后,就剩下前n-1个灯都是亮的,即剩下的任务就是把n-1个灯灭掉,需要步数 f[n-1]。 
      综上所述:f[n] = 2*f[n-2] + f[n-1] + 1。 (n>=3)  f[1]=1,f[2]=2。

(2)源程序。

#include <stdio.h>
#include <string.h>
#define MOD 200907
struct Matrix
{
       __int64 mat[4][4]; // 存储矩阵中各元素
};
Matrix matMul(Matrix a ,Matrix b,int n)
{
      Matrix c;
      memset(c.mat,0,sizeof(c.mat));
      int i,j,k;
      for (k = 1; k<=n ; k++)
          for (i=1 ;i<=n ; i++)
              if (a.mat[i][k]!=0)
                  for (j = 1 ;j<=n ;j++)
                      c.mat[i][j] = (c.mat[i][j] + a.mat[i][k] * b.mat[k][j]) % MOD;
      return c;
}
Matrix quickMatPow(Matrix a ,int n,int b) // n阶矩阵a快速b次幂
{
      Matrix c;
      memset(c.mat ,0 ,sizeof(c.mat));
      int i;
      for (i = 1 ;i <= n ;i++)
            c.mat[i][i] = 1;
      while (b!=0)
      {
           if (b & 1)
                c = matMul(c ,a ,n); // c=c*a;
           a = matMul(a ,a ,n); // a=a*a
           b /= 2;
      }
      return c;
}
int main()
{
      int n;
      __int64 ans;
      Matrix p;
      while(scanf("%d" ,&n) && n!=0)
      {
           memset(p.mat,0,sizeof(p.mat));
           p.mat[1][2]=2;
           p.mat[1][1]=p.mat[1][3]=1;
           p.mat[2][1]=p.mat[3][3]=1;
           if (n<3)
               printf("%d\n",n%MOD);
           else
           {
                p = quickMatPow(p,3,n-2);
                ans=(p.mat[1][1]*2+p.mat[1][2]+p.mat[1][3])%MOD;
                printf("%I64d\n" ,ans);
           }
      }
      return 0;
}

原文地址:https://www.cnblogs.com/cs-whut/p/11456718.html

时间: 2024-10-09 13:36:25

矩阵乘法(四):分析问题,确定递推式,采用矩阵快速幂求解的相关文章

递推求值【快速幂矩阵】

递推求值 描述 给你一个递推公式: f(x)=a*f(x-2)+b*f(x-1)+c 并给你f(1),f(2)的值,请求出f(n)的值,由于f(n)的值可能过大,求出f(n)对1000007取模后的值. 注意:-1对3取模后等于2   输入 第一行是一个整数T,表示测试数据的组数(T<=10000)随后每行有六个整数,分别表示f(1),f(2),a,b,c,n的值.其中0<=f(1),f(2)<100,-100<=a,b,c<=100,1<=n<=10000000

2018ICPC焦作- Poor God Water 求递推式+矩阵快速幂

题目链接:https://nanti.jisuanke.com/t/31721 题意:一个孩子吃饭,有meat, fish 和 chocolate 三种食物可以选.要求连续三顿饭食物不能完全相同,鱼和肉的前一顿和后一顿不能都是巧克力,巧克力的左右两边不能同时出现鱼和肉. 思路:分九种情况,求出递推式,写出标准矩阵,用矩阵快速幂. 代码: #include <iostream> #include <stdio.h> #include <string.h> #define

矩阵乘法来加速递推式计算

Codevs1281: 给你6个数,m, a, c, x0, n, g Xn+1 = ( aXn + c ) mod m,求Xn 计算递推式,运用矩阵来进行计算加速 然后注意用类似快速幂的方法写一个快速加,避免溢出 怎么把式子化成矩阵,日后再补 1 #include<cstdio> 2 long long mod,a,c,x0,n,g; 3 struct Mat 4 { 5 long long m[2][2]; 6 }base,X0; 7 long long quick_add(long lo

[题解][SHOI2013]超级跳马 动态规划/递推式/矩阵快速幂优化

这道题... 让我见识了纪中的强大 这道题是来纪中第二天(7.2)做的,这么晚写题解是因为 我去学矩阵乘法啦啦啦啦啦对矩阵乘法一窍不通的童鞋戳链接啦 层层递推会TLE,正解矩阵快速幂 首先题意就是给你一个 n 行m 列 的格子图 一只马从棋盘的左上角跳到右下角.每一步它向右跳奇数列,且跳到本行或相邻行. 题意很简单暴力dp的思路也很简单但是数据很恶心虽然远古一点,但毕竟是省选题 1 ≤ n ≤ 50,2 ≤ m ≤ 10^9 不过还是给了我们一点提示:n这么小? 总之我们先找出转移式对于每一个点

hdu 1757 A Simple Math Problem (构造矩阵解决递推式问题)

题意:有一个递推式f(x) 当 x < 10    f(x) = x.当 x >= 10  f(x) = a0 * f(x-1) + a1 * f(x-2) + a2 * f(x-3) + -- + a9 * f(x-10) 同时ai(0<=i<=9) 不是 0 就是 1: 现在给你 ai 的数字,以及k和mod,请你算出 f(x)%mod 的结果是多少 思路:线性递推关系是组合计数中常用的一种递推关系,如果直接利用递推式,需要很长的时间才能计算得出,时间无法承受,但是现在我们已知

hiho 1143 矩阵快速幂 求递推式

题目链接: hihocoder 1143 思路见题目上 快速幂模板: // m^n % k int quickpow(int m,int n,int k) { int b = 1; while (n > 0) { if (n & 1) b = (b*m)%k; n = n >> 1 ; m = (m*m)%k; } return b; } 题解: #include<iostream> #include<cstdio> #include<cstring

递推 +二项式定理(矩阵快速幂)

https://nanti.jisuanke.com/t/A2060 题意:第一个数为f[1] = a ,f[2] = b . 递推式:f[n] = f[n-1] + 2*f[n-2] + n4 . 求f[n]%2147493647. 数据:N,a,b<231 #include<stdio.h> #include<string.h> #include<math.h> #include<queue> #include<algorithm> #

【HDU4990】递推式

题目大意:给定序列 1, 2, 5, 10, 21, 42, 85, 170, 341 …… 求第n项 模 m的结果 递推式 f[i]  = f[i - 2] + 2 ^ (i - 1); 方法一: 构造矩阵, 求递推式 方法二: 直接推公式,递推式求和,得到 f[n] = [2 ^ (n + 1) - 1] / 3 奇数, f[n] = [2 ^ (n + 1) - 2] / 3 偶数: 其实还可以进一步化简, 注意到 2 ^ 2k % 3 = 1, 2 ^ (2k + 1) % 3 = 2,

递推式转化为矩阵形式

EXAMPLE: 递推式: d(n + 2) = p * d(n + 1) + (1 - p) * d(n); 令G(n) = (d(n + 2), d(n + 1))^T; 则 G(n + 1) = M * G(n); 解得 M = p   1 - p 1    0 G(n) = (M ^ n) * G(0); #