[SWEA - D3] 3456. 직사각형 길이 찾기

 

 

나의 코드

T = int(input())
for t in range(1, T + 1):
    arr = list(map(int,input().split()))
    cnt = [0]*101
    result = 0
    for x in arr:
        cnt[x] += 1
    for i in range(len(cnt)):
        if cnt[i] ==1 or cnt[i] ==3 :
            result = i
    print("#"+str(t) , result)

입력받은 수를 카운팅했다.

진짜 멍청하게 풀었다.

 

 

코드 수정 

T = int(input())
for t in range(1, T + 1):
    arr = list(map(int,input().split()))
    arr.sort()
    result = 0
    if arr[0] == arr[1]:
        result = arr[2]
    else:
        result = arr[0]
    print("#"+str(t) , result)

 

입력받은 수를 정렬하고 

0번째 요소와 1번째 요소가 서로 같다면 마지막 수만 혼자인 것이기 때문에 마지막수가 답이 되고

0번째 요소와 1번째 요소가 서로 다르다면 0번째 수가 답이 될 것이다.