poj3233 - Matrix Power Series - 矩阵快速幂+二分

最新推荐文章于 2020-10-17 09:49:19 发布

努力写题的tyf

最新推荐文章于 2020-10-17 09:49:19 发布

阅读量169

点赞数

分类专栏：二分搜索 dfs 快速幂文章标签：矩阵快速幂二分回溯

本文链接：https://blog.csdn.net/m0_37579232/article/details/99607610

版权

dfs 同时被 3 个专栏收录

17 篇文章 0 订阅

订阅专栏

二分搜索

7 篇文章 0 订阅

订阅专栏

快速幂

6 篇文章 0 订阅

订阅专栏

Matrix Power Series

Time Limit: 3000MS		Memory Limit: 131072K
Total Submissions: 31411		Accepted: 12658

Description

Given a n × n matrix A and a positive integer k, find the sum S = A + A2 + A3 + … + Ak.

Input

The input contains exactly one test case. The first line of input contains three positive integers n (n ≤ 30), k (k ≤ 109) and m (m < 104). Then follow n lines each containing n nonnegative integers below 32,768, giving A’s elements in row-major order.

Output

Output the elements of S modulo m in the same way as A is given.

Sample Input

2 2 4
0 1
1 1

Sample Output

1 2
2 3

思路：

题意还是很好理解的，就是算矩阵快速幂，唯一的难点就是k很大（1e9），所以这里用了二分与dfs相结合的方法。

1.n为偶数

$S_{10}=A_{1}+A_{2}+A_{3}+A_{4}+A_{5}+A_{6}+A_{7}+A_{8}+A_{9}+A_{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})$

2.n为奇数

$S_{5}=A_{1}+A_{2}+A_{3}+A_{4}+A_{5}$

$S_{5}=A_{1}+A_{2}+A_{3}+A_{3}*(A_{1}+A_{2})$

以n为偶数为例，eg.求 $S_{10}$ ，可用dfs回溯的方法

dfs(10):= $A_{1}+A_{2}+...+A_{10}$

求出ans=dfs(5)

则 $S_{10}$ =ans+ans* $A_{5}$ =ans+ans*(ksm(a,5))

代码如下：

#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cstring>
#include<set>
#include<map>
#include<queue>
#include<vector>

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ll long long
using namespace std;
const int N=35;
const int INF=0x3f3f3f3f;
struct mat{
    ll m[N][N];
}a;

int n,mod;

mat mul(mat a,mat b){
    mat c;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            c.m[i][j]=0;
            for(int k=0;k<n;k++){
                c.m[i][j]=(c.m[i][j]+a.m[i][k]*b.m[k][j])%mod;
            }
        }
    }
    return c;
}

mat ksm(mat c,int k){
    mat res;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            res.m[i][j]=(i==j?1:0);
        }
    }
    while(k){
        if(k&1)res=mul(res,c);
        c=mul(c,c);
        k>>=1;
    }
    return res;
}

mat add(mat a,mat b){
    mat c;
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            c.m[i][j]=(a.m[i][j]+b.m[i][j])%mod;
        }
    }
    return c;
}

mat dfs(int k){//求解:A + A2 + A3 + … + Ak.
    if(k==1){
        return a;
    }
    mat ans=dfs(k/2);
    if(k&1){
        mat tmp=ksm(a,k/2+1);
        ans=add(ans,mul(tmp,ans));
        ans=add(ans,tmp);
    }
    else{
        mat tmp=ksm(a,k/2);
        ans=add(ans,mul(ans,tmp));
    }
    return ans;
}

int main(){
    int k;
    scanf("%d%d%d",&n,&k,&mod);
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            scanf("%lld",&a.m[i][j]);
        }
    }
    mat ans=dfs(k);
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            printf("%lld ",ans.m[i][j]%mod);
        }
        printf("\n");
    }
}