結果

問題 No.2989 Fibonacci Prize
コンテスト
ユーザー chestnut_68
提出日時 2024-12-14 10:20:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 53 ms / 2,000 ms
コード長 695 bytes
コンパイル時間 1,822 ms
コンパイル使用メモリ 172,492 KB
実行使用メモリ 12,516 KB
最終ジャッジ日時 2024-12-14 10:20:21
合計ジャッジ時間 4,712 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 77
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
	int64_t N,M,T=-1;
	cin>>N>>M;
	
	if(M==1||M==2){
		if(N==1)cout<<2;
		else cout<<0;
		return 0;
	}
	if(N==1){
		cout<<(M-1)*(M-2)/2+3;
		return 0;
	}
	
	vector<int> F(3);
	vector<int64_t>R(N,0),RR(N,0);
	F[1]=F[2]=1;
	for(int i=3;;i++){
		F.push_back((F[i-1]+F[i-2])%N);
		if(F[i-1]==0&&F[i]==1){
			T=i-1;
			F.push_back(1);
			break;
		}
	}
	int64_t K=(M-2)/T,r=(M-2)%T;
	for(int i=1;i<=T;i++){
		R[(F[i+2]-1+N)%N]++;
		if(i<=r)RR[(F[i+2]-1+N)%N]++;
	}

	int64_t ans=0;
	for(int i=0;i<N;i++){
		R[i]*=K;
		R[i]+=RR[i];
		if(i==0)R[i]++;
		ans+=R[i]*(R[i]-1)/2;
	}
	if(F[(M-1)%T]%N==0)ans++;
	if(F[M%T]%N==0)ans+=2;
	cout<<ans<<endl;
}
0