結果

問題 No.854 公平なりんご分配
ユーザー Navier_Boltzmann
提出日時 2022-04-10 16:55:44
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,906 bytes
コンパイル時間 369 ms
コンパイル使用メモリ 82,476 KB
実行使用メモリ 849,748 KB
最終ジャッジ日時 2024-12-14 05:26:02
合計ジャッジ時間 162,551 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 MLE * 1
other AC * 47 TLE * 30 MLE * 15
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys 
input = sys.stdin.buffer.readline
N = int(input())
A = list(map(int,input().split()))
Q = int(input())

from collections import defaultdict,Counter
class prime_factorize():
    
    def __init__(self,M=10**6):
        self.sieve = [-1]*(M+1)
        self.sieve[1] = 1
        self.p = [False]*(M+1)
        for i in range(2,M+1):
            if self.sieve[i] == -1:
                self.p[i] = True
                for j in range(i,M+1,i):
                    self.sieve[j] = i
                    
    def factors(self,x):
        tmp = []
        while self.sieve[x] != x:
            tmp.append(self.sieve[x])
            x //= self.sieve[x]
        tmp.append(self.sieve[x])
        return tmp
        
    def is_prime(self,x):
        return self.p[x]
p = prime_factorize(M=2001)
p_list = []
for i in range(1,2001):
    if p.is_prime(i):
        p_list.append(i)
d = {i:[0]*N for i in p_list}
for i in range(N):
    a = A[i]
    for k,v in Counter(p.factors(a)).items():
        if k == 1:
            continue 
        d[k][i] += v
        
        
    if i !=0:
        for k in p_list:
            d[k][i] += d[k][i-1]


for _ in range(Q):
    P,L,R = map(int,input().split())
    L -= 1
    R -= 1
    dp = {i:0 for i in p_list}
    c_flg = True
    while P != 1 and c_flg:
        flg = True
        for i in p_list:
            if P%i==0:
                dp[i] += 1
                P //= i
                flg = False
                break
        if flg:
            print('NO')
            c_flg = False
    if not c_flg:
        continue
    
    if L != 0:
        if all(d[k][R]-d[k][L-1] >= dp[k] for k in dp.keys()):
            print('Yes')
        else:
            print('NO')
        continue
    else:
        if all(d[k][R] >= dp[k] for k in dp.keys()):
            print('Yes')
        else:
            print('NO')
        continue
        
            
            
0