結果

問題 No.1516 simple 門松列 problem Re:MASTER
ユーザー ansainansain
提出日時 2021-05-04 03:45:51
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 1,192 ms / 6,000 ms
コード長 3,224 bytes
コンパイル時間 252 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 44,612 KB
最終ジャッジ日時 2024-04-17 10:43:12
合計ジャッジ時間 19,029 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 533 ms
44,364 KB
testcase_01 AC 579 ms
44,364 KB
testcase_02 AC 797 ms
44,236 KB
testcase_03 AC 539 ms
44,236 KB
testcase_04 AC 532 ms
44,232 KB
testcase_05 AC 530 ms
44,232 KB
testcase_06 AC 534 ms
44,104 KB
testcase_07 AC 574 ms
44,500 KB
testcase_08 AC 599 ms
44,364 KB
testcase_09 AC 532 ms
44,108 KB
testcase_10 AC 544 ms
44,496 KB
testcase_11 AC 534 ms
44,240 KB
testcase_12 AC 534 ms
44,240 KB
testcase_13 AC 528 ms
44,232 KB
testcase_14 AC 1,192 ms
44,112 KB
testcase_15 AC 895 ms
44,492 KB
testcase_16 AC 754 ms
44,360 KB
testcase_17 AC 571 ms
44,368 KB
testcase_18 AC 540 ms
44,368 KB
testcase_19 AC 538 ms
44,244 KB
testcase_20 AC 1,142 ms
44,612 KB
testcase_21 AC 1,149 ms
44,364 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import defaultdict, Counter, deque
from itertools import permutations, combinations, product, combinations_with_replacement, groupby, accumulate
import operator
from math import sqrt, gcd, factorial
# from math import isqrt, prod,comb  # python3.8用(notpypy)
#from bisect import bisect_left,bisect_right
#from functools import lru_cache,reduce
#from heapq import heappush,heappop,heapify,heappushpop,heapreplace
import numpy as np
#import networkx as nx
#from networkx.utils import UnionFind
#from numba import njit, b1, i1, i4, i8, f8
#from scipy.sparse import csr_matrix
#from scipy.sparse.csgraph import shortest_path, floyd_warshall, dijkstra, bellman_ford, johnson, NegativeCycleError
# numba例 @njit(i1(i4[:], i8[:, :]),cache=True) 引数i4配列、i8 2次元配列,戻り値i1
def input(): return sys.stdin.readline().rstrip()
def divceil(n, k): return 1+(n-1)//k  # n/kの切り上げを返す
def yn(hantei, yes='Yes', no='No'): print(yes if hantei else no)


def check(a):
    return a[0] != a[1] and a[1] != a[2] and a[2] != a[0] and a[1] in [max(a), min(a)]


def dot2(mat1, mat2, MOD):
    """
    行列積a@bの10**9+7modをオーバーフロー回避するために上下15bitで分割して掛け算する 
    https://ikatakos.com/pot/programming_algorithm/python_tips/avoid_overflow
    """
    mask = (1 << 15) - 1
    mat1h = mat1 >> 15
    mat1l = mat1 & mask
    mat2h = mat2 >> 15
    mat2l = mat2 & mask
    mathh = mat1h @ mat2h % MOD
    matll = mat1l @ mat2l % MOD
    mathl = (mathh + matll - (mat1h - mat1l) @ (mat2h - mat2l)) % MOD
    res = (mathh << 30) + (mathl << 15) + matll
    res %= MOD
    return res


def matrix_multiplication(a, n, k, mod=10**9+7):  # n次正方行列a^k
    ans = np.eye(n, dtype=np.int64)
    while k:
        k, i = divmod(k, 2)
        if i:
            ans = dot2(ans, a, mod)
        a = dot2(a, a, mod)
    return ans


def main():
    mod = 10**9+7
    mod2 = 998244353
    n, k = map(int, input().split())
    mat = [[0]*2*k*k for i in range(2*k*k)]
    ans = [0]*2*k*k
    for a in range(k):
        for b in range(k):
            for c in range(k):
                if check([a, b, c]):
                    mat[a*k+b][b*k+c] += 1
                    mat[a*k+b+k**2][b*k+c+k**2] += 1
                    mat[a*k+b][b*k+c+k**2] += a
            if a != b:
                ans[a*k+b] += 1
                ans[a*k+b+k**2] += a+b
    newmat=[]
    newans=[]
    for x in range(2):
        for a in range(k):
            for b in range(k):
                if a!=b:
                    tmp=[]
                    for y in range(2):
                        for c in range(k):
                            for d in range(k):
                                if c!=d:
                                    tmp.append(mat[a*k+b+x*k**2][c*k+d+y*k**2])
                    newmat.append(tmp)
                    newans.append(ans[a*k+b+x*k**2])    
    mat2=matrix_multiplication(np.array(newmat,dtype=np.int64), 2*k*(k-1), n-2, mod2)
    ans2=dot2(np.array(newans,dtype=np.int64),mat2,mod2)
    #print(ans2[:k*k])
    #print(ans2[k*k:])
    print(ans2[:k*(k-1)].sum()%mod2,ans2[k*(k-1):].sum()%mod2)


if __name__ == '__main__':
    main()
0