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

Lake·2024년 5월 27일

Python 문제

목록 보기
15/22
post-thumbnail

개인정보 수집 유효기간 (2023 KAKAO BLIND RECRUITMENT)

  • 고객의 약관 동의를 얻어서 수집된 1~n번으로 분류되는 개인정보 n개가 있습니다. 약관 종류는 여러 가지 있으며 각 약관마다 개인정보 보관 유효기간이 정해져 있습니다. 당신은 각 개인정보가 어떤 약관으로 수집됐는지 알고 있습니다. 수집된 개인정보는 유효기간 전까지만 보관 가능하며, 유효기간이 지났다면 반드시 파기해야 합니다.

    예를 들어, A라는 약관의 유효기간이 12 달이고, 2021년 1월 5일에 수집된 개인정보가 A약관으로 수집되었다면 해당 개인정보는 2022년 1월 4일까지 보관 가능하며 2022년 1월 5일부터 파기해야 할 개인정보입니다.
    당신은 오늘 날짜로 파기해야 할 개인정보 번호들을 구하려 합니다.

    모든 달은 28일까지 있다고 가정합니다.

    다음은 오늘 날짜가 2022.05.19일 때의 예시입니다.

    약관 종류 유효기간
    A            6 달
    B            12 달
    C            3 달

    번호 개인정보 수집 일자 약관 종류
    1     2021.05.02             A
    2     2021.07.01             B
    3     2022.02.19             C
    4     2022.02.20             C

    • 첫 번째 개인정보는 A약관에 의해 2021년 11월 1일까지 보관 가능하며, 유효기간이 지났으므로 파기해야 할 개인정보입니다.
    • 두 번째 개인정보는 B약관에 의해 2022년 6월 28일까지 보관 가능하며, 유효기간이 지나지 않았으므로 아직 보관 가능합니다.
    • 세 번째 개인정보는 C약관에 의해 2022년 5월 18일까지 보관 가능하며, 유효기간이 지났으므로 파기해야 할 개인정보입니다.
    • 네 번째 개인정보는 C약관에 의해 2022년 5월 19일까지 보관 가능하며, 유효기간이 지나지 않았으므로 아직 보관 가능합니다.

      따라서 파기해야 할 개인정보 번호는 [1, 3]입니다.

      오늘 날짜를 의미하는 문자열 today, 약관의 유효기간을 담은
      1차원 문자열 배열 terms와 수집된 개인정보의 정보를 담은
      1차원 문자열 배열 privacies가 매개변수로 주어집니다.
      이때 파기해야 할 개인정보의 번호를 오름차순으로 1차원 정수
      배열에 담아 return 하도록 solution 함수를 완성해 주세요.

  • 제한사항
    • today는 "YYYY.MM.DD" 형태로 오늘 날짜를 나타냅니다.
    • 1 ≤ terms의 길이 ≤ 20
      • terms의 원소는 "약관 종류 유효기간" 형태의 약관 종류와 유효기간을 공백 하나로 구분한 문자열입니다.
      • 약관 종류는 A~Z중 알파벳 대문자 하나이며, terms 배열에서 약관 종류는 중복되지 않습니다.
      • 유효기간은 개인정보를 보관할 수 있는 달 수를 나타내는 정수이며, 1 이상 100 이하입니다.
    • 1 ≤ privacies의 길이 ≤ 100
      • privacies[i]는 i+1번 개인정보의 수집 일자와 약관 종류를 나타냅니다.
      • privacies의 원소는 "날짜 약관 종류" 형태의 날짜와 약관 종류를 공백 하나로 구분한 문자열입니다
      • 날짜는 "YYYY.MM.DD" 형태의 개인정보가 수집된 날짜를 나타내며, today 이전의 날짜만 주어집니다.
      • privacies의 약관 종류는 항상 terms에 나타난 약관 종류만 주어집니다.
    • today와 privacies에 등장하는 날짜의 YYYY는 연도, MM은 월, DD는 일을 나타내며 점(.) 하나로 구분되어 있습니다.
      • 2000 ≤ YYYY ≤ 2022
      • 1 ≤ MM ≤ 12
      • MM이 한 자릿수인 경우 앞에 0이 붙습니다.
      • 1 ≤ DD ≤ 28
      • DD가 한 자릿수인 경우 앞에 0이 붙습니다.
    • 파기해야 할 개인정보가 하나 이상 존재하는 입력만 주어집니다.

제출한 코드 :

from datetime import datetime, timedelta
from dateutil.relativedelta import relativedelta

def get_date(date):
    return datetime.strptime(date, "%Y.%m.%d")

def solution(today, terms, privacies):
    today = get_date(today)
    answer = []
    tm = {}

    for t in terms:
        term, mon = t.split(" ")
        tm[term] = int(mon)

    for i, p in enumerate(privacies):
        date, term = p.split(" ")
        time = tm[term]
        start_date = get_date(date)
        end_date = start_date + relativedelta(months=time)

        if end_date <= today:
            answer.append(i+1)

    return answer

달리기 경주 (연습문제)

  • 얀에서는 매년 달리기 경주가 열립니다. 해설진들은 선수들이 자기 바로 앞의 선수를 추월할 때 추월한 선수의 이름을 부릅니다. 예를 들어 1등부터 3등까지 "mumu", "soe", "poe" 선수들이 순서대로 달리고 있을 때, 해설진이 "soe"선수를 불렀다면 2등인 "soe" 선수가 1등인 "mumu" 선수를 추월했다는 것입니다. 즉 "soe" 선수가 1등, "mumu" 선수가 2등으로 바뀝니다.

    선수들의 이름이 1등부터 현재 등수 순서대로 담긴 문자열 배열 players와 해설진이 부른 이름을 담은 문자열 배열 callings가 매개변수로 주어질 때, 경주가 끝났을 때 선수들의 이름을 1등부터 등수 순서대로 배열에 담아 return 하는 solution 함수를 완성해주세요.

  • 제한사항

    • 5 ≤ players의 길이 ≤ 50,000
      • players[i]는 i번째 선수의 이름을 의미합니다.
      • players의 원소들은 알파벳 소문자로만 이루어져 있습니다.
      • players에는 중복된 값이 들어가 있지 않습니다.
      • 3 ≤ players[i]의 길이 ≤ 10
    • 2 ≤ callings의 길이 ≤ 1,000,000
      • callings는 players의 원소들로만 이루어져 있습니다.
      • 경주 진행중 1등인 선수의 이름은 불리지 않습니다.

제출한 코드 :

def solution(players, callings):
    # 선수들의 현재 위치를 기록하는 딕셔너리
    position = {player: idx for idx, player in enumerate(players)}

    for call in callings:
        # 추월한 선수의 현재 위치
        current_idx = position[call]

        if current_idx > 0:  # 현재 1등은 추월하지 않기 때문에
            # 앞선 선수의 이름
            prev_player = players[current_idx - 1]

            # 선수들의 위치를 서로 교환
            players[current_idx - 1], players[current_idx] = players[current_idx], players[current_idx - 1]

            # 위치 정보 업데이트
            position[call], position[prev_player] = position[prev_player], position[call]

    return players

공원 산책 (연습문제)

  • 지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.

    • ["방향 거리", "방향 거리" … ]

      예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸
      이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에
      다음 두 가지를 먼저 확인합니다.

  • 주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.

  • 주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.

    위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
    공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.
    공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.

  • 제한사항

    • 3 ≤ park의 길이 ≤ 50
      • 3 ≤ park[i]의 길이 ≤ 50
        • park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
          • S : 시작 지점
          • O : 이동 가능한 통로
          • X : 장애물
      • park는 직사각형 모양입니다.
    • 1 ≤ routes의 길이 ≤ 50
      • routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
      • 로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
      • routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
        • op는 다음 네 가지중 하나로 이루어져 있습니다.
          • N : 북쪽으로 주어진 칸만큼 이동합니다.
          • S : 남쪽으로 주어진 칸만큼 이동합니다.
          • W : 서쪽으로 주어진 칸만큼 이동합니다.
          • E : 동쪽으로 주어진 칸만큼 이동합니다.
        • 1 ≤ n ≤ 9

제출한 코드 :

def solution(park, routes):
    # 공원의 크기
    h = len(park)
    w = len(park[0])

    # 초기 위치 찾기
    for i in range(h):
        for j in range(w):
            if park[i][j] == 'S':
                start_pos = (i, j)
                break
    
    # 방향 설정
    directions = {
        'N': (-1, 0),
        'S': (1, 0),
        'W': (0, -1),
        'E': (0, 1)
    }
    
    current_pos = start_pos

    # 명령 처리
    for route in routes:
        direction, distance = route.split()
        distance = int(distance)
        new_pos = current_pos
        
        # 이동 가능 여부 확인
        for step in range(1, distance + 1):
            next_pos = (new_pos[0] + directions[direction][0], new_pos[1] + directions[direction][1])
            
            # 공원을 벗어나는지 확인
            if not (0 <= next_pos[0] < h and 0 <= next_pos[1] < w):
                break
            
            # 장애물인지 확인
            if park[next_pos[0]][next_pos[1]] == 'X':
                break
            
            new_pos = next_pos
        else:
            # 모두 성공한 경우 현재 위치 업데이트
            current_pos = new_pos

    return [current_pos[0], current_pos[1]]

신고 결과 받기 (2022 KAKAO BLIND RECRUITMENT)

  • 신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
    • 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
      • 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
      • 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
    • k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
      • 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.

       다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고,
       k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.

       유저 ID   유저가 신고한 ID   설명
       "muzi"     "frodo"                "muzi"가 "frodo"를 신고했습니다.
       "apeach"  "frodo"                "apeach"가 "frodo"를
                                             신고했습니다.
       "frodo"     "neo"                  "frodo"가 "neo"를 신고했습니다.
       "muzi"      "neo"                 "muzi"가 "neo"를 신고했습니다.
       "apeach"   "muzi"                "apeach"가 "muzi"를
                                             신고했습니다.

       각 유저별로 신고당한 횟수는 다음과 같습니다.

       유저 ID   신고당한 횟수
       "muzi"     1
       "frodo"    2
       "apeach"  0
       "neo"      2

       위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판
       이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된
       아이디를 정리하면 다음과 같습니다.

       유저 ID   유저가 신고한 ID 정지된 ID
       "muzi"    ["frodo", "neo"]    ["frodo", "neo"]
       "frodo"   ["neo"]               ["neo"]
       "apeach" ["muzi", "frodo"]   ["frodo"]
       "neo"     없음                  없음

       따라서 "muzi"는 처리 결과 메일을 2회, "frodo"와 "apeach"는
       각각 처리 결과 메일을 1회 받게 됩니다.

       이용자의 ID가 담긴 문자열 배열 id_list, 각 이용자가 신고한
       이용자의 ID 정보가 담긴 문자열 배열 report, 정지 기준이
       되는 신고 횟수 k가 매개변수로 주어질 때, 각 유저별로 처리
       결과 메일을 받은 횟수를 배열에 담아 return 하도록 solution
       함수를 완성해주세요.

  • 제한사항
    • 2 ≤ id_list의 길이 ≤ 1,000
      • 1 ≤ id_list의 원소 길이 ≤ 10
      • id_list의 원소는 이용자의 id를 나타내는 문자열이며 알파벳 소문자로만 이루어져 있습니다.
      • id_list에는 같은 아이디가 중복해서 들어있지 않습니다.
    • 1 ≤ report의 길이 ≤ 200,000
      • 3 ≤ report의 원소 길이 ≤ 21
      • report의 원소는 "이용자id 신고한id"형태의 문자열입니다.
      • 예를 들어 "muzi frodo"의 경우 "muzi"가 "frodo"를 신고했다는 의미입니다.
      • id는 알파벳 소문자로만 이루어져 있습니다.
      • 이용자id와 신고한id는 공백(스페이스)하나로 구분되어 있습니다.
      • 자기 자신을 신고하는 경우는 없습니다.
    • 1 ≤ k ≤ 200, k는 자연수입니다.
    • return 하는 배열은 id_list에 담긴 id 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.

제출한 코드 :

def solution(id_list, report, k):
    # 중복 제거된 신고 내역을 저장할 세트
    report_set = set(report)
    
    # 각 유저가 신고당한 횟수를 저장할 딕셔너리
    report_count = {user: 0 for user in id_list}
    
    # 각 유저가 신고한 유저 목록을 저장할 딕셔너리
    user_reports = {user: [] for user in id_list}
    
    # 신고 내역 처리
    for rep in report_set:
        reporter, reported = rep.split()
        report_count[reported] += 1
        user_reports[reporter].append(reported)
    
    # 정지된 유저 리스트
    suspended_users = {user for user, count in report_count.items() if count >= k}
    
    # 결과 메일 횟수 저장 리스트
    result = [0] * len(id_list)
    
    # 각 유저가 받은 메일 수 계산
    for i, user in enumerate(id_list):
        result[i] = len([reported for reported in user_reports[user] if reported in suspended_users])
    
    return result

최댓값과 최솟값 (연습문제)

  • 문자열 s에는 공백으로 구분된 숫자들이 저장되어 있습니다. str에 나타나는 숫자 중 최소값과 최대값을 찾아 이를 "(최소값) (최대값)"형태의 문자열을 반환하는 함수, solution을 완성하세요.
    예를들어 s가 "1 2 3 4"라면 "1 4"를 리턴하고, "-1 -2 -3 -4"라면 "-4 -1"을 리턴하면 됩니다.

    • 제한사항
      • s에는 둘 이상의 정수가 공백으로 구분되어 있습니다.

제출한 코드 :

def solution(s):
    # 문자열을 공백으로 분리하여 리스트로 만듦
    numbers = s.split()
    
    # 리스트의 원소들을 정수로 변환
    numbers = list(map(int, numbers))
    
    # 최소값과 최대값을 찾음
    min_num = min(numbers)
    max_num = max(numbers)
    
    # 형식에 맞게 문자열로 반환
    return f"{min_num} {max_num}"
profile
Lake
이전 포스트

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

다음 포스트

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

0개의 댓글