문제

여러 개의 시퀀스를 입력받아, 그중 가장 많은 요소를 가진 시퀀스를 반환하는 함수 longest()를 정의해 보라. 다음은 이 함수의 실행 예이다.

>>> longest([1, 2, 3], (4, 5), [], 'abcdefg', range(5))
'abcdefg'

>>> longest('파이썬', '프로그래밍')
'프로그래밍'

>>> longest(range(10), range(100), range(50))
range(0, 100)

답

코드

def longest(*sequences):
    """여러 개의 시퀀스를 전달받아 요소의 개수가 가장 많은 시퀀스를 반환한다."""
    winner = []
    for sequence in sequences:
        if len(winner) < len(sequence):
            winner = sequence
    return winner

# 함수의 동작 확인
print(longest([1, 2, 3], (4, 5), [], 'abcdefg', range(5)))
print(longest('파이썬', '프로그래밍'))
print(longest(range(10), range(100), range(50)))

실행 결과

abcdefg
프로그래밍
range(0, 100)

해설

• 함수에서 임의의 개수의 시퀀스를 인자로 입력받기 위해 패킹 매개변수 *sequences를 정의한다.
• 가장 큰 시퀀스를 기억하기 위한 변수 winner를 정의하고, 초깃값으로 요소의 개수가 0개인 시퀀스를 대입한다.
• for 문으로 sequences에 전달된 시퀀스들을 순회하면서 len() 함수를 이용해 시퀀스의 크기를 비교한다. 현재 시퀀스의 크기가 지금까지 발견한 가장 큰 시퀀스보다 더 큰 경우, 가장 큰 시퀀스를 기억하는 변수에 현재 시퀀스를 대입한다.
• 순회가 끝나면 winner 변수에는 가장 큰 시퀀스가 대입되어 있다. 이 시퀀스를 반환한다.
• 함수에 전달된 시퀀스의 개수가 0개인 경우에 어떻게 해야 하는지는 문제에 정의되어 있지 않다. 따라서 그러한 경우에는 임의로 처리하면 된다. 답안에서는 빈 리스트 []를 반환하고 있는데, None을 반환하게 했거나 오류가 발생하게 두었더라도 틀린 답이 아니다.
• 아래의 예와 같이 max() 함수를 이용하여 요소가 가장 많은 시퀀스를 구하는 것도 좋은 방법이다. max() 함수를 사용할 때 key 매개변수에 비교에 사용할 함수를 인자로 전달할 수 있다. 요소의 개수를 이용해 비교하기 위해서는 len() 함수를 인자로 지정하면 된다.

def longest(*sequences):
    return max(*sequences, key=len)