結果

問題 No.718 行列のできるフィボナッチ数列道場 (1)
コンテスト
ユーザー YamaKasa
提出日時 2018-07-29 12:18:53
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,792 bytes
コンパイル時間 2,012 ms
コンパイル使用メモリ 79,572 KB
実行使用メモリ 54,160 KB
最終ジャッジ日時 2024-07-08 14:12:53
合計ジャッジ時間 5,326 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Scanner;

public class Main {
	static long d = 1000000000 + 7;

	public static void main(String[] args) {

		Scanner scan = new Scanner(System.in);
		long N = scan.nextLong();
		scan.close();
		long m;
		if(N % 2 == 1) {
			m = N / 2 + 1;
		}else {
			m = N / 2;
		}
		long [][][]A = new long[40][2][2];
		A[0][0][0] = 1;
		A[0][0][1] = 1;
		A[0][1][0] = 1;
		A[0][1][1] = 0;
		for(int i = 1; i < 40; i++) {
			long a00 = A[i -1][0][0] % d;
			long a01 = A[i -1][0][1] % d;
			long a10 = A[i -1][1][0] % d;
			long a11 = A[i -1][1][1] % d;
			A[i][0][0] = ((a00 * a00) % d + (a01 * a10) % d) % d;
			A[i][0][1] = ((a00 * a01) % d + (a01 * a11) % d) % d;
			A[i][1][0] = ((a10 * a00) % d + (a11 * a01) % d) % d;
			A[i][1][1] = ((a10 * a01) % d + (a11 * a11) % d) % d;
		}

		String num = Long.toBinaryString(4 * m - 1);
		long [][]Am = new long[2][2];
		Am[0][0] = 1;
		Am[1][1] = 1;
		//System.out.println(num.length());
		for(int i = 0; i < num.length(); i++) {
			char c = num.charAt(num.length() - i - 1);
			//char c = num.charAt(i);
			if(c == '1') {
				long [][]At = new long[2][2];
				At = square(Am, A[i]);
				for(int j = 0; j < 2; j++) {
					for(int k = 0; k < 2; k++) {
						Am[j][k] = At[j][k];
					}
				}
				for(int j = 0; j < 2; j++) {
					for(int k = 0; k < 2; k++) {
						Am[j][k] = Am[j][k] % d;
					}
				}
			}
		}

		long a = Am[0][0] % d;
		long b = Am[0][1] % d;
		//System.out.println(a);
		//System.out.println(b);

		// y = 5^(d - 2) mod m を求める
//		long []f = new long[35];
//		f[0] = 1;
//		f[1] = 5;
//		String num1 = Long.toBinaryString(d - 2);
//		for(int i = 2; i < 34; i++) {
//			f[i] = (f[i - 1] * f[i - 1]) % d;
//		}
//		for(int i = 0; i < 5; i++) {
//			System.out.println(f[i]);
//		}
//		long y = 0;
//		for(int i = 0; i < num1.length(); i++) {
//			char c = num1.charAt(num1.length() - i - 1);
//			if(c == '1') {
//				y += f[i];
//				y = y % d;
//			}
//		}

		if(N % 2 == 1) {
			long  ans = (1L - b  + 2L * a) % d;
			long x =  ((2 * d + 1) / 5) % d;
			ans = (ans * x) % d;
			System.out.println(ans);
		}else {
//			long  ans = (-1L + b  + 3L * a) / 5L;
//			System.out.println(ans);
			long ans1 = (-1L + b  + 3L * a) % d;

			long x = ((2 * d + 1) / 5) % d;
			long x2 = (7 * d +  1) / 10;

			x2 = x2 % d;


			ans1 = (ans1 * x) % d;
			System.out.println(ans1);


		}
	}
	static long[][] square(long[][]A1, long[][]A2) {
		long[][]A = new long[2][2];
		for(int i = 0; i < 2; i++) {

			for(int j = 0; j < 2; j++) {
				long t = 0;
				for(int k = 0; k < 2; k++) {
					t += (A1[i][k] * A2[k][j]) % d;
				}
				t = t % d;
				A[i][j] = t;
			}
		}
		return A;
	}
	static void disp(long [][]A) {
		for(int i = 0; i < 2; i++) {
			System.out.println(A[i][0] + " " + A[i][1]);
		}
		System.out.println();
	}
}
0