矩阵乘法快速幂

今天员工小c写题的时候遇到了在第四黑厂时期就没填的坑,今日再次见面,自然是无奈,但是他请教了他的好朋友枫系,原来需要矩阵乘法快速幂的知识。

Number Sequence
时间限制: 1 Sec  内存限制: 256 MB
题目描述
此题HDUOJ数据过水,网上题解中直接将n%49的做法是错误的。
A number sequence is defined as follows:

f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.

Given A, B, and n, you are to calculate the value of f(n).

输入
The input consists of multiple test cases. Each test case contains 3 integers A, B and n on a single line (1 <= A, B <= 1000, 1 <= n <= 100,000,000). Three zeros signal the end of input and this test case is not to be processed.

输出
For each test case, print the value of f(n) on a single line.

样例输入
1 1 3
1 2 10
0 0 0

样例输出
2
5

这就是小c遇到的题目,小c最早找规律半天也没找出来,就去问度娘,结果度娘所有答案都是%49,小C一看,稳了,马上又A一题。(第n次不读题,题上说了%49错的) Ctrl C+Ctrl V = WA。非常尴尬。

然后小c没办法呀,只能请教他的新朋友枫系,枫系二话不说,就很仗义地把自己的代码展示给了小C,小C一看两看三看四看都看不明白,后来才知道这是矩阵乘法快速幂,可惜网上的博客并没有教会小C,于是小C只能再次求助枫系。

原来矩阵乘法快速幂需要线性代数的知识,还好小C也是个带学生。这题看似是斐波那契递推的升级版,实则应该采用矩阵的方法优化。

图片来自枫系。说实话这个图清晰明了,网上博客要是有个这图,小C早会了(逃

 这是代码,也可以当作模板采用,注释的非常详细

#include<bits/stdc++.h>//本代码来自于一位小c的好朋友枫系,向他表示感谢,他教会了我,我才能教给大家 
using namespace std;
struct juzhen
{
	int x[2][2];//定义一个矩阵 
};
juzhen mul(juzhen a,juzhen b)//这个部分是用作矩阵相乘的模拟; 
{
	juzhen c;//定义一个矩阵类型的c 
	memset(c.x,0,sizeof(c.x));//将c初始化 
	for(int i=0;i<2;i++)//矩阵乘法大模拟 
		for(int j=0;j<2;j++)//这个需要学过线性代数才能模拟 
			for(int k=0;k<2;k++)//线性代数应该是大一的东西 
				c.x[i][j]=(c.x[i][j]+a.x[i][k]*b.x[k][j])%7;//当然也不是很难,百度看一看就会了 
	return c;
}
juzhen pow(juzhen now,int n)//矩阵版本的快速幂 
{
	juzhen ans;//定义一个juzhen类型的ans,矩阵初始化*4,main函数那里说了 
	ans.x[0][0]=1;//左上角a       矩阵 
	ans.x[0][1]=0;//右上角b      a   b 
	ans.x[1][0]=0;//左下角1      1   0
	ans.x[1][1]=1;//右下角0 
	while(n)
	{
		if(n&1)//如果这一位是1 
			ans=mul(ans,now);//ans就乘以一下 
		now=mul(now,now);//now进位 
		n=n/2;//n右移一位 
	}
	return ans;//返回ans 
}
void print(juzhen a)//此处是用来在写矩阵的时候检查用的,交代码的时候不调用,不是我的代码,就把它留作纪念吧 
{
	for(int i=0;i<2;i++)
	{
		for(int j=0;j<2;j++)
			cout<<a.x[i][j]<<' ';
		cout<<endl;
	}
	cout<<endl;
}
int main()
{
	int a,b,c;
	for(cin>>a>>b>>c;!(a==0&&b==0&&c==0);cin>>a>>b>>c)
	{
		if(c==1||c==2)//一和二就直接输出吧 
			cout<<1;
		else  
		{
			juzhen ans;//定义一个juzhen类型的ans 
			ans.x[0][0]=a;//左上角a       矩阵 
			ans.x[0][1]=b;//右上角b      a   b 
			ans.x[1][0]=1;//左下角1      1   0
			ans.x[1][1]=0;//右下角0 
			ans=pow(ans,c-1);//跟快速幂一样 
			cout<<(ans.x[1][0]+ans.x[1][1])%7;//解释看下方图片 
		}
		cout<<endl; 
	}
	return 0;
}

再给一波普通的快速乘和快速幂模板吧。

快速乘:

//快速乘
int fast_mul(int a,int b,int c)
{
    int ans=0;
    a%=c;
    while(b)
    {
        if(b&1)
            ans=(a+ans)%c;
        a=(a+a)%c;
        b>>=1;
    }
    return ans%c;
}
//快速乘之神奇写法 
long long mul (long long x,long long y)
{
	return ((x*y-(long long)(((long double)x*y+0.5)/mod)*mod)%mod+mod)%mod;
}

快速幂:

//快速幂
long long qpow(long long a,long long b,long long p)
{
    long long x=a;
    long long ans=1;
    while(b)
    {
        if(b%2!=0)
            ans*=x;
        ans%=p;
        x*=x;
        x%=p;
        b/=2;
    }
    return ans;
}

 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值