Python/Python 문제
귤 고르기
서영환
2024. 6. 24. 18:04
- 문제 설명
- 경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 'k'개를 골라 상자 하나에 담아 판매하려고 합니다. 그런데 수확한 귤의 크기가 일정하지 않아 보기에 좋지 않다고 생각한 경화는 귤을 크기별로 분류했을 때 서로 다른 종류의 수를 최소화하고 싶습니다.
- 예를 들어, 경화가 수확한 귤 8개의 크기가 [1, 3, 2, 5, 4, 5, 2, 3] 이라고 합시다. 경화가 귤 6개를 판매하고 싶다면, 크기가 1, 4인 귤을 제외한 여섯 개의 귤을 상자에 담으면, 귤의 크기의 종류가 2, 3, 5로 총 3가지가 되며 이때가 서로 다른 종류가 최소일 때입니다.
- 경화가 한 상자에 담으려는 귤의 개수 k와 귤의 크기를 담은 배열 tangerine이 매개변수로 주어집니다. 경화가 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return 하도록 solution 함수를 작성해주세요.
- 제한 조건
- 1 ≤ k ≤ tangerine의 길이 ≤ 100,000
- 1 ≤ tangerine의 원소 ≤ 10,000,000
- 입출력 예
- 오류코드
- 문제에서 1,4인 귤을 제외 한다고 하여 고정으로 1과 4를 제외하도록 코드를 작성
- 딕셔너리를 이용하여 귤들의 갯수를 카운트
- 이후 갯수를 k만큼 담는데 그 종류를 answer에 저장하여 몇 가지의 귤이 담기는 확인
- 문제에 나와있는 데로 리턴은 하나 제출시 17점이라는 낮은 점수를 받게 되었음
- 질문하기 쪽 확인하니 1,4인 귤을 고정으로 제거 하는게 아닌 낮은 수를 제외하는 방식임을 확인
- 이후 수정 후 제출 하였으나 똑같이 17점이라는 낮은 점수를 받게 됨
#1차
def solution(k, tangerine):
answer = 0
dict_t = {}
for i in tangerine:
if i == 1 or i == 4 : continue
if i in dict_t:
dict_t[i] += 1
else:
dict_t[i] = 1
for name,value in dict_t.items():
k = k - value
answer += 1
if k <= 0: break
return answer
#2차
def solution(k, tangerine):
answer = 0
dict_t = {}
for i in tangerine:
if i not in dict_t: dict_t[i] = 1
else: dict_t[i] += 1
for name,value in dict_t.items():
if value < 2 : continue
k = k - value
answer += 1
if k <= 0: break
return answer
- 수정코드
- 방식을 수정하여 내림차순으로 정렬하여 큰 수부터 k만큼 제외 하고 그 가짓수를 answer에 저장하게 수정
- 이후 정상적으로 통과됨을 확인
def solution(k, tangerine):
answer = 0
dict_t = {}
for i in tangerine:
if i not in dict_t: dict_t[i] = 1
else: dict_t[i] += 1
check=sorted(dict_t.values(),reverse=True)
for value in check:
k = k - value
answer += 1
if k <= 0: break
return answer
- 결론
- 해당 문제는 귤의 크기를 먼저 나누고 크기에 맞게 귤을 센 후 많은 귤을 먼저 빼는 방식으로 해야지만 해당 문제를 해결 할 수 있음
- 경화가 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return 하는 문제라서 귤의 갯수가 많은거 부터 빼야지만 최소값을 구할 수 있고 반대로 다른 종류의 수가 최대 값을 return 하게 하는 문제 였다면 정렬을 오름차순 정령 후 그 것부터 빼는 방식으로 하면 최대값을 구 할 수 있음