結果
問題 |
No.1797 永遠のグリッド
|
ユーザー |
![]() |
提出日時 | 2022-01-27 22:57:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 260 ms / 2,000 ms |
コード長 | 1,436 bytes |
コンパイル時間 | 1,545 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 93,056 KB |
最終ジャッジ日時 | 2024-12-26 05:54:28 |
合計ジャッジ時間 | 4,811 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
import sys #input = sys.stdin.readline #文字列につけてはダメ #input = sys.stdin.buffer.readline #文字列につけてはダメ #sys.setrecursionlimit(1000000) #import bisect import itertools #import random #from heapq import heapify, heappop, heappush #from collections import defaultdict #from collections import deque #import copy #import math #from functools import lru_cache #@lru_cache(maxsize=None) #MOD = pow(10,9) + 7 #MOD = 998244353 #dx = [1,0,-1,0] #dy = [0,1,0,-1] def check(x,k): s = set([]) for v in x: s.add(v) if len(s) == k: return True return False def main(): H,W,K = map(int,input().split()) N = H*W base = [i for i in range(K)] L = itertools.product(base,repeat=N) S = set([]) ans = 0 for X in L: if not check(X,K): continue TX = tuple(X) if TX in S: continue #print(TX) ans += 1 NX = [-1]*N for di in range(H): for dj in range(W): for i in range(H): for j in range(W): idx = i * W + j ni = (i + di) % H nj = (j + dj) % W nidx = ni * W + nj NX[nidx] = X[idx] TNX = tuple(NX) #print("T",TNX) S.add(TNX) print(ans) if __name__ == '__main__': main()