結果

問題 No.1955 Not Prime
ユーザー titiatitia
提出日時 2022-05-23 02:01:54
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 2,340 bytes
コンパイル時間 231 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 126,172 KB
最終ジャッジ日時 2024-09-20 13:01:06
合計ジャッジ時間 27,162 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 584 ms
53,376 KB
testcase_01 AC 587 ms
53,632 KB
testcase_02 AC 603 ms
53,376 KB
testcase_03 AC 588 ms
53,504 KB
testcase_04 AC 596 ms
53,376 KB
testcase_05 AC 621 ms
53,504 KB
testcase_06 AC 662 ms
53,376 KB
testcase_07 AC 1,182 ms
54,508 KB
testcase_08 AC 1,106 ms
54,256 KB
testcase_09 AC 1,475 ms
54,004 KB
testcase_10 AC 1,424 ms
53,504 KB
testcase_11 TLE -
testcase_12 AC 840 ms
53,504 KB
testcase_13 AC 1,549 ms
54,904 KB
testcase_14 AC 765 ms
53,504 KB
testcase_15 AC 761 ms
53,632 KB
testcase_16 AC 1,006 ms
54,384 KB
testcase_17 AC 1,423 ms
57,204 KB
testcase_18 AC 1,490 ms
54,904 KB
testcase_19 AC 1,555 ms
56,572 KB
testcase_20 AC 622 ms
53,504 KB
testcase_21 AC 900 ms
53,376 KB
testcase_22 AC 591 ms
53,504 KB
testcase_23 AC 605 ms
53,504 KB
testcase_24 AC 582 ms
53,376 KB
testcase_25 AC 618 ms
53,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
import math 
N=int(input())

AB=[tuple(map(int,input().split())) for i in range(N)]

x=10**6+10

L=math.floor(math.sqrt(x)) # 平方根を求める

Primelist=[i for i in range(x+1)]
Primelist[1]=0 # 1は素数でないので0にする.
 
for i in Primelist:
    if i>L:
        break
    if i==0:
        continue
    for j in range(2*i,x+1,i):
        Primelist[j]=0

Primes=[Primelist[j] for j in range(x+1) if Primelist[j]!=0]
Primes=set(Primes)

EDGE=[[] for i in range(2*N)]
EDGE_INV=[[] for i in range(2*N)]

# 0...N-1: x
# N...2N-1: ￢x

for i in range(N):
    a,b=AB[i]

    for j in range(N):
        c,d=AB[j]

        if int(str(a)+str(c)) in Primes:
            # ￢(x ⋀ y)
            EDGE[i].append(j)
            EDGE[j+N].append(i+N)

            EDGE_INV[j].append(i)
            EDGE_INV[i+N].append(j+N)

        if int(str(a)+str(d)) in Primes:
            # ￢(x ⋀ ￢y)
            EDGE[i].append(j+N)
            EDGE[j].append(i+N)

            EDGE_INV[j+N].append(i)
            EDGE_INV[i+N].append(j)


        if int(str(b)+str(c)) in Primes:
            # ￢(￢x ⋀ y)
            EDGE[i+N].append(j)
            EDGE[j+N].append(i)

            EDGE_INV[j].append(i+N)
            EDGE_INV[i].append(j+N)

        if int(str(b)+str(d)) in Primes:
            # ￢(￢x ⋀ ￢y)
            EDGE[i+N].append(j+N)
            EDGE[j].append(i)

            EDGE_INV[j+N].append(i+N)
            EDGE_INV[i].append(j)
            
            
QUE = list(range(2*N))
check=[0]*(2*N)
TOP_SORT=[]
 
def dfs(x):
    if check[x]==1:
        return
    check[x]=1
    
    for to in EDGE[x]:
        if check[to]==0:
            dfs(to)
 
    TOP_SORT.append(x) # 全ての点からDFSを行い, 帰りがけに点を答えに入れる
    check[x]=1
 
while QUE:
    x=QUE.pop()
    dfs(x)
 
USE=[0]*(2*N)
SCC=[]
 
def dfs2(x):
    Q=[x]
    USE[x]=1
    ANS=[]
 
    while Q:
        x=Q.pop()
        ANS.append(x)
        for to in EDGE_INV[x]:
            if USE[to]==0:
                USE[to]=1
                Q.append(to)
    return ANS
 
for x in TOP_SORT[::-1]:
    if USE[x]==0:
        SCC.append(dfs2(x))


for scc in SCC:
    SET=set(scc)

    for x in scc:
        if x+N in SET or x-N in SET:
            print("No")
            exit()

print("Yes")
0