WA,但是自我感觉代码没问题,求帮改
查看原帖
WA,但是自我感觉代码没问题,求帮改
344382
lmrttx楼主2020/12/21 22:01

谢谢wtcl

#include<bits/stdc++.h>
#define mod 1000000007
using namespace std;
int n;
void mul(int f[2],int a[2][2]){
	//int c[2]={0,0};
	int c[2];
	memset(c,0,sizeof(c));
	for(register int j=0;j<2;j++)
	for(register int k=0;k<2;k++)
	c[j]=(c[j]+(long long)f[k]*a[k][j])%mod;
	memcpy(f,c,sizeof(c));
}
void mulself(int a[2][2]){
	//int c[2][2]={(0,0),(0,0)};
	int c[2][2];
	memset(c,0,sizeof(c));
	for(register int i=0;i<2;i++)
	for(register int j=0;j<2;j++)
	for(register int k=0;k<2;k++){
		c[i][j]=(c[i][j]+(long long)a[i][k]*a[k][j])%mod;
	}
	memcpy(a,c,sizeof(c));
}
int main(){
	scanf("%d",&n);
	int f[2]={0,1};
	int a[2][2]={(0,1),(1,1)};
	for(;n;n>>=1){
		if(n&1)mul(f,a);
		mulself(a);
	}
	printf("%d",f[0]);
	return 0;
} 
2020/12/21 22:01
加载中...