結果

問題 No.1349 Subset Product Queries
コンテスト
ユーザー KoD
提出日時 2021-01-12 21:14:38
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 554 bytes
コンパイル時間 265 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 283,492 KB
最終ジャッジ日時 2024-11-25 23:15:05
合計ジャッジ時間 73,782 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 1
other AC * 1 WA * 9 TLE * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np

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