結果

問題 No.1115 二つの数列 / Two Sequences
ユーザー 👑 Kazun
提出日時 2020-08-13 22:21:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 2,229 bytes
コンパイル時間 360 ms
コンパイル使用メモリ 82,168 KB
実行使用メモリ 102,464 KB
最終ジャッジ日時 2024-10-09 19:35:05
合計ジャッジ時間 5,609 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

class Permutation():
    def __init__(self,n,p=[]):
        if p==[]:
            self.p=[i for i in range(n)]
        else:
            self.p=p

        self.n=n

    def __str__(self):
        return "["+", ".join(map(str,self.p))+"]"

    def __mul__(self,other):
        T=[]
        n=max(self.n,other.n)

        for i in range(n):
            T.append(self.Replace(other.Replace(i)))

        return Permutation(n,T)

    def __truediv__(self,other):
        pass

    def __sgn__(self):
        if self.Minimum_Transposition()%2:
            return -1
        else:
            return 1

    def Inverse(self):
        Q=list(range(self.n))
        for k in range(self.n):
            Q[self.p[k]]=k
        return Permutation(self.n,Q)

    def Inversion(self):
        X=["*"]+[0]*self.n

        A=0
        Y=(self.n*(self.n-1))//2

        for a in self.p:
            s=a
            while 1<=s:
                Y-=X[s]
                s-=s&(-s)

            r=a+1
            while r<=self.n:
                X[r]+=1
                r+=r&(-r)
        return Y


    def Transposition(self,u,v):
        a=self.p.index(u)
        b=self.p.index(v)

        self.p[a]=v
        self.p[b]=u

    def Minimum_Transposition(self):
        X=self.Cycle_Division()

        T=0
        for d in X:
            T+=len(d)-1
        return T

    def Cycle_Multiplication(self,*C):
        X=[self.p.index(k) for k in C]

        N=len(C)
        for i in range(N):
            self.p[X[i]]=C[(i+1)%N]

    def Cycle_Division(self):
        T=[False]*self.n
        k=0
        v=0

        A=[]
        for k in range(self.n):
            if (not T[k]) and self.p[k]!=k:
                v=k
                B=[k]
                while self.p[v]!=k:
                    v=self.p[v]
                    T[v]=True
                    B.append(v)
                A.append(B)
        return A

    def Replace(self,x):
        if x<self.n:
            return self.p[x]
        else:
            return x
#----------------------------------------------
N=int(input())
A=Permutation(N,list(map(lambda x:int(x)-1,input().split())))
B=Permutation(N,list(map(lambda x:int(x)-1,input().split())))
C=B.Inverse()*A
print(C.Inversion())
0