結果

問題 No.2758 RDQ
ユーザー katonyonkokatonyonko
提出日時 2024-05-18 01:00:04
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 641 ms / 2,000 ms
コード長 3,499 bytes
コンパイル時間 667 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 226,904 KB
最終ジャッジ日時 2024-05-18 01:00:17
合計ジャッジ時間 10,323 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 393 ms
104,440 KB
testcase_01 AC 42 ms
55,904 KB
testcase_02 AC 40 ms
57,244 KB
testcase_03 AC 39 ms
56,664 KB
testcase_04 AC 40 ms
56,632 KB
testcase_05 AC 41 ms
55,980 KB
testcase_06 AC 601 ms
222,360 KB
testcase_07 AC 641 ms
226,904 KB
testcase_08 AC 558 ms
221,524 KB
testcase_09 AC 577 ms
221,684 KB
testcase_10 AC 620 ms
225,808 KB
testcase_11 AC 451 ms
108,468 KB
testcase_12 AC 422 ms
106,752 KB
testcase_13 AC 455 ms
108,332 KB
testcase_14 AC 435 ms
107,960 KB
testcase_15 AC 442 ms
108,768 KB
testcase_16 AC 449 ms
107,416 KB
testcase_17 AC 451 ms
108,220 KB
testcase_18 AC 449 ms
108,820 KB
testcase_19 AC 460 ms
108,516 KB
testcase_20 AC 431 ms
108,036 KB
testcase_21 AC 41 ms
56,096 KB
testcase_22 AC 40 ms
55,556 KB
testcase_23 AC 40 ms
56,084 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import io
import sys
from collections import defaultdict, deque, Counter
from itertools import permutations, combinations, accumulate
from heapq import heappush, heappop
sys.setrecursionlimit(10**6)
from bisect import bisect_right, bisect_left
from math import gcd
import math

_INPUT = """\
6
5 3
2 6 7 19 14
1 4 2
3 5 4
1 5 7
5 4
7 6 5 4 3
1 5 2
2 3 3
2 4 4
1 3 5
3 1
1000 1000 100
1 3 10
"""

def gcd(a, b):
  while b: a, b = b, a % b
  return a
def lcm(a, b):
  return a // gcd(a, b) * b
def isPrimeMR(n):
  d = n - 1
  d = d // (d & -d)
  L = [2, 7, 61] if n < 1<<32 else [2, 3, 5, 7, 11, 13, 17] if n < 1<<48 else [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37]
  for a in L:
    t = d
    y = pow(a, t, n)
    if y == 1: continue
    while y != n - 1:
      y = y * y % n
      if y == 1 or t == n - 1: return 0
      t <<= 1
  return 1
def findFactorRho(n):
  m = 1 << n.bit_length() // 8
  for c in range(1, 99):
      f = lambda x: (x * x + c) % n
      y, r, q, g = 2, 1, 1, 1
      while g == 1:
          x = y
          for i in range(r):
              y = f(y)
          k = 0
          while k < r and g == 1:
              ys = y
              for i in range(min(m, r - k)):
                  y = f(y)
                  q = q * abs(x - y) % n
              g = gcd(q, n)
              k += m
          r <<= 1
      if g == n:
          g = 1
          while g == 1:
              ys = f(ys)
              g = gcd(abs(x - ys), n)
      if g < n:
          if isPrimeMR(g): return g
          elif isPrimeMR(n // g): return n // g
          return findFactorRho(g)
def primeFactor(n):
  i = 2
  ret = {}
  rhoFlg = 0
  while i * i <= n:
      k = 0
      while n % i == 0:
          n //= i
          k += 1
      if k: ret[i] = k
      i += i % 2 + (3 if i % 3 == 1 else 1)
      if i == 101 and n >= 2 ** 20:
          while n > 1:
              if isPrimeMR(n):
                  ret[n], n = 1, 1
              else:
                  rhoFlg = 1
                  j = findFactorRho(n)
                  k = 0
                  while n % j == 0:
                      n //= j
                      k += 1
                  ret[j] = k
  if n > 1: ret[n] = 1
  if rhoFlg: ret = {x: ret[x] for x in sorted(ret)}
  return ret
def divisors(N):
  pf = primeFactor(N)
  ret = [1]
  for p in pf:
      ret_prev = ret
      ret = []
      for i in range(pf[p]+1):
          for r in ret_prev:
              ret.append(r * (p ** i))
  return sorted(ret)

def solve(test):
  N,Q=map(int,input().split())
  A=list(map(int,input().split()))
  d=defaultdict(list)
  for i in range(N):
    di=divisors(A[i])
    for j in di:
      d[j].append(i)
  for _ in range(Q):
    L,R,X=map(int,input().split())
    L-=1
    print(bisect_left(d[X],R)-bisect_left(d[X],L))

def random_input():
  from random import randint,shuffle
  N=randint(1,10)
  M=randint(1,N)
  A=list(range(1,M+1))+[randint(1,M) for _ in range(N-M)]
  shuffle(A)
  return (" ".join(map(str, [N,M]))+"\n"+" ".join(map(str, A))+"\n")*3

def simple_solve():
  return []

def main(test):
  if test==0:
    solve(0)
  elif test==1:
    sys.stdin = io.StringIO(_INPUT)
    case_no=int(input())
    for _ in range(case_no):
      solve(0)
  else:
    for i in range(1000):
      sys.stdin = io.StringIO(random_input())
      x=solve(1)
      y=simple_solve()
      if x!=y:
        print(i,x,y)
        print(*[line for line in sys.stdin],sep='')
        break

#0:提出用、1:与えられたテスト用、2:ストレステスト用
main(0)
0