체육복 (Programmers 42862)

🧑‍💻 문제

  • 점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.
  • 전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.

제한사항

  • 전체 학생의 수는 2명 이상 30명 이하입니다.
  • 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
  • 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
  • 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
  • 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

🧑‍💻 해결방법

  • 카테고리를 보면 알 수 있듯, 그리디 알고리즘이다.
  • 비교적 간단한 문제

🧑‍💻 탐욕법 (Greedy Algorithm)

  • 탐욕법은 최적의 해를 구하는데 주로 사용된다.
  • 하지만 이 알고리즘은 꼭 적절하다고는 할 수 없다. 그 이유는 오히려 최적의 해를 구하지 못하는 경우가 더 많기 때문이다.

🧑‍💻 실패코드

입력받은 lost 리스트를 직접적으로 건드렸기 때문에 실행 테스트 케이스에서 실패한 것으로 보인다. 또한 시간 복잡도가 n(O^2)가 되기 때문에 좋은 코드가 아니다.

def solution(n, lost, reserve):
    answer = 0
    for res_stu in reserve:
        for lo_stu in lost:
            if (res_stu - 1) == lo_stu or (res_stu + 1) == lo_stu:
                answer += 1
                lost.remove(lo_stu)
    answer = n - len(lost)
    return answer

🧑‍💻 코드

def solution(n, lost, reserve):
    answer = 0

    los = [l for l in lost if l not in reserve]
    res = [r for r in reserve if r not in lost]

    for r in res:
        if r-1 in los:
            los.remove(r-1)
        elif r+1 in los:
            los.remove(r+1)
    answer = n - len(los)

    return answer

🧑‍💻 총평

  • 탐욕법은 그저 들어보기만 했던 알고리즘 이름이었다.
  • 허나 푸는 방법을 보았을 때는 그저 내가 아는 문제풀이 방법이었다.
  • 문제가 간단했으나 나는 시간복잡도의 문제도 있었고, 입력받은 리스트에 직접적으로 가하니 문제가 생겼던 것 같다.
  • 리스트가 들어오면 한 번 재생성 하는 방법도 좋은듯 하다.
  • 고민도 중요하지만 문제를 많이 접해보는 것도 나에게는 필요해보인다.

좋은 웹페이지 즐겨찾기