码迷,mamicode.com
首页 > 其他好文 > 详细

Hdu1575Tr A矩阵

时间:2014-10-29 00:04:18      阅读:267      评论:0      收藏:0      [点我收藏+]

标签:style   blog   io   color   os   for   sp   div   on   

矩阵快速幂,就是快速幂的乘法变成矩阵乘法,其余的都一样。

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <string.h>
using namespace std;
const int mod = 9973;
const int maxn = 12;
struct Matrix
{
    int m[maxn][maxn];
};
int n;
Matrix Mul(Matrix a, Matrix b)
{
    Matrix ans; 
    for (int i = 0; i < n; i++){
        for (int j = 0; j < n; j++){
            ans.m[i][j] = 0;
            for (int k = 0; k < n; k++){
                ans.m[i][j] += (a.m[i][k] * b.m[k][j]);
                ans.m[i][j] %= mod;
            }
        }
    }
    return ans;
}

Matrix fast(Matrix  a, int sum)
{
    Matrix  ans;
    for (int i = 0; i < n;i++)
    for (int j = 0; j < n; j++)
        ans.m[i][j] = (i == j);
    while (sum){
        if (sum & 1) ans = Mul(ans, a);
        a = Mul(a, a);
        sum /= 2;
    }
    return ans;
}

int main()
{
    int t,k;
    cin >> t;
    Matrix a;
    while (t--){
        cin >> n >> k;
        for (int i = 0; i < n;i++)
        for (int j = 0; j < n; j++)
            cin >> a.m[i][j];
        Matrix gao = fast(a, k);
        int ans = 0;
        for (int i = 0; i < n; i++)
            ans += gao.m[i][i], ans %= mod;
        printf("%d\n", ans);
    }
    return 0;
}

 

Hdu1575Tr A矩阵

标签:style   blog   io   color   os   for   sp   div   on   

原文地址:http://www.cnblogs.com/yigexigua/p/4058143.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!