結果
問題 | No.1772 Many DELETEQs |
ユーザー | 👑 SPD_9X2 |
提出日時 | 2021-11-23 18:46:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,103 ms / 3,500 ms |
コード長 | 1,611 bytes |
コンパイル時間 | 612 ms |
コンパイル使用メモリ | 87,292 KB |
実行使用メモリ | 339,612 KB |
最終ジャッジ日時 | 2023-09-07 22:00:19 |
合計ジャッジ時間 | 9,835 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 980 ms
328,420 KB |
testcase_01 | AC | 982 ms
328,620 KB |
testcase_02 | AC | 977 ms
328,236 KB |
testcase_03 | AC | 1,103 ms
339,172 KB |
testcase_04 | AC | 1,096 ms
339,612 KB |
testcase_05 | AC | 1,102 ms
339,092 KB |
testcase_06 | AC | 1,055 ms
339,584 KB |
ソースコード
""" Many Deleteqs 想定解v1.1 Qの制約チェックを付けた 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 sys from sys import stdin mod = 998244353 dp = [[0] * 4001 for i in range(4001)] dp[0][0] = 1 for i in range(4001): for j in range(4001): if i != 0 and j != 0: #add AA or BB dp[i][j] += 2 * dp[i-1][j-1] if i != 0: #add AB dp[i][j] += dp[i-1][j] if j != 0: dp[i][j] += dp[i][j-1] dp[i][j] %= mod dp2 = [ [0] * (4002) for i in range(4002) ] for i in range(4001): for j in range(i+1): a = i - j dp2[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] %= mod Q = 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 <= 4000 assert 1 <= y <= 4000 if x < y: x,y = y,x a = x-y ilim = min(x , y+a) ANS.append( dp2[a][ilim] ) print ("\n".join(map(str,ANS)))