結果
問題 | No.2631 Rectangle Grid Game |
ユーザー |
![]() |
提出日時 | 2024-02-16 23:38:25 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 44 ms / 2,000 ms |
コード長 | 1,777 bytes |
コンパイル時間 | 466 ms |
コンパイル使用メモリ | 82,568 KB |
実行使用メモリ | 56,344 KB |
最終ジャッジ日時 | 2024-09-28 22:41:29 |
合計ジャッジ時間 | 2,624 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
import ioimport sysfrom collections import defaultdict, deque, Counterfrom itertools import permutations, combinations, accumulatefrom heapq import heappush, heappopfrom bisect import bisect_right, bisect_leftfrom math import gcdimport math_INPUT = """\63 42 23141 5926"""mod=998244353def solve(test):H,W=map(int, input().split())ans=pow(H*W,2,mod)if H%2==0:if W%2==0:h,w=H//2-1,W//2-1ans-=4*(h*(h+1)//2*w*(w+1)//2)%modans-=2*2*H*w*(w+1)//2%modans-=2*2*W*h*(h+1)//2%modans-=4*H*W%modelse:h,w=H//2-1,W//2ans-=4*(h*(h+1)//2*w*(w+1)//2)%modans-=2*2*H*w*(w+1)//2%modans-=2*W*h*(h+1)//2%modans-=2*H*W%modelse:if W%2==0:h,w=H//2,W//2-1ans-=4*(h*(h+1)//2*w*(w+1)//2)%modans-=2*H*w*(w+1)//2%modans-=2*2*W*h*(h+1)//2%modans-=2*H*W%modelse:h,w=H//2,W//2ans-=4*h*(h+1)//2*w*(w+1)//2%modans-=2*H*w*(w+1)//2%modans-=2*W*h*(h+1)//2%modans-=H*W%modprint(ans%mod)def random_input():from random import randint,shuffleN=randint(1,10)M=randint(1,N)A=list(range(1,M+1))+[randint(1,M) for _ in range(N-M)]shuffle(A)return (" ".join(map(str, [N,M]))+"\n"+" ".join(map(str, A))+"\n")*3def simple_solve():return []def main(test):if test==0:solve(0)elif test==1:sys.stdin = io.StringIO(_INPUT)case_no=int(input())for _ in range(case_no):solve(0)else:for i in range(1000):sys.stdin = io.StringIO(random_input())x=solve(1)y=simple_solve()if x!=y:print(i,x,y)print(*[line for line in sys.stdin],sep='')break#0:提出用、1:与えられたテスト用、2:ストレステスト用main(0)