1. 程式人生 > >hdu1575 Tr A 矩陣快速冪

hdu1575 Tr A 矩陣快速冪

Tr A
Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 7383    Accepted Submission(s): 5423


Problem Description
A為一個方陣,則Tr A表示A的跡(就是主對角線上各項的和),現要求Tr(A^k)%9973。

 

Input
資料的第一行是一個T,表示有T組資料。
每組資料的第一行有n(2 <= n <= 10)和k(2 <= k < 10^9)兩個資料。接下來有n行,每行有n個數據,每個資料的範圍是[0,9],表示方陣A的內容。

 

Output
對應每組資料,輸出Tr(A^k)%9973。
 

Sample Input
2
2 2
1 0
0 1
3 99999999
1 2 3
4 5 6
7 8 9
 

Sample Output
2
2686

思路;

很簡單的一道矩陣快速冪,直接套模板就行了

程式碼:

#include<bits/stdc++.h>
using namespace std;
const int mod = 9973;
int n;
struct Mtrix
{
    int a[10][10];
};
Mtrix mul(Mtrix x,Mtrix y)
{
    Mtrix ans;
    for (int i = 0;i < n;i ++)
        for (int  j= 0;j < n;j ++)
        {
            ans.a[i][j] = 0;
            for (int k = 0;k < n;k ++)
            {
                ans.a[i][j] += x.a[i][k] * y.a[k][j];
                ans.a[i][j] %= mod;
            }
        }
    return ans;
}
int quick(Mtrix x,int k)
{
    Mtrix ans;
    for (int i = 0;i < n;i ++)
        for (int j = 0;j < n;j ++)
            ans.a[i][j] = x.a[i][j];
    k --;
    while (k)
    {
        if (k & 1) ans = mul(ans,x);
        k >>= 1;
        x = mul(x,x);
    }
    int res = 0;
    for (int i = 0;i < n;i ++)
        for (int j = 0;j < n;j ++)
            if (i == j)
                res = (res + ans.a[i][j]) % mod;
    return res;
}
int main()
{
    int t;
    scanf("%d",&t);
    while (t --)
    {
        Mtrix c;
        int k;
        scanf("%d %d",&n,&k);
        for (int i = 0;i < n;i ++)
            for (int j = 0;j < n;j ++)
                scanf("%d",&c.a[i][j]);
        printf("%d\n",quick(c,k));
    }
    return 0;
}