[ baekjoon ] 11652. 카드(정렬 lambda)

백준 11652. 카드 문제 링크

문제

준규는 숫자 카드 N장을 가지고 있다. 숫자 카드에는 정수가 하나 적혀있는데, 적혀있는 수는 -262보다 크거나 같고, 262보다 작거나 같다.
준규가 가지고 있는 카드가 주어졌을 때, 가장 많이 가지고 있는 정수를 구하는 프로그램을 작성하시오. 만약, 가장 많이 가지고 있는 정수가 여러 가지라면, 작은 것을 출력한다.

입력

첫째 줄에 준규가 가지고 있는 숫자 카드의 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개 줄에는 숫자 카드에 적혀있는 정수가 주어진다.

출력

첫째 줄에 준규가 가장 많이 가지고 있는 정수를 출력한다.

import sys
n = int(input())

cards = {}
for i in range(n):
  input = int(sys.stdin.readline())
  if input in cards:
    cards[input] += 1
  else:
    cards[input] = 1
    
arr = sorted(cards.items(), key = lambda arr : (-arr[1], arr[0]))
print(arr[0][0])
arr = sorted(cards.items(), key = lambda arr : (-arr[1], arr[0]))

value중 최대를 고르고, 만약 가장 많이 가지고 있는 정수가 여러 가지이면 key 중 작은 것을 출력하므로 lambda식을 이용할 수 있다.

cf) dictionary

print(cards)
print(cards.items()) # (튜플)리스트
print(sorted(cards.items())) # (튜플)리스트, key 정렬
arr = cards.items()
print(sorted(arr, key = lambda arr : (-arr[1])))
# value을 기준으로 내림차순
print(sorted(arr, key = lambda arr : (-arr[1], arr[0]))) 
# value을 기준으로 내림차순, 2차로 key를 기준으로 오름차순
arr = sorted(arr, key = lambda arr : (-arr[0], arr[1])) 
# key를 기준으로 내림차순, 2차로 value을 기준으로 오름차순
print(sorted(cards)) # key 정렬
print(sorted(cards.values())) # value 정렬

좋은 웹페이지 즐겨찾기