結果

問題 No.1955 Not Prime
ユーザー rlangevinrlangevin
提出日時 2024-04-15 12:17:59
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 4,357 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 82,372 KB
実行使用メモリ 284,508 KB
最終ジャッジ日時 2024-04-15 12:18:06
合計ジャッジ時間 6,086 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 78 ms
73,252 KB
testcase_01 AC 75 ms
66,356 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 77 ms
67,240 KB
testcase_05 AC 75 ms
66,932 KB
testcase_06 AC 159 ms
85,856 KB
testcase_07 AC 568 ms
109,048 KB
testcase_08 AC 461 ms
101,324 KB
testcase_09 AC 419 ms
99,280 KB
testcase_10 AC 154 ms
84,096 KB
testcase_11 TLE -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from math import sqrt, ceil


def Sieve(n):
    lst = [True] * (n + 1)
    lst[0] = lst[1] = False
    for i in range(2, ceil(sqrt(n)) + 1):
        if lst[i]:
            for j in range(2 * i, n + 1, i):
                lst[j] = False
    return lst

class DirectedGraph():
    def __init__(self, N):
        self.N = N
        self.G = [[] for i in range(N)]
        self.rG = [[] for i in range(N)]
        self.order = []
        self.used1 = [0] * N
        self.used2 = [0] * N
        self.group = [-1] * N
        self.label = 0
        self.seen = [0] * N
        self.Edge = set()

    def add_edge(self, u, v):
        #多重辺は排除する
        if (u, v) not in self.Edge:
            self.G[u].append(v)
            self.rG[v].append(u)
            self.Edge.add((u, v))

    def dfs(self, s):
        stack = [~s, s]
        while stack:
            u = stack.pop()
            if u >= 0:
                if self.used1[u]:
                    continue
                self.used1[u] = 1
                for v in self.G[u]:
                    if self.used1[v]:
                        continue
                    stack.append(~v)
                    stack.append(v)
            else:
                u = ~u
                if self.seen[u]:
                    continue
                self.seen[u]= 1
                self.order.append(u)

    def rdfs(self, s, num):
        stack = [s]
        while stack:
            u = stack.pop()
            if u >= 0:
                self.used2[u] = 1
                self.group[u] = num
                for v in self.rG[u]:
                    if self.used2[v]:
                        continue
                    stack.append(v)

    def scc(self):
        for i in range(self.N):
            if self.used1[i]:
                continue
            self.dfs(i)
        for s in reversed(self.order):
            if self.used2[s]:
                continue
            self.rdfs(s, self.label)
            self.label += 1
        return self.label, self.group

    def construct(self):
        nG = [set() for _ in range(self.label)]
        mem = [[] for i in range(self.label)]
        for s in range(self.N):
            now = self.group[s]
            for u in self.G[s]:
                if now == self.group[u]:
                    continue
                nG[now].add(self.group[u])
            mem[now].append(s)
        return nG, mem


class TwoSAT():
    def __init__(self, N):
        self.N = N
        self.G = DirectedGraph(2 * N)
        
    def add(self, x1, x2, f1, f2):
        if f1 == True and f2 == True:
            # ￢x1∪￢x2
            # (x1⇒￢x2)∩(x2⇒￢x1)
            self.G.add_edge(x1, x2 + self.N)
            self.G.add_edge(x2, x1 + self.N)
            
        if f1 == True and f2 == False:
            # ￢x1∪x2
            # (x1⇒x2)∩(￢x2⇒￢x1)
            self.G.add_edge(x1, x2)
            self.G.add_edge(x2 + self.N, x1 + self.N)
        
        if f1 == False and f2 == True:
            # x1∪￢x2
            # (￢x1⇒￢x2)∩(x2⇒x1)
            self.G.add_edge(x1 + self.N, x2 + self.N)
            self.G.add_edge(x2, x1)
            
        if f1 == False and f2 == False:
            # x1∪x2
            # (￢x1⇒x2)∩(￢x2⇒x1)
            self.G.add_edge(x1 + self.N, x2)
            self.G.add_edge(x2 + self.N, x1)
            
    def check(self):
        _, group = self.G.scc()
        ans = []
        for i in range(self.N):
            if group[i] == group[i + self.N]:
                print("No")
                exit()
            if group[i] > group[i + self.N]:
                ans.append(1)
            else:
                ans.append(0)
        return ans
    
    
N = int(input())
A, B = [None] * N, [None] * N
for i in range(N):
    A[i], B[i] = input().split()
    
D = Sieve(10**6+5)
TS = TwoSAT(N)
for i in range(N):
    for j in range(i + 1, N):
        if D[int(A[i]+B[j])] or D[int(A[j]+B[i])]:
            TS.add(i, j, True, True)
        if D[int(B[i]+B[j])] or D[int(A[j]+A[i])]:
            TS.add(i, j, False, True)
        if D[int(A[i]+A[j])] or D[int(B[j]+B[i])]:
            TS.add(i, j, True, False)
        if D[int(B[i]+A[j])] or D[int(B[j]+A[i])]:
            TS.add(i, j, False, False)
          
if TS.check():
    print("Yes")
0