結果

問題 No.854 公平なりんご分配
ユーザー qib
提出日時 2022-12-29 19:46:48
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,242 bytes
コンパイル時間 556 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 318,500 KB
最終ジャッジ日時 2024-11-24 20:05:29
合計ジャッジ時間 49,044 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 80 TLE * 1 MLE * 11
権限があれば一括ダウンロードができます

ソースコード

diff #

import math

n = int(input())
a = list(map(int, input().split()))

m = 2000
k = int(math.sqrt(10 ** 9))

dp = [None for _ in range(k + 1)]
s = {}
for x in range(2, k + 1):
  if not dp[x] is None:
    continue
  dp[x] = x
  if x <= m:
    s[x] = [0 for _ in range(n + 1)]
  for y in range(2 * x, k + 1, x):
    if not dp[y] is None:
      continue
    dp[y] = x

for i in range(n):
  cur = a[i]
  while cur > 1:
    s[dp[cur]][i + 1] += 1
    cur //= dp[cur]

s[0] = [0 for _ in range(n + 1)]
for i in range(n):
  if a[i] == 0:
    s[0][i + 1] += 1

for x in range(m + 1):
  if x == 0 or dp[x] == x:
    for i in range(1, n + 1):
      s[x][i] += s[x][i - 1]

q = int(input())
for i in range(q):
  p, l, r = map(int, input().split())
  l -= 1
  r -= 1
  if p == 1 or s[0][r + 1] - s[0][l] > 0:
    print("Yes")
    continue

  cnt = {}
  cur = p
  flg = True
  for x in range(2, k + 1):
    if cur == 1:
      break
    if dp[x] != x or cur % x != 0:
      continue

    if x > m:
      flg = False
      break

    cnt[x] = 0
    while cur % x == 0:
      cnt[x] += 1
      cur //= x

  if not flg or cur > 1:
    print("NO")
    continue

  if all(s[x][r + 1] - s[x][l] >= v for x, v in cnt.items()):
    print("Yes")
  else:
    print("NO")
0