結果

問題 No.1955 Not Prime
ユーザー KazunKazun
提出日時 2022-05-21 13:08:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,686 ms / 2,000 ms
コード長 5,482 bytes
コンパイル時間 293 ms
コンパイル使用メモリ 82,584 KB
実行使用メモリ 229,688 KB
最終ジャッジ日時 2024-09-20 11:37:18
合計ジャッジ時間 15,684 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 54 ms
67,200 KB
testcase_01 AC 53 ms
67,200 KB
testcase_02 AC 55 ms
67,072 KB
testcase_03 AC 53 ms
67,072 KB
testcase_04 AC 54 ms
67,072 KB
testcase_05 AC 54 ms
67,072 KB
testcase_06 AC 216 ms
86,660 KB
testcase_07 AC 1,477 ms
121,956 KB
testcase_08 AC 1,217 ms
115,988 KB
testcase_09 AC 1,197 ms
114,836 KB
testcase_10 AC 97 ms
84,224 KB
testcase_11 AC 510 ms
229,688 KB
testcase_12 AC 744 ms
99,280 KB
testcase_13 AC 1,686 ms
129,724 KB
testcase_14 AC 567 ms
96,676 KB
testcase_15 AC 533 ms
95,208 KB
testcase_16 AC 803 ms
107,888 KB
testcase_17 AC 1,569 ms
133,072 KB
testcase_18 AC 1,248 ms
117,400 KB
testcase_19 AC 1,114 ms
136,492 KB
testcase_20 AC 58 ms
69,120 KB
testcase_21 AC 841 ms
102,428 KB
testcase_22 AC 56 ms
67,712 KB
testcase_23 AC 54 ms
67,200 KB
testcase_24 AC 55 ms
67,328 KB
testcase_25 AC 53 ms
67,200 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Two_SAT:
    def __init__(self,N=0):
        """ N 変数の 2-SAT を定義する.

        N: int
        """

        self.N=N
        self.var_num=N
        self.__cost=2*N

        self.arc=[[] for _ in range(2*N)]
        self.rev=[[] for _ in range(2*N)]

    def cost(self):
        return self.__cost

    def var_to_index(self,v):
        if v>=0:
            return 2*v
        else:
            return 2*(-v-1)+1

    def index_to_var(self,i):
        if i%2:
            return -(i+1)//2
        else:
            return i//2

    def add_variable(self,k=1):
        """ 新たに変数 k 個を加える.
        """

        m=self.var_num
        self.var_num+=k
        self.__cost+=2*k

        self.arc+=[[] for _ in range(2*k)]
        self.rev+=[[] for _ in range(2*k)]

        return list(range(m,m+k))

    def __add_clause(self,i,j):
        self.__cost+=1
        self.arc[self.var_to_index(i)].append(self.var_to_index(j))
        self.rev[self.var_to_index(j)].append(self.var_to_index(i))

    def add_imply(self,i,j):
        """ X_i -> X_j を加える.
        """
        self.__add_clause(i,j)
        self.__add_clause(~j,~i)

    def add_or(self,i,j):
        """ X_i or X_j を加える.
        """

        self.add_imply(~i,j)

    def add_nand(self,i,j):
        """ not (X_i and X_j) を加える.
        """

        self.add_imply(i,~j)

    def add_equivalent(self,*I):
        """ I=[i_0, ..., i_{k-1}] に対して, X_{i_0}=...=X_{i_{k-1}} を追加する.
        """

        k=len(I)

        if k<=1:
            return

        for j in range(k-1):
            self.add_imply(I[j],I[j+1])
        self.add_imply(I[-1],I[0])

    def add_not_equal(self,i,j):
        """ X_i != X_j を追加する.
        """
        self.add_equal(i,~j)

    def set_true(self,i):
        """ 変数 X_i を True にする.
        """

        self.__add_clause(~i,i)

    def set_false(self,i):
        """ 変数 X_i を False にする.
        """

        self.__add_clause(i,~i)

    def at_most_one(self,*I):
        """ X_i (i in I) を満たすような i は高々1つだけという条件を追加する.
        """

        k=len(I)

        if k<=1:
            return

        A=self.add_variable(k)

        self.add_imply(I[0],A[0])
        for i in range(1,k):
            self.add_imply(A[i-1],A[i])
            self.add_imply(I[i],A[i])
            self.add_nand(A[i-1],I[i])

    def is_satisfy(self,Mode=0):
        """ Two-SAT は充足可能?

        Mode:
        0 (Defalt): 充足可能?
        1: 充足可能ならば,その変数の割当を与える (不可能なときはNone).
        2: 充足不能の原因である変数を全て挙げる.
        """
        N=self.var_num
        Group=[0]*(2*N)
        Order=[]

        for s in range(2*N):
            if Group[s]:continue

            S=[s]
            Group[s]=-1

            while S:
                u=S.pop()
                for v in self.arc[u]:
                    if Group[v]:continue
                    Group[v]=-1
                    S.append(u);S.append(v)
                    break
                else:
                    Order.append(u)

        K=0
        for s in Order[::-1]:
            if Group[s]!=-1:continue

            S=[s]
            Group[s]=K

            while S:
                u=S.pop()
                for v in self.rev[u]:
                    if Group[v]!=-1:continue

                    Group[v]=K
                    S.append(v)
            K+=1

        if Mode==0:
            for i in range(N):
                if Group[2*i]==Group[2*i+1]:
                    return False
            return True
        elif Mode==1:
            T=[0]*N
            for i in range(N):
                if Group[2*i]>Group[2*i+1]:
                    T[i]=1
                elif Group[2*i]==Group[2*i+1]:
                    return None
            return T
        elif Mode==2:
            return [i for i in range(self.var_num) if Group[2*i]==Group[2*i+1]]

    def solve(self):
        return self.is_satisfy(1)

def Sieve_of_Eratosthenes(N):
    """ N までのエラトステネスの篩を実行

    [Input]
    N:自然数

    [Output]
    素数かどうかのリスト ([0,0,1,1,0,1,...])
    """

    if N==0:
        return [0]

    T=[1]*(N+1)
    T[0]=T[1]=0

    for x in range(4,N+1,2): T[x]=0
    for x in range(9,N+1,3): T[x]=0

    a=5
    Flag=0
    while a*a<=N:
        if T[a]:
            b=a*a
            c=2*a
            while b<=N:
                T[b]=0
                b+=c
        a+=2+2*Flag
        Flag^=1
    return T

#==================================================
def f(a,b):
    if b<10:
        x=10*a+b
    elif b<100:
        x=100*a+b
    else:
        x=1000*a+b
    return P[x]

#==================================================
N=int(input())

A=[0]*N; B=[0]*N
for i in range(N):
    A[i],B[i]=map(int,input().split())

P=Sieve_of_Eratosthenes(10**6)

T=Two_SAT(N)
for i in range(N):
    for j in range(N):
        if f(A[i],B[i]) or f(A[i],B[j]) or f(A[j],B[i]) or f(A[j],B[j]):
            T.add_nand(i,j)

        if f(A[i],B[i]) or f(A[i],A[j]) or f(B[j],B[i]) or f(B[j],A[j]):
            T.add_nand(i,~j)

        if f(B[i],A[i]) or f(B[i],B[j]) or f(A[j],A[i]) or f(A[j],B[j]):
            T.add_nand(~i,j)

        if f(B[i],A[i]) or f(B[i],A[j]) or f(B[j],A[i]) or f(B[j],A[j]):
            T.add_nand(~i,~j)

print("Yes" if T.is_satisfy() else "No")
0