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

Lake·2024년 5월 28일

Python 문제

목록 보기
18/22
post-thumbnail

H-Index (정렬)

  • H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과에 따르면, H-Index는 다음과 같이 구합니다.

    어떤 과학자가 발표한 논문 n편 중, h번 이상 인용된 논문이 h편 이상이고 나머지 논문이 h번 이하 인용되었다면 h의 최댓값이 이 과학자의 H-Index입니다.

    어떤 과학자가 발표한 논문의 인용 횟수를 담은 배열 citations가 매개변수로 주어질 때, 이 과학자의 H-Index를 return 하도록 solution 함수를 작성해주세요.

    • 제한사항
      • 과학자가 발표한 논문의 수는 1편 이상 1,000편 이하입니다.
      • 논문별 인용 횟수는 0회 이상 10,000회 이하입니다.

제출한 코드 :

def solution(citations):
    citations.sort(reverse=True)  
    
    h_index = 0
    for i, citation in enumerate(citations):
        if citation >= i + 1:  # 현재 논문의 인용 횟수가 현재 순위보다 크거나 같으면 h-index 갱신
            h_index = i + 1
        else:
            break
    
    return h_index  

n^2 배열 자르기 (월간 코드 챌린지 시즌3)

  • 정수 n, left, right가 주어집니다. 다음 과정을 거쳐서 1차원 배열을 만들고자 합니다.

    • nn열 크기의 비어있는 2차원 배열을 만듭니다.

    • i = 1, 2, 3, ..., n에 대해서, 다음 과정을 반복합니다.

    • 1행 1열부터 ii열까지의 영역 내의 모든 빈 칸을 숫자 i로 채웁니다.

    • 1행, 2행, ..., n행을 잘라내어 모두 이어붙인 새로운 1차원 배열을 만듭니다.

    • 새로운 1차원 배열을 arr이라 할 때, arr[left], arr[left+1], ..., arr[right]만 남기고 나머지는 지웁니다.

    • 정수 n, left, right가 매개변수로 주어집니다. 주어진 과정대로 만들어진 1차원 배열을 return 하도록 solution 함수를 완성해주세요.

    • 제한사항

      • 1 ≤ n ≤ 10^7
      • 0 ≤ leftright < n^2
      • right - left < 10^5

제출한 코드 :

def solution(n, left, right):
    result = []
    for i in range(left, right + 1):
        row = i // n
        col = i % n
        result.append(max(row, col) + 1)
    
    return result

행렬의 곱셈 (연습문제)

  • 2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

    • 제한사항
      • 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
      • 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
      • 곱할 수 있는 배열만 주어집니다.

제출한 코드 :

import numpy as np

def solution(arr1, arr2):
    arr1_np = np.array(arr1)
    arr2_np = np.array(arr2)
    result = np.dot(arr1_np, arr2_np)
    
    return result.tolist()  

할인행사 (연습문제)

  • XYZ 마트는 일정한 금액을 지불하면 10일 동안 회원 자격을 부여합니다. XYZ 마트에서는 회원을 대상으로 매일 한 가지 제품을 할인하는 행사를 합니다. 할인하는 제품은 하루에 하나씩만 구매할 수 있습니다. 알뜰한 정현이는 자신이 원하는 제품과 수량이 할인하는 날짜와 10일 연속으로 일치할 경우에 맞춰서 회원가입을 하려 합니다.

    예를 들어, 정현이가 원하는 제품이 바나나 3개, 사과 2개, 쌀 2개, 돼지고기 2개, 냄비 1개이며, XYZ 마트에서 14일간 회원을 대상으로 할인하는 제품이 날짜 순서대로 치킨, 사과, 사과, 바나나, 쌀, 사과, 돼지고기, 바나나, 돼지고기, 쌀, 냄비, 바나나, 사과, 바나나인 경우에 대해 알아봅시다. 첫째 날부터 열흘 간에는 냄비가 할인하지 않기 때문에 첫째 날에는 회원가입을 하지 않습니다. 둘째 날부터 열흘 간에는 바나나를 원하는 만큼 할인구매할 수 없기 때문에 둘째 날에도 회원가입을 하지 않습니다. 셋째 날, 넷째 날, 다섯째 날부터 각각 열흘은 원하는 제품과 수량이 일치하기 때문에 셋 중 하루에 회원가입을 하려 합니다.

    정현이가 원하는 제품을 나타내는 문자열 배열 want와 정현이가 원하는 제품의 수량을 나타내는 정수 배열 number, XYZ 마트에서 할인하는 제품을 나타내는 문자열 배열 discount가 주어졌을 때, 회원등록시 정현이가 원하는 제품을 모두 할인 받을 수 있는 회원등록 날짜의 총 일수를 return 하는 solution 함수를 완성하시오. 가능한 날이 없으면 0을 return 합니다.

    • 제한사항
      • 1 ≤ want의 길이 = number의 길이 ≤ 10
        • 1 ≤ number의 원소 ≤ 10
        • number[i]want[i]의 수량을 의미하며, number의 원소의 합은 10입니다.
      • 10 ≤ discount의 길이 ≤ 100,000
      • wantdiscount의 원소들은 알파벳 소문자로 이루어진 문자열입니다
        • 1 ≤ want의 원소의 길이, discount의 원소의 길이 ≤ 12

제출한 코드 :

def solution(want, number, discount):
    answer = 0
    dict_wishlist = {}

    for i in range(len(want)):
        dict_wishlist[want[i]] = number[i]

    for i in range(len(discount) - 9):
        dict_tmp = dict_wishlist.copy()
        for j in range(i, i + 10):
            if discount[j] in dict_tmp and dict_tmp[discount[j]] != 0:
                dict_tmp[discount[j]] -= 1
            else:
                break
        if sum(dict_tmp.values()) == 0:
            answer += 1

    return answer  

의상 (해시)

  • 코니는 매일 다른 옷을 조합하여 입는것을 좋아합니다.

    예를 들어 코니가 가진 옷이 아래와 같고, 오늘 코니가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야합니다.

    종류 이름
    얼굴 동그란 안경, 검정 선글라스
    상의 파란색 티셔츠
    하의 청바지
    겉옷 긴 코트

  • 코니는 각 종류별로 최대 1가지 의상만 착용할 수 있습니다. 예를 들어 위 예시의 경우 동그란 안경과 검정 선글라스를 동시에 착용할 수는 없습니다.
  • 착용한 의상의 일부가 겹치더라도, 다른 의상이 겹치지 않거나, 혹은 의상을 추가로 더 착용한 경우에는 서로 다른 방법으로 옷을 착용한 것으로 계산합니다.
  • 코니는 하루에 최소 한 개의 의상은 입습니다.

      코니가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른
      옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

       - 제한사항
          - clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
          - 코니가 가진 의상의 수는 1개 이상 30개 이하입니다.
          - 같은 이름을 가진 의상은 존재하지 않습니다.
          - clothes의 모든 원소는 문자열로 이루어져 있습니다.
          - 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자
            또는 '_' 로만 이루어져 있습니다.

제출한 코드 :

from collections import defaultdict

def solution(clothes):
    clothes_dict = defaultdict(int)
    
    for _, category in clothes:
        clothes_dict[category] += 1
    
    answer = 1
    
    for count in clothes_dict.values():
        answer *= (count + 1)
    
    return answer - 1 # 모든 종류에서 아무 의상도 입지 않은 경우를 제외하고 반환  

0개의 댓글