[Python] 알고리즘 연습 13 (프로그래머스)

Lake·2024년 5월 23일

Python 문제

목록 보기
13/22
post-thumbnail

로또의 최고 순위와 최저 순위 (2021 Dev-Matching: 웹 백엔드 개발자(상반기))

  • 로또 6/45(이하 '로또'로 표기)는 1부터 45까지의 숫자 중 6개를 찍어서 맞히는 대표적인 복권입니다. 아래는 로또의 순위를 정하는 방식입니다. (실제로 사용되는 로또 순위의 결정 방식과는 약간 다르지만, 이 문제에서는 지문에 명시된 대로 로또 순위를 결정하도록 합니다. )

순위 당첨 내용
  1   6개 번호가 모두 일치
  2   5개 번호가 일치
  3   4개 번호가 일치
  4   3개 번호가 일치
  5   2개 번호가 일치
  6  (낙첨) 그 외

로또를 구매한 민우는 당첨 번호 발표일을 학수고대하고 있었습니다. 하지만, 민우의 동생이 로또에 낙서를 하여, 일부 번호를 알아볼 수 없게 되었습니다. 당첨 번호 발표 후, 민우는 자신이 구매했던 로또로 당첨이 가능했던 최고 순위와 최저 순위를 알아보고 싶어 졌습니다.
알아볼 수 없는 번호를 0으로 표기하기로 하고, 민우가 구매한 로또 번호 6개가 44, 1, 0, 0, 31 25라고 가정해보겠습니다. 당첨 번호 6개가 31, 10, 45, 1, 6, 19라면, 당첨 가능한 최고 순위와 최저 순위의 한 예는 아래와 같습니다.

당첨 번호        31  10    45 1   6   19       결과
최고 순위 번호 31 0→10 44 1 0→6 25 4개 번호 일치, 3등
최저 순위 번호 31 0→11 44 1 0→7 25 2개 번호 일치, 5등

  • 순서와 상관없이, 구매한 로또에 당첨 번호와 일치하는 번호가 있으면 맞힌 걸로 인정됩니다.
  • 알아볼 수 없는 두 개의 번호를 각각 10, 6이라고 가정하면 3등에 당첨될 수 있습니다.
    • 3등을 만드는 다른 방법들도 존재합니다. 하지만, 2등 이상으로 만드는 것은 불가능합니다.
  • 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다.
    • 5등을 만드는 다른 방법들도 존재합니다. 하지만, 6등(낙첨)으로 만드는 것은 불가능합니다.

민우가 구매한 로또 번호를 담은 배열 lottos, 당첨 번호를 담은 배열 win_nums가 매개변수로 주어집니다. 이때, 당첨 가능한 최고 순위와 최저 순위를 차례대로 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

  • 제한사항
    • lottos는 길이 6인 정수 배열입니다.
    • lottos의 모든 원소는 0 이상 45 이하인 정수입니다.
      • 0은 알아볼 수 없는 숫자를 의미합니다.
      • 0을 제외한 다른 숫자들은 lottos에 2개 이상 담겨있지 않습니다.
      • ottos의 원소들은 정렬되어 있지 않을 수도 있습니다.
    • win_nums은 길이 6인 정수 배열입니다.
    • win_nums의 모든 원소는 1 이상 45 이하인 정수입니다
      • win_nums에는 같은 숫자가 2개 이상 담겨있지 않습니다.
      • win_nums의 원소들은 정렬되어 있지 않을 수도 있습니다.

제출한 코드 :

def solution(lottos, win_nums):
    # 당첨 번호와 구매한 로또 번호를 집합으로 변환
    win_nums_set = set(win_nums)
    lottos_set = set(lottos)
    
    # 알아볼 수 없는 번호의 개수 세기
    unknown_count = lottos.count(0)
    
    # 최고 순위: 알아볼 수 없는 번호를 모두 맞았다고 가정하여 계산
    # 최저 순위: 알아볼 수 없는 번호를 모두 맞추지 못했다고 가정하여 계산
    max_match = len(win_nums_set.intersection(lottos_set)) + unknown_count
    min_match = len(win_nums_set.intersection(lottos_set))
    
    # 등수 계산
    max_rank = min(7 - max_match, 6)
    min_rank = min(7 - min_match, 6)
    
    return [max_rank, min_rank]

옹알이(2) (연습문제)

  • 머쓱이는 태어난 지 11개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음과 네 가지 발음을 조합해서 만들 수 있는 발음밖에 하지 못하고 연속해서 같은 발음을 하는 것을 어려워합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.

  • 제한사항

    • 1 ≤ babbling의 길이 ≤ 100
    • 1 ≤ babbling[i]의 길이 ≤ 30
    • 문자열은 알파벳 소문자로만 이루어져 있습니다.

제출한 코드 :

def solution(babbling):
    answer = 0
    init = {"aya", "ye", "woo", "ma"}  

    for word in babbling:
        comp = ""  # 비교할 단어
        temp = ""  # 임시 저장할 단어

        for char in word:
            comp += char

            # 연속해서 같은 단어가 나왔을 때 반복문 탈출
            if comp == temp:
                break

            # 4가지 단어가 포함 돼있을 때 현재 단어 임시 저장 후 비교단어 초기화
            if comp in init:
                temp = comp
                comp = ""

        # 비교할 단어가 존재 하지 않을때 카운트
        if comp == "":
            answer += 1

    return answer

숫자 짝꿍 (연습문제)

  • 두 정수 X, Y의 임의의 자리에서 공통으로 나타나는 정수 k(0 ≤ k ≤ 9)들을 이용하여 만들 수 있는 가장 큰 정수를 두 수의 짝꿍이라 합니다(단, 공통으로 나타나는 정수 중 서로 짝지을 수 있는 숫자만 사용합니다). X, Y의 짝꿍이 존재하지 않으면, 짝꿍은 -1입니다. X, Y의 짝꿍이 0으로만 구성되어 있다면, 짝꿍은 0입니다.

    예를 들어, X = 3403이고 Y = 13203이라면, XY의 짝꿍은 XY에서 공통으로 나타나는 3, 0, 3으로 만들 수 있는 가장 큰 정수인 330입니다. 다른 예시로 X = 5525이고 Y = 1255이면 XY의 짝꿍은 XY에서 공통으로 나타나는 2, 5, 5로 만들 수 있는 가장 큰 정수인 552입니다(X개, Y에는 5가 2개 나타나므로 남는 5 한 개는 짝 지을 수 없습니다.)
    두 정수 X, Y가 주어졌을 때, X, Y의 짝꿍을 return하는 solution 함수를 완성해주세요.

  • 제한사항

    • 3 ≤ X, Y의 길이(자릿수) ≤ 3,000,000입니다.
    • X, Y는 0으로 시작하지 않습니다.
    • X, Y의 짝꿍은 상당히 큰 정수일 수 있으므로, 문자열로 반환합니다.

제출한 코드 :

from collections import Counter

def solution(X, Y):
    answer = ""

    z = Counter(X) & Counter(Y)
    z = sorted([[k,v,k*v] for k,v in z.items()], key=lambda x: x[0], reverse=True)

    if not z:
        return "-1"

    answer = "".join(str(item[2]) for item in z).lstrip('0')

    if not answer:
        return "0"

    return answer

체육복 (탐욕법(Greedy))

  • 점심시간에 도둑이 들어, 일부 학생이 체육복을 도난당했습니다. 다행히 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 합니다. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있습니다. 예를 들어, 4번 학생은 3번 학생이나 5번 학생에게만 체육복을 빌려줄 수 있습니다. 체육복이 없으면 수업을 들을 수 없기 때문에 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 들어야 합니다.

    전체 학생의 수 n, 체육복을 도난당한 학생들의 번호가 담긴 배열 lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve가 매개변수로 주어질 때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 함수를 작성해주세요.

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

제출한 코드 :

def solution(n, lost, reserve):
    answer = 0
    students = [1] * n  # 체육복이 있는 학생들을 1로 초기화

    # 체육복을 도난당한 학생들은 0으로 표시
    for l in lost:
        students[l - 1] -= 1

    # 여벌의 체육복을 가진 학생들은 2로 표시
    for r in reserve:
        students[r - 1] += 1

    # 체육복 빌려주기
    for i in range(n):
        if students[i] == 0:  # 체육복이 없는 경우
            if i > 0 and students[i - 1] == 2:  # 바로 앞번호의 학생이 여벌의 체육복을 가진 경우
                students[i] = 1
                students[i - 1] = 1
            elif i < n - 1 and students[i + 1] == 2:  # 바로 뒷번호의 학생이 여벌의 체육복을 가진 경우
                students[i] = 1
                students[i + 1] = 1

    # 체육수업을 들을 수 있는 학생 수 계산
    for s in students:
        if s > 0:
            answer += 1

    return answer

문자열 나누기 (연습문제)

  • 문자열 s가 입력되었을 때 다음 규칙을 따라서 이 문자열을 여러 문자열로 분해하려고 합니다.
    • 먼저 첫 글자를 읽습니다. 이 글자를 x라고 합시다.
    • 이제 이 문자열을 왼쪽에서 오른쪽으로 읽어나가면서, x와 x가 아닌 다른 글자들이 나온 횟수를 각각 셉니다. 처음으로 두 횟수가 같아지는 순간 멈추고, 지금까지 읽은 문자열을 분리합니다.
    • s에서 분리한 문자열을 빼고 남은 부분에 대해서 이 과정을 반복합니다. 남은 부분이 없다면 종료합니다.
    • 만약 두 횟수가 다른 상태에서 더 이상 읽을 글자가 없다면, 역시 지금까지 읽은 문자열을 분리하고, 종료합니다.

      문자열 s가 매개변수로 주어질 때, 위 과정과 같이 문자열들로 분해하고, 분해한 문자열의 개수를 return 하는 함수 solution을 완성하세요.
  • 제한사항
    • 1 ≤ s의 길이 ≤ 10,000
    • s는 영어 소문자로만 이루어져 있습니다.

제출한 코드 :

def solution(s):
    answer = 0
    idx = 0
    
    while idx < len(s):
        count_x = 1
        count_not_x = 0
        x = s[idx]
        for i in range(idx + 1, len(s)):
            if s[i] == x:
                count_x += 1
            else:
                count_not_x += 1
            if count_x == count_not_x:
                answer += 1
                idx = i + 1
                break
        else:
            answer += 1
            break

    return answer

0개의 댓글