初心者のプログラミング日記

プログラミング初心者の日記

プログラミングに関することを書いていきます。

ABC133(A,B)

A - T or T

https://atcoder.jp/contests/abc133/tasks/abc133_a
かかった時間 1分
実行時間 24ms

N,A,B=map(int,input().split())
print(min(N*A,B))

B - Good Distance

https://atcoder.jp/contests/abc133/tasks/abc133_b
かかった時間 30分
実行時間 185ms

import math
N,D=map(int,input().split())
List=[]

for i in range(N):
    x=list(map(int,input().split()))
    List.append(x)

ans=0
for i in range(N-1):
    y=List[i]

    for j in range(i+1,N):
        z=List[j]
        sum=0
        for j in range(D):
            sum+=abs(y[j]-z[j])**2
            s=math.sqrt(sum)
            
        if s.is_integer():
            ans+=1
                
print(ans)

数が大きくないので全探索で