递归

最新推荐文章于 2023-05-01 16:07:22 发布

dlxdecs

最新推荐文章于 2023-05-01 16:07:22 发布

阅读量411

点赞数

分类专栏：数据结构文章标签：矩阵乘法快速幂

本文链接：https://blog.csdn.net/u012629369/article/details/25974833

版权

数据结构专栏收录该内容

37 篇文章 0 订阅

订阅专栏

描述 Description

考虑一下定义在非负整数n上的递推关系：
       | f0                 当n=0时
F(n)= | f1                 当n=1时
       | a*F(n-1)+b*F(n-2)  其它

给定f0,f1,a,b,n,写一个程序计算F(n)，-10^9≤F(n)≤10^9（四舍五入），保证一定有解。

输入格式 InputFormat

一行有五个数，f0,f1,a,b,n。-10^9≤n,f0,f1≤10^9，10^6≤a,b≤10^6。
a,b是实数，n,f0,f1是整数。

输出格式 OutputFormat

一个数F(n)。

样例输入 SampleInput

样例1：0 1 1 1 20

样例2：0 1 -1 0 1000000000

样例3：-1 1 4 -3 18

样例输出 SampleOutput

样例1：6765

样例2：-1

样例3：387420487

思路：

已无力吐槽， 搞了好几个小时， 就是wa， 查数据才

知道原来是A，B可以是小数。。。

巧妙的运用矩阵乘法递推， 加快速幂。

{0 1}      {f0}       {f1}          {f0}

{B A} 乘以 {f1} 可得到 {f2} ...所以  {f1}的n-

          {f0}        {fn-1}

1次幂再乘以{f1}即可得到{fn  }, 即可求fn。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

typedef struct Node
{
	double data[2][2];
}node;

node multy(node matrix1, node matrix2)      //矩阵乘法
{
	node ans;
	int i, j, k;
	double temp;
	for(i = 0; i < 2; i++)
	{
		for(j = 0; j < 2; j++)
		{
			temp = 0;
			for(k = 0; k < 2; k++)
			{
				temp += matrix1.data[i][k]*matrix2.data[k][j];
			}
			ans.data[i][j] = temp;
		}
	}
	return ans;
}

node ksm(node item, int n)          //快速幂
{
	node temp;
	temp.data[0][0] = 1.0;
	temp.data[0][1] = 0.0;
	temp.data[1][0] = 0.0;
	temp.data[1][1] = 1.0;
	while(n != 0)
	{
		if(n%2 != 0)
		{
			temp = multy(temp, item);
		}
		item = multy(item, item);
		n /= 2;
	}
	return temp;
}

int main()
{
	node temp = {0, 1, 0, 0};
	int i, n;
	double f0, f1;
	scanf("%lf%lf%lf%lf%d", &f0, &f1, &temp.data[1][1], &temp.data[1][0], &n);
	temp = ksm(temp, n-1);
	printf("%.0lf\n", 1.0*temp.data[1][0]*f0+1.0*temp.data[1][1]*f1);
	return 0;
}