改了CCF书上的代码最后几个点wa了求助
查看原帖
改了CCF书上的代码最后几个点wa了求助
287297
Lyr_ids楼主2021/7/23 11:12

//书本模板

#include<bits/stdc++.h>
using namespace std;
const int mo=1000000007;
const int SZ=2;
struct node{
	long long v[SZ][SZ];
}I,A,B,ans,C,A2,res;
node Get_mul(node a,node b)
{
	memset(res.v,0,sizeof(res.v));
	for(int i=0;i<SZ;i++)
	{
		for(int j=0;j<SZ;j++)
		{
			for(int k=0;k<SZ;k++)
				res.v[i][j]+=(a.v[i][k]*b.v[k][j])%mo;
			res.v[i][j]%=mo;
		}
	}
	return res;
}
node power(int n)
{
	C=I;
	A2=A;
	while(n)
	{
		if(n%2)
			C=Get_mul(C,A2);
		n/=2;
		A2=Get_mul(A2,A2);
	}
	return C;
}
unsigned long long n;
int t;
int main()
{
	memset(I.v,0,sizeof(I.v));
	memset(A.v,0,sizeof(A.v));
	memset(B.v,0,sizeof(B.v));
	I.v[0][0]=I.v[1][1]=1;
	A.v[0][0]=A.v[0][1]=A.v[1][0]=1;
	B.v[0][0]=1;
	B.v[0][1]=2;
	cin>>t;
	while(t--)
	{
		scanf("%lld",&n);
		if(!n)
		{
			printf("0\n");
			continue;
		}
		ans=Get_mul(B,power(--n));
		printf("%lld\n",ans.v[0][0]%mo);
	}
	return 0;
}
2021/7/23 11:12
加载中...