結果

問題 No.1265 Balloon Survival
コンテスト
ユーザー uni_python
提出日時 2020-11-01 22:55:33
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 1,627 ms / 2,000 ms
コード長 1,146 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 359 ms
コンパイル使用メモリ 84,924 KB
実行使用メモリ 130,016 KB
最終ジャッジ日時 2026-04-03 07:43:57
合計ジャッジ時間 22,047 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge4_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def main():
    import sys
    input=sys.stdin.readline
    def I(): return int(input())
    def MI(): return map(int, input().split())
    def LI(): return list(map(int, input().split()))



    N=I()

    if N==1:
        print(0)
        exit()
    if N==2:
        print(1)
        exit()

    X=[0]*N
    Y=[0]*N
    for i in range(N):
        X[i],Y[i]=MI()

    M=10**3+1
    M2=M*M

    def enc(d,i,j):
        temp=d*M2 + i*M + j
        return temp

    def dec(temp):
        d,ij=divmod(temp,M2)
        i,j=divmod(ij,M)
        return d,i,j




    # 取り除いた or 消えた
    used=[0]*N

    L=[0]*(N*(N-1)//2)
    k=0

    for i in range(N):
        for j in range(i+1,N):
            d=(X[i]-X[j])**2 + (Y[i]-Y[j])**2
            L[k]=enc(d,i,j)
            k+=1

    ans=0
    L.sort()
    S=0

    for aaa in L:
        _,i,j=dec(aaa)
        if i==0:
            if used[j]==0:
                ans+=1
                used[j]=1
                S+=1
        if used[i]==0 and used[j]==0:
            used[i]=1
            used[j]=1
            S+=2
        if S==N-1:
            break
            
    print(ans)

main()
0