結果
問題 | No.2631 Rectangle Grid Game |
ユーザー |
![]() |
提出日時 | 2024-02-16 23:37:54 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,775 bytes |
コンパイル時間 | 152 ms |
コンパイル使用メモリ | 82,368 KB |
実行使用メモリ | 57,232 KB |
最終ジャッジ日時 | 2024-09-28 22:38:51 |
合計ジャッジ時間 | 2,563 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 WA * 8 |
ソースコード
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-=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)