結果

問題 No.2783 4-33 Easy
コンテスト
ユーザー katonyonko
提出日時 2024-06-14 22:23:13
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 249 ms / 2,000 ms
コード長 2,461 bytes
コンパイル時間 521 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 103,936 KB
最終ジャッジ日時 2024-06-14 22:23:21
合計ジャッジ時間 6,965 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import io
import sys
from collections import defaultdict, deque, Counter
from itertools import permutations, combinations, accumulate
from heapq import heappush, heappop
from bisect import bisect_right, bisect_left
from math import gcd
import math

_INPUT = """\
6
14
1 0 1 0 0 2 0 1 0 3 0 0 1 0
2 7 1 8 2 8 1 8 2 8 4 5X 9X X
81
4 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
33 0 0 0 0 0 0 0 X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X 0 0 0 0 0 0 0 0 1X
52
0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 0 0 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
1 0 0 0 3 1 5X 1 1 0 0 0 5 0 3 X 0 1 0 2 0 0 7 0 0 0 2 0 1 0 0 0 0 0 0 7 1 10 2 0 0 1 X 5 4 0 1 0 0 0 2 X
"""

mod=998244353
def input():
  return sys.stdin.readline()[:-1]

def solve(test):
  N=int(input())
  A=list(map(int,input().split()))
  B=list(input().split())
  d=defaultdict(int)
  ll=[]
  for i in range(N):
    if B[i]=='X':
      d[A[i]]+=1
    if B[i][-1]!='X':
      ll.append((A[i],int(B[i])))
  dp=[0]*(len(ll)+1)*5*34*9
  def idx(x,y,z,w):
    return x*5*34*9+y*34*9+z*9+w
  dp[idx(0,0,0,0)]=1
  for i in range(len(ll)):
    for j in range(5):
      for k in range(34):
        for l in range(9):
          dp[idx(i+1,j,k,l)]+=dp[idx(i,j,k,l)]
          dp[idx(i+1,j,k,l)]%=mod
          if ll[i][0]+j<5 and ll[i][1]+k<34 and l+1<9:
            dp[idx(i+1,j+ll[i][0],k+ll[i][1],l+1)]+=dp[idx(i,j,k,l)]
            dp[idx(i+1,j+ll[i][0],k+ll[i][1],l+1)]%=mod
  ans=sum([d[i]*dp[idx(len(ll),4-i,33,8)] for i in range(5)])%mod
  print(ans)

def random_input():
  from random import randint,shuffle
  N=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")*3

def 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)
0