萌新求助玄学问题
查看原帖
萌新求助玄学问题
282624
Alexia_Cosecant楼主2021/11/8 14:50
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int mod=1e9+7;
long long t[3][3],A[3][3],ans[3][3];
#define re register int
inline void ksm(int x)
{
	while(x)
	{
		if(x&1)
		{
			memset(t,0,sizeof(t));
			for(re i=1;i<=2;i++)
				for(re j=1;j<=2;j++)
					for(re k=1;k<=2;k++)
						t[i][j]=(t[i][j]+ans[i][k]*A[k][j])%mod;
			for(re i=1;i<=2;i++)
                for(re j=1;j<=2;j++)
                    ans[i][j]=t[i][j];
		}
		memset(t,0,sizeof(t));
		for(re i=1;i<=2;i++)
			for(re j=1;j<=2;j++)
				for(re k=1;k<=2;k++)
					t[i][j]=(t[i][j]+A[i][k]*A[k][j])%mod;
		for(re i=1;i<=2;i++)
                for(re j=1;j<=2;j++)
                    A[i][j]=t[i][j];
		x>>=1;
	}
}
int main()
{
	long long n;
    scanf("%lld",&n);
	ans[1][1]=ans[2][2]=1;
	A[1][2]=A[2][1]=A[2][2]=1;
	ksm(n-2);
	int res=(ans[2][1]+ans[2][2])%mod;
	printf("%d",res);
	return 0;
}

看跟一篇题解写的类似,现在改的跟题解基本上相同,却TLE两个点,WA一个点。太玄学了!求助。

2021/11/8 14:50
加载中...