2. 小强爱数学

Guapifang

于 2023-03-13 13:11:54 发布

阅读量207

点赞数

分类专栏： LeetCode 文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/weixin_43918046/article/details/129491595

版权

LeetCode 专栏收录该内容

481 篇文章

订阅专栏

在这里插入图片描述

题解

感觉蛮有趣的一个题目，需要推导下通项式。
具体推导如下。

an = x^n + y^n;
a1 = x + y;
a2 = a1(x+y) - xy*2;
a3 = a2(x+y) - xy(x+y);
a4 = a3(x+y) - xy(x^2 + y^2);
....
an = an-1*A - an-2*B;

有了这个推导关系，就很好处理了。

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll mod = 1e9+7;
int T,n;
ll A,B;
ll arr[3];
int main()
{
	cin>>T;
	while(T--)
	{
		cin>>A>>B>>n;
		arr[0] = 2;
		arr[1] = A;
		for(int i=2;i<=n;i++)
		{
			arr[2] = ((arr[1]*A)%mod - (arr[0]*B)%mod + mod)%mod;
			arr[0] = arr[1];
			arr[1] = arr[2];
		}
		cout<<arr[2]<<endl;
	}
	return 0;
}