結果

問題 No.854 公平なりんご分配
ユーザー Navier_Boltzmann
提出日時 2022-04-10 17:00:18
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
MLE  
実行時間 -
コード長 2,236 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 600,304 KB
最終ジャッジ日時 2024-12-14 05:39:19
合計ジャッジ時間 229,528 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 23 TLE * 8 MLE * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    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
main()
0