結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
コンテスト
ユーザー fmhr
提出日時 2017-06-09 23:04:44
言語 Go
(1.23.4)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 753 bytes
コンパイル時間 14,702 ms
コンパイル使用メモリ 221,900 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-22 17:09:51
合計ジャッジ時間 15,111 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

import (
	"fmt"
	"log"
)

var N, M int

func main() {
	log.SetFlags(log.Lshortfile)
	fmt.Scan(&N, &M)
	A := make([][]int, 2)
	for i := range A {
		A[i] = make([]int, 2)
	}
	A[0][0] = 1
	A[0][1] = 1
	A[1][0] = 1
	A[1][1] = 0
	B := make([][]int, 2)
	for i := range B {
		B[i] = make([]int, 2)
	}
	for i := range A {
		B[i][i] = 1
	}
	for N > 0 {
		if N&1 == 1 {
			B = mul(B, A)
		}
		A = mul(A, A)
		N >>= 1
	}
	// log.Println(B)
	fmt.Println(B[1][1])
}

func mul(a, b [][]int) [][]int {
	c := make([][]int, len(a))
	for i := range c {
		c[i] = make([]int, len(b[0]))
	}
	for i := 0; i < len(a); i++ {
		for k := 0; k < len(b); k++ {
			for j := 0; j < len(b[0]); j++ {
				c[i][j] = (c[i][j] + a[i][k]*b[k][j]) % M
			}
		}
	}
	return c
}
0