矩阵的快速幂

#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<iostream>
using namespace std;

struct matrix
{
    long long a[3][3];
}origin,res;

matrix multiply(matrix x,matrix y)
{
    matrix temp;
    memset(temp.a,0,sizeof(temp.a));
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
        {
            for(int k=0;k<3;k++)
            {
                temp.a[i][j] += x.a[i][k] * y.a[k][j];
            }
        }
    }
    return temp;
}
void init()
{
    int i,j;
    cout<<"随机数组如下:"<<endl;
    for(i=0;i<3;i++)
    {
        for(j=0;j<3;j++)
        {
            origin.a[i][j] = rand() % 10;  //随机数生成一个原始矩阵
            cout<<"  "<<origin.a[i][j];
        }
        cout<<endl;
        memset(res.a,0,sizeof(res.a));
        res.a[0][0] = res.a[1][1] = res.a[2][2] = 1; //初始化单位矩阵
    }
}

void calc(int n)  //矩阵的快速幂
{
    while(n)
    {
        if(n & 1) //奇数  代码每进行一次，二进制数就少了最后面的一个1。二进制数有多少个1就第3行代码就执行多少次。
        {
            res = multiply(res,origin); //单位矩阵和原始矩阵相乘,奇数可以拆分成偶数次+一次,例如n^5 = n * n ^ 4
        }
        n >>= 1;
        origin = multiply(origin,origin);
    }
    cout<<n<<"次幂结果如下:"<<endl;
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
        {
            cout<<"   "<<res.a[i][j];
        }
        cout<<endl;
    }
}

int main()
{
    int N;
    while(cin>>N)
    {
        init();
        calc(N);
    }
    return 0;
}