結果

問題 No.14 最小公倍数ソート
コンテスト
ユーザー mkawa2
提出日時 2020-05-09 00:42:15
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 282 ms / 5,000 ms
コード長 1,889 bytes
コンパイル時間 76 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 29,440 KB
最終ジャッジ日時 2024-07-04 02:00:20
合計ジャッジ時間 5,110 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from heapq import *
import sys

sys.setrecursionlimit(10 ** 6)
int1 = lambda x: int(x) - 1
p2D = lambda x: print(*x, sep="\n")
def II(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def SI(): return sys.stdin.readline()[:-1]

def factors(a):
    res=[]
    for d in range(1,a+1):
        if d**2>a:break
        if a%d==0:
            res.append(d)
            res.append(a//d)
    if res[-1]==res[-2]:res.pop()
    return res

def main():
    n=II()
    aa=LI()
    # 固定された数をA、未固定の数をBとする
    # 最大公約数をGとすると、A=Ga,B=Gbとして最小公倍数L=Gab
    # 各数ごとに、約数fが最大公約数となったとして、
    # そのときのbとインデックスiをヒープのbi[f]に入れる
    # また、各数の約数集合ffもfffに入れる
    bi=[[] for _ in range(100005)]
    fff=[]
    for i,a in enumerate(aa):
        ff=factors(a)
        fff.append(ff)
        for f in ff:
            heappush(bi[f],(a//f,i))
    # print(fff)
    # print(bi)
    ans=[aa[0]]
    fix=0
    used=[False]*n
    used[0]=True
    for _ in range(n-2):
        ff=fff[fix]
        mnb=10**9
        mnf=10**9
        mni=0
        # 最大公約数を決め打ちしてL=Gabのbが最小になる数を探す
        # 最大公約数の候補は固定された数の約数
        for f in ff:
            while bi[f] and used[bi[f][0][1]]:heappop(bi[f])
            if not bi[f]:continue
            b,i=bi[f][0]
            if (b,f)<(mnb,mnf):
                mnb=b
                mnf=f
                mni=i
        ans.append(mnb*mnf)
        used[mni]=True
        fix=mni
    for a,u in zip(aa,used):
        if not u:ans.append(a)
    print(*ans)

main()
0