結果
問題 | No.1772 Many DELETEQs |
ユーザー |
👑 ![]() |
提出日時 | 2021-11-23 18:46:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,087 ms / 3,500 ms |
コード長 | 1,611 bytes |
コンパイル時間 | 389 ms |
コンパイル使用メモリ | 81,996 KB |
実行使用メモリ | 338,560 KB |
最終ジャッジ日時 | 2024-06-25 15:37:43 |
合計ジャッジ時間 | 9,513 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 5 |
ソースコード
"""Many Deleteqs 想定解v1.1Qの制約チェックを付けたdp[i][j] = ABをi個、BAをj個最小で使うことで作成可能な文字列の個数AB,BAは対消滅可能なので、i <= x , j <= y , i-j = x-yとなる i,jに関して総和を求めればよい。x-y >= 0 となるように、x,yを交換する。iを決めると、jも決まる。ここから、iの範囲が決まる。後は、x-yごとにindexをiとして累積和を取っておくとクエリO(1)dp2[a][b] = (i-j==a) かつ、i <= b の場合の dp[i][j] の総和"""import sysfrom sys import stdinmod = 998244353dp = [[0] * 4001 for i in range(4001)]dp[0][0] = 1for i in range(4001):for j in range(4001):if i != 0 and j != 0:#add AA or BBdp[i][j] += 2 * dp[i-1][j-1]if i != 0:#add ABdp[i][j] += dp[i-1][j]if j != 0:dp[i][j] += dp[i][j-1]dp[i][j] %= moddp2 = [ [0] * (4002) for i in range(4002) ]for i in range(4001):for j in range(i+1):a = i - jdp2[a][i] += dp[i][j]dp2[a][i] %= mod#累積和for a in range(4001):for b in range(4000):dp2[a][b+1] += dp2[a][b]dp2[a][b+1] %= modQ = int(stdin.readline())assert 1 <= Q <= 2*(10**5)ANS = []for loop in range(Q):x,y = map(int,stdin.readline().split())#制約チェックassert 1 <= x <= 4000assert 1 <= y <= 4000if x < y:x,y = y,xa = x-yilim = min(x , y+a)ANS.append( dp2[a][ilim] )print ("\n".join(map(str,ANS)))