結果

問題 No.1349 Subset Product Queries
ユーザー KoD
提出日時 2021-01-12 21:14:00
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 583 bytes
コンパイル時間 218 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,616 KB
最終ジャッジ日時 2024-11-25 23:11:06
合計ジャッジ時間 18,368 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 1
other RE * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np
from numba import njit

@njit
def solve(stdin):
  N, Q = stdin[:2]
  A = stdin[2: 2 + N]
  dp = np.full((N + 1, 5001), -1)
  dp[0][0] = 0
  for i in range(N):
    for j in range(5001):
      dp[i + 1][j] = max(dp[i + 1][j], dp[i][j])
      if j + A[i] <= 5000:
        dp[i + 1][j + A[i]] = max(dp[i + 1][j + A[i]], dp[i][j])
    dp[i + 1][0] = i + 1

  for i in range(Q):
    l, r, k = stdin[2 + N + 3 * i: 2 + N + 3 * (i + 1)]
    if dp[r][k] >= l - 1:
      print("Yes")
    else:
      print("No")

solve(np.fromstring(open(0).read(), dtype=np.int64, sep=' '))
0