poj3233经典二分矩阵快速幂

这个有个难点,就是他那个是到10的9次方,所以就是说,应该不能一个一个加,看了别人的方法就是用二分的方法,
思路:这是一道典型的矩阵快速幂的问题。用到两次二分,相当经典。矩阵快速幂。首先我们知道 A^x 可以用矩阵快速幂求出来。其次可以对k进行二分,每次将规模减半,分k为奇偶两种情况,如当k = 10和k = 5时有:

k = 10 有: S(10) = ( A^1+A^2+A^3+A^4+ A^5 ) + A^5 * ( A^1+A^2+A^3+A^4+A^5 ) = S(5) + A^5 * S(5)
k = 5 有: S(5) = ( A^1+A^2 ) + A^3 + A^3 * ( A^1+A^2 ) = S(2) + A^3 + A^3 * S(2)
k = 2 有 : S(2) = A^1 + A^2 = S(1) + A^1 * S(1)。
应用这个式子后,规模k减小了一半。我们二分求出后再递归地计算,即可得到原问题的答案。
然后呢?就是不要乱mol,因为数据分为只有10的4次方,所以不用那么乱mol,我乱mol就tle了

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

int n , mol ;
struct martix{
    int m[50][50];
}ans ,base;

martix mult(martix x,martix y)//矩阵乘法
{
    martix tmp;
    for(int i = 0 ; i < n ; i++)
    for(int j = 0 ; j < n ; j++){
        tmp.m[i][j] = 0;
        for(int k = 0 ; k < n ; k++){
            tmp.m[i][j] = (tmp.m[i][j] + (x.m[i][k])*(y.m[k][j])) % mol;//我中间mol了,但是t了
        }
    }
    return tmp;
}

martix expo(martix x ,int k)//矩阵的幂
{
    martix tmp;
    int i , j;
    for(int i = 0 ; i < n ; i++)
    for(int j = 0 ; j < n ; j++){
        if(i == j) tmp.m[i][j] = 1;
        else tmp.m[i][j] = 0;
    }
    while(k){
        if(k & 1) tmp = mult(tmp , x);
        x = mult(x , x);
        k >>= 1;
    }
    return tmp;
}
martix add(martix a , martix b)//矩阵加法
{
    for(int i = 0 ; i < n ; i++)
    for(int j = 0 ; j < n ; j++)
        a.m[i][j] = (a.m[i][j] % mol + b.m[i][j] % mol ) % mol;
    return a;
}
martix sum( martix x, int k)
{
    martix tmp , y;
    if(k == 1) return x;
    tmp = sum(x , k / 2);
    if(k & 1){
        y = expo(x , k / 2 + 1);
        tmp = add(mult(y , tmp) , tmp);
        return add(tmp , y);
    }
    else{
        y = expo(x , k / 2);
        return add(mult(y , tmp) , tmp);
    }
}
int main()
{
    int k;
    while(~scanf("%d%d%d" , &n , &k , &mol)){
        for(int i = 0 ; i < n ; i++)
            for(int j = 0 ; j < n ; j++)
               scanf("%d",&base.m[i][j]);
        ans = sum(base ,k);
        for(int i = 0 ; i < n ; i++){
            for(int j = 0 ; j < n ; j++)
               printf("%d ",ans.m[i][j]);
            printf("\n");
        }
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
以下是Java解决POJ3233矩阵幂序列问题的代码和解释: ```java import java.util.Scanner; public class Main { static int n, k, m; static int[][] A, E; public static void main(String[] args) { Scanner sc = new Scanner(System.in); n = sc.nextInt(); k = sc.nextInt(); m = sc.nextInt(); A = new int[n][n]; E = new int[n][n]; for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { A[i][j] = sc.nextInt() % m; E[i][j] = (i == j) ? 1 : 0; } } int[][] res = matrixPow(A, k); int[][] ans = matrixAdd(res, E); printMatrix(ans); } // 矩阵乘法 public static int[][] matrixMul(int[][] a, int[][] b) { int[][] c = new int[n][n]; for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { for (int k = 0; k < n; k++) { c[i][j] = (c[i][j] + a[i][k] * b[k][j]) % m; } } } return c; } // 矩阵快速幂 public static int[][] matrixPow(int[][] a, int b) { int[][] res = E; while (b > 0) { if ((b & 1) == 1) { res = matrixMul(res, a); } a = matrixMul(a, a); b >>= 1; } return res; } // 矩阵加法 public static int[][] matrixAdd(int[][] a, int[][] b) { int[][] c = new int[n][n]; for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { c[i][j] = (a[i][j] + b[i][j]) % m; } } return c; } // 输出矩阵 public static void printMatrix(int[][] a) { for (int i = 0; i < n; i++) { for (int j = 0; j < n; j++) { System.out.print(a[i][j] + " "); } System.out.println(); } } } ``` 解释: 1. 首先读入输入的n、k、m和矩阵A,同时初始化单位矩阵E。 2. 然后调用matrixPow函数求出A的k次幂矩阵res。 3. 最后将res和E相加得到结果ans,并输出。 4. matrixMul函数实现矩阵乘法,matrixPow函数实现矩阵快速幂,matrixAdd函数实现矩阵加法,printMatrix函数实现输出矩阵

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值