結果

問題 No.826 連絡網
ユーザー hedwig100
提出日時 2020-04-20 14:31:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,941 ms / 2,000 ms
コード長 948 bytes
コンパイル時間 384 ms
コンパイル使用メモリ 82,328 KB
実行使用メモリ 273,036 KB
最終ジャッジ日時 2024-10-06 08:44:30
合計ジャッジ時間 17,885 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

INF = 10 ** 10
import sys
sys.setrecursionlimit(100000000)
dy = (-1,0,1,0)
dx = (0,1,0,-1)
from  collections import deque

def sieve(n):
    prime = []
    limit = n ** 0.5
    data = [i + 1 for i in range(1,n)]
    while True:
        p = data[0]
        if limit <= p:
            return prime + data
        prime.append(p)
        data = [e for e in data if e%p != 0]

def main():
    n, p = map(int,input().split())
    primes = sieve(n + 1)
    G = [[] for _ in range(n + 1)]
    for q in primes:
        for i in range(2 * q,n + 1,q):
            G[q].append(i)
            G[i].append(q)
    
    stack = []
    stack.append(p)
    visited = [0] * (n + 1)
    visited[p] = 1
    while stack:
        p = stack.pop()
        for v in G[p]:
            if visited[v]:
                continue
            else:
                visited[v] = 1
                stack.append(v)
    
    print(sum(visited))

if __name__ == '__main__':
    main()
0