1. 가장 큰 수

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

제한 사항

• numbers의 길이는 1 이상 100,000 이하입니다.
• numbers의 원소는 0 이상 1,000 이하입니다.
• 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

입출력 예

numbers	return
[6, 10, 2]	"6210"
[3, 30, 34, 5, 9]	"9534330"

문제 풀이

def solution(numbers):
    strings = map(str, numbers)
    strings = sorted(strings, key=lambda x: x*4, reverse=True)
    answer = ''.join(strings)
    return str(int(answer))

3과 30을 비교할 때 문자열을 곱해주는 식으로 3333과 30303030을 문자로 비교하면 3333이 먼저오므로 이 점을 이용해 정렬하면 된다.

단 0, 0, 0이 들어온다면 “000”이 답이 되므로 str(int(…))를 씌워준다

시행 착오

def small_sort(a, b):
    if a == b:
        return a
    A = str(a); B = str(b)
    len_A = len(A); len_B = len(B)
    for i in range(min(len_A, len_B)):
        if A[i] > B[i]:
            return b
        elif A[i] < B[i]:
            return a
    if len_A < len_B:
        # 33        3322
        if A[0] > B[len_A]:
            return b
        # 33        3344
        else:
            return a
    else:
        # 3322     33
        if A[len_B] < B[0]:
            return a
        # 3344     33
        else:
            return b

def solution(numbers):
    for i in range(len(numbers)-1):
        for j in range(len(numbers)-1):
            if numbers[j] == small_sort(numbers[j], numbers[j+1]):
                numbers[j], numbers[j+1] = numbers[j+1], numbers[j]
    strings = map(str, numbers)
    answer = ''.join(strings)
    return answer

비교하는 로직 자체는 떠올라서 solution 함수로 만들고 버블소트로 구현했다.

당연히 시간초과였다.

로직 자체는 33과 3344가 있을 때 이어붙이면 33,3344와 3344,33이 되기 때문에 3344가 먼저와야한다.

이 때 비교할 곳은 두 숫자의 첫번째 자리수와, (짧은 숫자의 자리수 + 1)번째 자리수를 비교하면 된다.

인덱스상으론 0 과 짧은 숫자의 길이이므로 그런식으로 구현해봤지만 로직에도 문제가 있는 것 같고, 버블소트라 시간 초과도 있었다.

2. H_index

문제 설명

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과1에 따르면, H-Index는 다음과 같이 구합니다.

어떤 과학자가 발표한 논문 n편 중, h번 이상 인용된 논문이 h편 이상이고 나머지 논문이 h번 이하 인용되었다면 h의 최댓값이 이 과학자의 H-Index입니다.

어떤 과학자가 발표한 논문의 인용 횟수를 담은 배열 citations가 매개변수로 주어질 때, 이 과학자의 H-Index를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 과학자가 발표한 논문의 수는 1편 이상 1,000편 이하입니다.
• 논문별 인용 횟수는 0회 이상 10,000회 이하입니다.

입출력 예

citations	return
[3, 0, 6, 1, 5]	3

입출력 예 설명

이 과학자가 발표한 논문의 수는 5편이고, 그중 3편의 논문은 3회 이상 인용되었습니다. 그리고 나머지 2편의 논문은 3회 이하 인용되었기 때문에 이 과학자의 H-Index는 3입니다.

문제풀이

def solution(citations):
    citations = sorted(citations)
    H_index = 0
    for i in range(len(citations)):
        if citations[i] <= len(citations) - i:
            H_index = citations[i]
        else:
            H_index = max(H_index, len(citations) - i)
    return H_index

풀이 방법은 먼저 배열을 정렬시켜놓는다.

만약 citations 값이 본인 포함 이후 citations 값 개수 이하라면

citations값이 h라고 할 때 h번 이상 인용된 논문이 h편 이상이라는 뜻이므로 H_index에 갱신시킨다.

근데 citations가 [10, 11, 12]라면 H_index가 0으로 끝날 수 있다.

그러므로 위 조건에 해당되지 않는다면 해당 자리를 기존 H_index와 비교해서 갱신시키면 된다.

시행착오

def solution(citations):
    citations = sorted(citations)
    H_index = 0
    last_index = 0
    for i in range(len(citations)):
        if citations[i] <= len(citations) - i:
            H_index = citations[i]
        else:
            last_index = i
            break

    start = H_index + 1
    for i in range(start, last_index):
        if i <= len(citations) - last_index:
            H_index = i
    return H_index

마지막으로 H_index로 잡히는 citations 값과 그 이후 citations 값을 알아낸 뒤 그 사이 숫자를 하나하나 대조해보는 방식을 사용하려 했으나 이 방법은 [10, 11, 12]와 같은 케이스를 잡아내지 못한다.

GPT 픽 파이썬 모범답안

def solution(citations):
    citations.sort(reverse=True)  # 인용 수 내림차순
    h = 0
    for i, c in enumerate(citations, start=1):  # i = 논문 개수
        if c >= i:
            h = i
    return h

enumerate를 사용해 아름답게 푼 모습이다.

앞으로 enumerate를 좀 애용해야겠다.