[Python] 탐욕법(Greedy) : 큰 수 만들기

https://programmers.co.kr/learn/courses/30/lessons/42883

코딩테스트 연습 - 큰 수 만들기

 

 

 

💡 문제

---

문자열 형식으로 숫자 number와 제거할 수의 개수 k가 solution 함수의 매개변수로 주어진다. number에서 k개의 수를 제거했을 때 만들 수 있는 수 중 가장 큰 숫자를 문자열 형태로 return 하시오.

 

 

[제한 조건]

• number는 2자리 이상, 1,000,000자리 이하인 숫자입니다.
• k는 1 이상 number의 자릿수 미만인 자연수입니다.

 

 

 

 

※ 문제 푸는 방법 ※

1. 주어진 number 배열에서 하나씩 꺼내 배열에 담는다.

2. 이때, 지금 담으려는 것보다 배열에 담긴 수가 작다면, 또 k=0이 아니라면  배열에서 해당 수를 빼낸다.

 

 

 

[Python Code 👩🏻‍💻]

CODE REVIEW]

1. number를 차례로 담을 배열 collected을 생성한다.

2. enumerate() 함수를 사용하여 해당 수와 인덱스를 기준으로 for문

3. collected 배열에 수가 있고, collected 배열 내 마지막 원소가 현재 수(=num)보다 작고, k=0이 아니라면 (아직 수를 뺄 수 있다면) collected에서 수를 빼내고, k를 1 감소시킨다.

4. number을 다 돌지 않았는데 더 이상 수를 뺄 수 없다면 i에서부터 끝까지 collected 배열에 추가시키고 for문을 빠져나온다.

5. collected에 수를 하나씩 담는다. (조건에 해당할 때만 수를 빼거나 붙이고, 수는 계속 추가해줘야함)

6. 만약 number을 다 돌았음에도 k=0이 아니라면, collected를 collected의 k번째 전까지로 한다.

7. 합친다.