矩阵快速幂

斐波那契

poj3070

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
struct mat
{
    int m[2][2];
};
mat I=
{
    1,0,
    0,1
};
mat I1=
{
    1,1,
    1,0
};
mat mul(mat a,mat b)
{
    mat c;
    for(int i=0;i<2;i++)
    {
        for(int j=0;j<2;j++)
        {
            c.m[i][j]=0;
            for(int k=0;k<2;k++)
            {
                c.m[i][j]+=(a.m[i][k]*b.m[k][j]);
                c.m[i][j]%=10000;
            }
        }
    }
    return c;
}
mat quick(int m)
{
    mat ans=I,gg=I1;
    while(m)
    {
        if(m&1)
            ans=mul(ans,gg);
        gg=mul(gg,gg);
        m>>=1;
    }
    return ans;
}

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        if(n<0) return 0;
        if(n==0) printf("0\n");
        else if(n==1) printf("1\n");
        else if(n==2) printf("1\n");
        else
        {
            mat anss=quick(n);
            printf("%d\n",anss.m[0][1]);
        }
    }
    return 0;
}

点的平移,缩放、翻转和旋转


#include<stdio.h>
#include<math.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define PI acos(-1.0)
const int maxx=3;
struct mat
{
    double m[maxx][maxx];
};
mat I;
mat ans;
double x[10005];
double y[10005];
mat mul(mat a,mat b)
{
    mat c;
    for(int i=0; i<3; i++)
    {
        for(int j=0; j<3; j++)
        {
            c.m[i][j]=0;
            for(int k=0; k<3; k++)
            {
                c.m[i][j]+=(a.m[i][k]*b.m[k][j]);
            }
        }
    }
    return c;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=0; i<n; i++)
        {
            scanf("%lf%lf",&x[i],&y[i]);
        }
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++)
            i==j?ans.m[i][j]=1:ans.m[i][j]=0;
        for(int i=0; i<m; i++)
        {
            memset(I.m,0,sizeof(I.m));
            getchar();
            char a;
            scanf("%c",&a);
            if(a=='X')//绕x轴旋转
            {
                I.m[0][0]=1.0;
                I.m[1][1]=-1.0;
                I.m[2][2]=1.0;
            }
            else if(a=='Y')//绕y轴旋转
            {
                I.m[0][0]=-1.0;
                I.m[1][1]=1.0;
                I.m[2][2]=1.0;
            }
            else if(a=='M')//移动
            {
                double p,q;
                scanf("%lf%lf",&p,&q);
                I.m[0][0]=1.0;
                I.m[1][1]=1.0;
                I.m[2][2]=1.0;
                I.m[0][2]=p;
                I.m[1][2]=q;
            }
            else if(a=='S')//放缩
            {
                double p;
                scanf("%lf",&p);
                I.m[0][0]=p;
                I.m[1][1]=p;
                I.m[2][2]=1.0;
            }
            else if(a=='R')//旋转
            {
                double p;
                scanf("%lf",&p);
                I.m[0][0]=cos(PI*p/180.0);
                I.m[0][1]=-sin(PI*p/180.0);
                I.m[1][0]=sin(PI*p/180.0);
                I.m[1][1]=cos(PI*p/180.0);
                I.m[2][2]=1.0;
            }
            ans=mul(I,ans);
        }
        for(int i=0; i<n; i++)
        {
            memset(I.m,0,sizeof(I.m));
            I.m[0][0]=x[i];
            I.m[1][0]=y[i];
            I.m[2][0]=1;
            I=mul(ans,I);
            printf("%.1lf %.1lf\n",I.m[0][0],I.m[1][0]);
        }
    }
    return 0;
}

求A + A^2 + A^3 + … + A^k的结果

poj3233

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <string.h>
using namespace std;
#define MAXV 70

typedef struct
{
    int r,c;            //c行r列
    int mat[MAXV][MAXV];
} Matrix;

Matrix ans,cnt;
int n,k,m;

void Input()
{
    int i,j;
    /*构造B矩阵*/
    memset(cnt.mat,0,sizeof(cnt.mat));
    memset(ans.mat,0,sizeof(ans.mat));
    for(i=0; i<n; i++)
    {
        for(j=0; j<n; j++)
            scanf("%d",&cnt.mat[i][j]);
    }
    for(i=0; i<n; i++)
    {
        cnt.mat[i+n][i+n]=cnt.mat[i][i+n]=1;
        ans.mat[i][i]=ans.mat[i+n][i+n]=1;
    }

    /*对矩阵B和B^(k+1)初始化*/
    cnt.c=cnt.r=2*n;
    ans.c=ans.r=2*n;
}

Matrix MatrixMul(Matrix x,Matrix y)     //矩阵乘法
{
    Matrix t;
    int i,j,v;
    memset(t.mat,0,sizeof(t.mat));
    t.c=x.c;
    t.r=y.r;

    for(i=0; i<t.c; i++)
        for(j=0; j<t.r; j++)
        {
            for(v=0; v<x.r; v++)
                t.mat[i][j]+=((x.mat[i][v]*y.mat[v][j])%m);
            t.mat[i][j]=t.mat[i][j]%m;
        }
    return t;
}

void Binary()
{
    //二分快速幂
    k++;
    while(k)
    {
        if(k & 1) ans=MatrixMul(ans,cnt);
        cnt=MatrixMul(cnt,cnt);
        k=k>>1;
    }
}

void Output()
{
    /*输出的时候要减去一个单位矩阵*/
    int i,j;
    for(i=0; i<n; i++)
    {
        for(j=0; j<n; j++)
            if(i!=j)
                printf("%d ",ans.mat[i][j+n]);
            else
                printf("%d ",ans.mat[i][j+n]-1);
        printf("\n");
    }
}

int main()
{
    while(~scanf("%d%d%d",&n,&k,&m))
    {
        Input();
        Binary();
        Output();
    }
    return 0;
}

给定一个有向图,问从A点恰好走k步(允许重复经过边)到达B点的方案数mod p的值

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
#define MOD 1000
struct matrix
{
    int mat[31][31];
    matrix(){memset(mat,0,sizeof(mat));}
};
int n;
matrix mul(matrix A,matrix B)
{
    int i,j,k;
    matrix C;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            for(k=1;k<=n;k++)
            {
                C.mat[i][j]=(C.mat[i][j]+A.mat[i][k]*B.mat[k][j])%MOD;
            }
        }
    }
    return C;
}
matrix powmul(matrix A,int k)
{
    matrix B;
    for(int i=1;i<=n;i++)B.mat[i][i]=1;
    while(k)
    {
        if(k&1)B=mul(B,A);
        A=mul(A,A);
        k>>=1;
    }
    return B;
}
int main()
{
    int s,t,m,T,a,b,k;
    while(~scanf("%d%d",&n,&m)&&(n||m))
    {
        matrix A,B;
        while(m--)
        {
            scanf("%d%d",&s,&t);
            A.mat[s+1][t+1]=1;
        }
        scanf("%d",&T);
        while(T--)
        {
            scanf("%d%d%d",&a,&b,&k);
            B=A;
            B=powmul(A,k);
            cout<<B.mat[a+1][b+1]<<endl;
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值