[Softeer] GPT식 숫자 비교 - Python
얼마 전 GPT의 실수 비교 방식이 화제가 된 적이 있었다.
질문) "3.9와 3.11 중에 뭐가 더 커?" / 답변) "3.11이 더 큽니다."
수학 시간에 졸지 않은 사람들은 3.9가 3.11보다 크다고 생각하지만,
GPT의 눈으로 보면 Python 3.9와 Python 3.11 중 후자를 더 크게 보는 학습 데이터가 많아 저렇게 생각할 수 있다.
GPT의 세상에서 3.1은 3보다 크고, 마찬가지로 3.9는 3.2보다 크지만, 3.10은 3.2보다 큰 값으로 처리된다.
구체적으로, 소수점을 기준으로 왼쪽을 수로 읽은 값을 x,
오른쪽을 수로 읽은 값을 y라고 할 때
두 수의 비교가 다음과 같이 이루어진다:
x값이 더 작으면 더 작은 수이다.
x값이 같을 경우 y값이 더 작으면 더 작은 수이다.
소수점이 없는 경우는 같은 수의 소수점이 있는 경우보다 항상 작게 취급된다. (다시 말해, GPT에게 3은 3.0보다 작다.)
N개의 수들이 주어졌을 때, 이를 GPT의 기준에 따라 비내림차순으로 정렬해보자.
제약조건
[문제 제약 조건]
[조건 1] N은 1 이상 1,000 이하이다.
[조건 2] 각 수는 실수 혹은 정수로 표현되고, 0 이상 100 이하이며,
소수점이 없거나 소수점 아래 최대 3자리까지 주어진다.
입력형식
첫 번째 줄에 N이 주어진다.
두 번째 줄부터 N개의 줄에 걸쳐, 각 수가 한 줄에 하나씩 주어진다.
01.23, 3.001과 같이 소수점을 기준으로 각 파트의 0이 아닌 수 이전에 leading zero가 나오는 경우는 주어지지 않는다.
추가로, 3.00, 3.000, 또는 00.1과 같이 소수점을 기준으로 한 파트에 두 개 이상의 0만 주어지는 입력은 없다.
출력형식
첫 번째 줄부터 N개의 줄에 걸쳐, 입력으로 주어진 수들을 GPT의 기준으로 비내림차순으로 정렬한 순서대로 한 줄에 하나씩 출력한다.
문제풀이
본 문제는 처음 보았을 때, 정렬문제라는 것을 알 수 있었다.
간단한 정렬문제 였는데, 해당 문제를 해결하기 위해선 정렬 키 값에 대한 이해가 필요했다.
3과 3.0, 3.01, 3.1, 3.9, 3.10 과 같은 숫자가 있을 때,
3보단 3.0이 큰 수가 될 테고
3.10이 3.9보다 큰 값이 된다.
3.10이라는 수치를 3과 10으로 봐야만 한다.
소수점 앞의 수와 뒤의 수를 독립적으로 비교해서 정렬해주면 된다.
따라서 간단히 split해주고 두 값을 넣은 전체 배열을 만들었다.
여기서, 소수점 뒷 자리가 없는 숫자의 경우 -1을 넣어서 항상 작은 수로 나오게 의도했다.
뿐만 아니라, 길이또한 중요하기 때문에,
이를 정렬 기준에 넣어서 해결했다.
(지금와서 보니 별로 필요없는 것 같다..)
정답코드
import sys
n = int(input())
total = []
for _ in range(n):
num = input().split('.')
if len(num) == 1:
num.append(-1)
for i in range(len(num)):
num[i] = int(num[i])
total.append(num)
total.sort(key = lambda x: (x[0], x[-1], len(str(x))))
for t in total:
if t[-1] == -1:
print(t[0])
else:
print(f"{t[0]}.{t[-1]}")
https://www.softeer.ai/practice/11001
Softeer - 현대자동차그룹 SW인재확보플랫폼
www.softeer.ai