[BOJ] 정사각형_1485.py

문제

네 점이 주어졌을 때, 네 점을 이용해 정사각형을 만들 수 있는지 없는지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 네 줄로 이루어져 있으며, 점의 좌표가 한 줄에 하나씩 주어진다. 점의 좌표는 -100,000보다 크거나 같고, 100,000보다 작거나 같은 정수이다. 같은 점이 두 번 이상 주어지지 않는다.

출력

각 테스트 케이스마다 입력으로 주어진 네 점을 이용해서 정사각형을 만들 수 있으면 1을, 없으면 0을 출력한다.

 

import math
class Square:
    def __init__(self, point:list):
        self.p0 = point[0]
        self.p1 = point[1]
        self.p2 = point[2]
        self.p3 = point[3]
    def is_square(self):
        stool1 = distance(self.p0, self.p1)
        stool2 = distance(self.p0, self.p2)
        stool3 = distance(self.p2, self.p3)
        stool4 = distance(self.p1, self.p3)
        if stool1 == stool2 and stool1 == stool3 and stool1 == stool4:
            if distance(self.p0, self.p3) == distance(self.p1, self.p2):
                return 1
            else:
                return 0
        else:
            return 0
        
def distance(a:list, b:list):
    D = math.sqrt(pow((b[1]-a[1]), 2) + pow((b[0]-a[0]), 2))
    return D
 
if __name__ == "__main__":
    n = int(input())
    for i in range(n):
        s = []
        for _ in range(4):
            s.append(list(map(int, input().split())))
        s.sort()
        sq = Square(s)
        print(sq.is_square())
 

기말고사 시험치러 가기 직전에 푼 문제

정사각형이 되려면 서로 이웃한 두 변의 길이가 같고 + 두 대각선의 길이도 같으면 정사각형이라고 할 수 있다

사각형의 네 점을 속성으로 가지는 square 클래스를 만들고 해당 인스턴스가 정사각형인지 판별하는 함수도 만들어 넣었다