結果

問題 No.1810 RGB Biscuits
ユーザー MasKoaTSMasKoaTS
提出日時 2022-01-13 22:05:35
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 332 ms / 2,000 ms
コード長 1,235 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,776 KB
最終ジャッジ日時 2024-11-17 15:20:25
合計ジャッジ時間 4,025 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 32 ms
11,520 KB
testcase_01 AC 32 ms
11,648 KB
testcase_02 AC 32 ms
11,520 KB
testcase_03 AC 34 ms
11,648 KB
testcase_04 AC 35 ms
11,648 KB
testcase_05 AC 302 ms
11,648 KB
testcase_06 AC 332 ms
11,776 KB
testcase_07 AC 327 ms
11,648 KB
testcase_08 AC 321 ms
11,648 KB
testcase_09 AC 323 ms
11,776 KB
testcase_10 AC 207 ms
11,648 KB
testcase_11 AC 35 ms
11,648 KB
testcase_12 AC 37 ms
11,648 KB
testcase_13 AC 38 ms
11,520 KB
testcase_14 AC 80 ms
11,520 KB
testcase_15 AC 39 ms
11,648 KB
testcase_16 AC 46 ms
11,520 KB
testcase_17 AC 163 ms
11,648 KB
testcase_18 AC 169 ms
11,648 KB
testcase_19 AC 35 ms
11,520 KB
testcase_20 AC 32 ms
11,520 KB
testcase_21 AC 61 ms
11,520 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools as iter
import collections as coll
import heapq as hq
import bisect as bis
from decimal import Decimal as dec
from copy import deepcopy as dcopy
import math
import sys
sys.setrecursionlimit(10**6)
def input():
    return sys.stdin.readline().rstrip()
def getN():
	return int(sys.stdin.readline())
def getNs():
	return map(int,sys.stdin.readline().split())
def getList():
	return list(map(int,sys.stdin.readline().split()))
def strinps(n):
	return [sys.stdin.readline().rstrip() for _ in range(n)]
pi = 3.141592653589793
mod = 10**9+7
MOD = 998244353
INF = math.inf
dx = [1,0,-1,0];	dy = [0,1,0,-1]

def prod_mat(A, B):
	res = [[0, 0],
		[0, 0]]
	for i in range(2):
		for j in range(2):
			for k in range(2):
				res[i][j] += A[i][k] * B[k][j]
				res[i][j] %= mod
	return res

def exp_mat(A, n):
	res = [[1, 0],
		[0, 1]]
	while(n):
		if(n & 1):
			res = prod_mat(A, res)
		n >>= 1
		A = prod_mat(A, A)
	return res

"""
Main Code
"""

a,b = getNs()
n = getN()
t = [getN() for _ in [0] * n]
p = [[a, b],
	 [1, 0]]
for i in t:
	k = i >> 1
	r = i & 1
	q = exp_mat(p, k)
	v = [q[0][0] + q[0][1], q[1][0] + q[1][1]]
	if(r):
		print((v[0] + v[1] + v[0] * a % mod + v[1] * b % mod) % mod)
	else:
		print((v[0] + v[1]) % mod)
0