공부블로그

TODAY TIL

안녕하세요! 이번 포스팅에서는 폰켓몬 선택과 관련된 흥미로운 코딩 문제를 다뤄보겠습니다. 단계적으로 설명하고 코드 구현을 공유합니다.

문제 설명

여러분은 오랜 여행 끝에 홍 박사님의 연구실에 도착했습니다. 홍 박사님은 여러분에게 연구실에 있는 N마리의 폰켓몬 중 N/2마리를 선택해도 좋다고 했습니다. 같은 종류의 폰켓몬은 같은 번호를 가지며, 여러분은 최대한 다양한 종류의 폰켓몬을 선택하고 싶습니다.

문제의 목표: N마리의 폰켓몬 중 최대 N/2마리를 선택할 때, 가장 다양한 종류의 폰켓몬을 선택하는 경우를 찾고, 그때의 폰켓몬 종류의 수를 반환하는 것입니다.

제한사항

• nums는 폰켓몬의 종류 번호가 담긴 1차원 배열입니다.
• N은 항상 짝수이며, 1 이상 10,000 이하의 자연수입니다.
• 폰켓몬의 종류 번호는 1 이상 200,000 이하의 자연수입니다.

단계별 해결 방법

1. 입력 분석:
   • nums 배열에는 각 폰켓몬의 종류 번호가 포함되어 있습니다.
   • 배열의 길이 N은 항상 짝수입니다. 따라서 N/2마리의 폰켓몬을 선택할 수 있습니다.
2. 중복 제거:
   • set 자료형을 사용하여 배열에서 중복을 제거하고 고유한 폰켓몬 종류의 수를 구합니다.
3. 최대 선택 가능한 폰켓몬 수 계산:
   • 선택할 수 있는 최대 종류의 개수는 min(len(unique_pokemon), N // 2)입니다.
     • len(unique_pokemon)은 중복 제거 후 고유한 종류의 개수입니다.
     • N // 2는 선택할 수 있는 폰켓몬의 총 수입니다.
4. 코드 구현

def solution(nums):
    # 중복 제거 후 고유한 폰켓몬 종류의 수 계산
    unique_pokemon = set(nums)
    
    # 선택할 수 있는 폰켓몬 종류의 최댓값 계산
    return min(len(unique_pokemon), len(nums) // 2)

코드 설명

• set(nums)를 사용하여 nums 배열에서 중복된 종류 번호를 제거합니다.
• min(len(unique_pokemon), len(nums) // 2)를 사용해 선택할 수 있는 폰켓몬 종류의 최대 개수를 계산합니다.
  • 고유한 폰켓몬 종류의 수(len(unique_pokemon))가 선택 가능한 수(N/2)보다 작으면 모든 고유 종류를 선택할 수 있습니다.
  • 고유한 폰켓몬 종류의 수가 N/2보다 크다면 N/2만큼의 고유 종류만 선택할 수 있습니다.

예제

입력: nums = [3, 1, 2, 3]

1. set(nums) 결과: {1, 2, 3} (고유한 폰켓몬 종류: 3개)
2. 선택할 수 있는 최대 개수: len(nums) // 2 = 2
3. 반환값: min(3, 2) = 2

결과: 최대 2종류의 폰켓몬을 선택할 수 있습니다.

결론

이 문제는 배열에서 중복된 항목을 제거하고 선택할 수 있는 폰켓몬의 종류를 비교하는 간단한 문제였습니다. 중복 제거를 위한 set 사용과 최소값 계산을 통해 다양한 경우를 쉽게 처리할 수 있었습니다.

이제 여러분도 다양한 알고리즘 문제를 해결할 때 중복 제거와 최소값 계산을 활용해 보세요!

TODAY TIL

안녕하세요! 이번 포스팅에서는 비밀번호 찾기 문제 와 관련된 흥미로운 코딩 문제를 다뤄보겠습니다. 단계적으로 설명하고 코드 구현을 공유합니다.

문제 설명

창영이는 민균이의 컴퓨터에서 비밀번호가 포함된 텍스트 파일을 얻었습니다. 이 파일에는 여러 단어가 줄마다 적혀있으며, 민균이의 비밀번호는 특정 조건을 만족해야 합니다:

1. 목록에 포함된 단어의 길이는 모두 홀수입니다.
2. 비밀번호를 뒤집은 형태의 문자열도 목록에 포함되어 있습니다.

목표는 목록에서 비밀번호를 찾아 비밀번호의 길이와 가운데 글자를 출력하는 것입니다.

문제 해결을 위한 단계별 접근 방법

1단계: 입력받기

• 첫째 줄에 단어의 개수 NNN이 주어집니다.
• 이후 NNN개의 단어를 입력받아 words 리스트에 저장합니다.

# 입력 받기
N = int(input())
words = [input().strip() for _ in range(N)]

2단계: 단어와 뒤집은 단어 비교

• 모든 단어에 대해 해당 단어의 뒤집힌 형태가 리스트에 있는지 확인합니다.
  • 예를 들어, 단어가 "tulipan"이라면, 뒤집은 형태는 "napilut"입니다.
• 이를 위해 for 루프를 사용해 words 리스트의 각 단어를 순회하며 뒤집힌 단어가 words에 있는지 검사합니다.

# 비밀번호 찾기
password = ""
for word in words:
    reversed_word = word[::-1]  # 단어 뒤집기
    if reversed_word in words:  # 뒤집은 단어가 목록에 있는지 확인
        password = word
        break

3단계: 비밀번호 길이와 가운데 글자 찾기

• 비밀번호를 찾았다면:
  • 길이는 len(password)로 구할 수 있습니다.
  • 가운데 글자는 (길이 // 2) 인덱스를 통해 찾습니다.

# 비밀번호 길이와 가운데 글자 찾기
length = len(password)
middle_char = password[length // 2]

4단계: 결과 출력

• 비밀번호의 길이와 가운데 글자를 한 줄에 출력합니다.

# 결과 출력
print(length, middle_char)

전체 코드

아래는 전체 코드를 정리한 것입니다.

# 1. 입력 받기
N = int(input())
words = [input().strip() for _ in range(N)]

# 2. 비밀번호 찾기
password = ""
for word in words:
    reversed_word = word[::-1]
    if reversed_word in words:
        password = word
        break

# 3. 비밀번호 길이와 가운데 글자 찾기
length = len(password)
middle_char = password[length // 2]

# 4. 결과 출력
print(length, middle_char)

예제와 결과

예제 입력

4
tulipan
napilut
apple
banana

예제 출력

7 i

요약

이 문제는 문자열을 뒤집어 비교하는 방법과 문자열의 가운데 글자를 찾는 기본적인 접근 방식을 통해 해결할 수 있습니다. 단계별로 문제를 나누고 각 단계에서 필요한 작업을 순차적으로 구현하면 쉽게 답을 구할 수 있습니다.

TODAY TIL

안녕하세요! 이번 포스팅에서는 군대 근무 시간 공평성 확인 문제와 관련된 흥미로운 코딩 문제를 다뤄보겠습니다. 단계적으로 설명하고 코드 구현을 공유합니다.

군대 근무 시간 공평성 확인 문제 풀이 (단계별 접근법)

문제 개요

푸앙이와 동기들은 군대에서 4교대 근무를 합니다. 각 시간대(08:00 :12:00, 12:00 : 18:00, 18:00 :22:00, 22:00 :08:00)별로 근무 시간이 다르고, 각자의 근무 시간이 최대한 공평하게 배분되기를 원합니다. 주어진 근무표가 공평하게 분배되었는지, 즉 각 근무자의 총 근무 시간 차이가 12시간 이하인지 판단하는 프로그램을 작성해보겠습니다.

단계별 문제 해결 과정

1단계: 문제 이해하기

• 주어진 정보
  • 주 수(N)가 주어지고, 각 주마다 4개의 근무 시간이 제공됩니다.
  • 시간대는 각각 4시간, 6시간, 4시간, 10시간으로 구성됩니다.
  • 근무표에서 -는 근무자가 없는 것을 의미하고, 근무자가 여러 명일 경우 공백으로 구분됩니다.
• 목표
  • 근무표에서 모든 근무자의 총 근무 시간을 계산하여, 최대 근무 시간과 최소 근무 시간의 차이가 12시간 이하인지 확인합니다.
  • 근무자가 없는 경우는 공평한 것으로 간주합니다.

2단계: 계획 세우기

1. 입력 처리: 주의 개수 N을 입력받고, 각 주별로 4줄씩 근무표를 읽어옵니다.
2. 근무 시간 계산하기: 각 시간대의 근무자를 확인하여 근무 시간을 누적합니다. 각 시간대에 따라 4, 6, 4, 10시간을 해당 근무자에게 추가합니다.
3. 공평성 검사: 모든 근무자의 근무 시간을 비교하여 차이가 12시간 이하인지 확인하고, 결과를 출력합니다.

3단계: 코드 작성 (단계별로 해결)

각 단계를 코드로 작성하면서 이해해보겠습니다

# 1. 입력 처리
N = int(input())  # 주의 개수 입력받기
schedule = [input().strip() for _ in range(N * 4)]  # 전체 근무표 입력받기

설명: N은 주의 개수이고, 근무표는 N * 4 줄로 주어집니다. schedule 리스트에 각 줄을 저장합니다.

# 2. 근무 시간 정의
hours = [4, 6, 4, 10]  # 각 시간대의 근무 시간 (08:00~12:00, 12:00~18:00, 18:00~22:00, 22:00~08:00)
work_time = {}  # 각 사람의 총 근무 시간을 저장할 딕셔너리

• 설명: 각 시간대별 근무 시간을 리스트에 저장하여 반복문에서 참조하기 쉽게 합니다.
• 딕셔너리 생성: work_time은 각 근무자의 이름을 키로 하고 근무 시간을 값으로 저장합니다.

# 3. 근무 시간 누적 계산
for i in range(N):  # 매 주마다 4개의 시간대가 있음
    for j in range(4):  # 각 시간대별로 근무자 확인
        people = schedule[i * 4 + j].split()  # 각 시간대별 근무자를 공백으로 분리
        for person in people:
            if person != '-':  # 근무자가 있는 경우만 처리
                if person not in work_time:
                    work_time[person] = 0
                work_time[person] += hours[j]  # 해당 시간대의 근무 시간을 더해줌

설명: i와 j를 이용해 각 주의 근무표에서 4개의 시간대를 차례로 가져오고, split()으로 각 시간대별 근무자를 분리합니다. -이 아닌 근무자에게만 시간을 누적합니다.

# 4. 공평성 검사
if work_time:  # 근무자가 있는 경우
    max_hours = max(work_time.values())
    min_hours = min(work_time.values())
    print("Yes" if max_hours - min_hours <= 12 else "No")
else:  # 근무자가 없는 경우
    print("Yes")

• 설명: 모든 근무자의 근무 시간을 비교하여 차이가 12시간 이하인 경우 “Yes”, 그렇지 않으면 “No”를 출력합니다.
• 예외 처리: 근무자가 없다면 바로 "Yes"를 출력합니다.

코드 전체

위의 각 단계를 하나의 코드로 합치면 다음과 같습니다.

# 입력 처리
N = int(input())  # 주의 개수
schedule = [input().strip() for _ in range(N * 4)]  # 전체 근무표 입력받기

# 각 시간대에 해당하는 근무 시간 정의
hours = [4, 6, 4, 10]  # 08:00~12:00, 12:00~18:00, 18:00~22:00, 22:00~08:00
work_time = {}  # 각 사람의 총 근무 시간을 저장할 딕셔너리

# 근무 시간 계산
for i in range(N):  # 매 주마다 4개의 시간대가 있음
    for j in range(4):  # 각 시간대별로 근무자 확인
        people = schedule[i * 4 + j].split()  # 각 시간대별 근무자를 분리
        for person in people:
            if person != '-':  # 근무자가 있는 경우만
                if person not in work_time:
                    work_time[person] = 0
                work_time[person] += hours[j]  # 해당 시간대의 근무 시간을 추가

# 공평성 검사
if work_time:  # 근무자가 있는 경우
    max_hours = max(work_time.values())
    min_hours = min(work_time.values())
    print("Yes" if max_hours - min_hours <= 12 else "No")
else:  # 근무자가 없는 경우
    print("Yes")

결론

이 문제를 통해 단계별로 문제를 나누어 풀어보았습니다. 입력을 먼저 처리한 후, 근무 시간을 누적 계산하고 공평성을 검사하는 방식으로 해결했습니다. 각 단계마다 목표를 명확히 설정하고 코드를 작성해 가면 더욱 이해하기 쉬울 것입니다.

TODAY TIL

안녕하세요! 이번 포스팅에서는 전주 듣고 노래 맞히기」 프로그램 와 관련된 흥미로운 코딩 문제를 다뤄보겠습니다. 첫 세 음만으로 노래 제목을 맞히는 프로그램을 단계적으로 설명하고 코드 구현을 공유합니다.

문제 설명

정환이 알고 있는 노래는 총 NNN개이고, 각각의 노래는 제목과 일곱 개의 음으로 이루어진 전주를 가지고 있습니다. 정환은 이 전주의 첫 세 음을 듣고 제목을 맞히려고 합니다.

요구사항

1. 주어진 첫 세 음에 맞는 노래가 딱 하나 있다면 해당 제목을 출력합니다.
2. 같은 첫 세 음으로 시작하는 노래가 여러 개면 ?를 출력합니다.
3. 해당하는 노래가 전혀 없다면 !를 출력합니다.

입력 및 출력 형식

• 첫 줄에는 정환이 알고 있는 노래의 개수 NNN과 맞히기를 시도할 노래의 개수 MMM이 주어집니다.
• 다음으로, NNN개의 노래 정보가 각각 제목과 전주의 일곱 개 음으로 주어집니다.
• 그 뒤에 맞히기를 시도할 첫 세 음이 MMM개 주어집니다.

출력 형식: 각 시도에 대해 노래 제목, ?, 또는 ! 중 하나를 한 줄에 출력합니다.

해결 방법

이 문제는 특정 패턴(첫 세 음)을 기준으로 노래 제목을 빠르게 찾는 문제입니다. 노래 정보를 사전에 딕셔너리에 저장해 두고, 맞히기 시도마다 해당 패턴이 있는지 빠르게 탐색하도록 합니다.

1. 아이디어 정리
   • 딕셔너리 자료 구조를 사용하여 각 노래의 첫 세 음을 키로 설정하고, 제목을 값으로 저장합니다.
   • 맞히기 시도 시, 입력된 첫 세 음으로 시작하는 노래를 딕셔너리에서 조회해 결과를 도출합니다.
   • 만약 해당 첫 세 음으로 시작하는 제목이 여러 개라면 ?, 하나라면 제목, 없으면 !를 출력합니다.
2. 코드 구현 단계
   • 노래 데이터 저장: 노래 정보를 입력받아 딕셔너리에 첫 세 음을 키로 하고, 제목을 리스트로 저장합니다. 리스트에 저장함으로써 다중 매칭이 가능하도록 합니다.
   • 맞히기 시도 처리: 맞히기 시도 시 딕셔너리를 조회하여 조건에 맞는 결과를 출력합니다.

# 1. 입력 받기
N, M = map(int, input().split())

# 2. 노래 정보를 저장할 딕셔너리 생성
song_dict = {}

# 3. 각 노래 정보를 입력 받아 첫 세 음을 키로 노래 제목을 딕셔너리에 저장
for _ in range(N):
    data = input().split()
    title = data[1]
    melody = ''.join(data[2:5])  # 첫 세 음만 추출
    
    if melody in song_dict:
        song_dict[melody].append(title)
    else:
        song_dict[melody] = [title]

# 4. 맞히기 시도 처리
for _ in range(M):
    guess_melody = ''.join(input().split())
    
    if guess_melody in song_dict:
        titles = song_dict[guess_melody]
        if len(titles) == 1:
            print(titles[0])  # 유일하게 일치하는 제목
        else:
            print('?')  # 여러 개의 제목이 일치
    else:
        print('!')  # 일치하는 제목이 없음

코드 설명

• 노래 정보 입력: 입력받은 각 노래 정보에서 첫 세 음만 추출해 딕셔너리에 저장합니다. 딕셔너리의 키는 첫 세 음, 값은 해당 제목의 리스트입니다.
• 맞히기 시도 처리: 첫 세 음을 기반으로 딕셔너리를 조회하여 조건에 맞는 출력을 도출합니다.

이 코드와 설명을 통해 전주 듣고 노래 맞히기 프로그램을 성공적으로 구현할 수 있습니다.

TODAY TIL

안녕하세요! 이번 포스팅에서는 보드게임 할리갈리와 관련된 흥미로운 코딩 문제를 다뤄보겠습니다. 게임에서 한별이가 종을 쳐야 할 시점을 자동으로 판단하는 코드를 작성하는 방법을 단계별로 설명하겠습니다.

📋 문제 설명

할리갈리 게임은 과일이 그려진 카드들과 종으로 이루어져 있습니다. 카드를 차례로 펼쳐서 한 종류의 과일이 정확히 5개가 될 때 가장 먼저 종을 쳐야 하는 게임입니다. 만약 한 종류의 과일이 5개가 되는 순간을 놓치지 않고 종을 칠 수 있도록 도와주는 프로그램을 만들어봅시다!

입력 조건

• 첫 번째 줄: 펼쳐진 카드의 개수 NNN
• 이후 각 줄에는 과일 종류 SSS와 과일 개수 XXX가 주어집니다.
• 과일 종류는 STRAWBERRY, BANANA, LIME, PLUM 중 하나입니다.

출력 조건

• 종을 쳐야 하면 "YES"를, 그렇지 않으면 "NO"를 출력합니다.

📝 문제 접근 방법

문제를 풀기 위해 각 과일의 개수를 누적한 후, 어떤 과일이 정확히 5개인지 확인하는 과정을 거칩니다.

1. 입력 데이터 처리: 카드의 개수 NNN과 카드별 과일 정보를 입력받습니다.
2. 과일 개수 누적: 각 과일에 대한 개수를 누적하여 관리합니다.
3. 종을 칠 조건 확인: 모든 카드의 정보를 누적한 후, 과일의 총 개수가 정확히 5인 과일이 있는지 확인합니다.

이제 코드로 문제를 해결해 보겠습니다.

# 입력 받기
N = int(input())  # 펼쳐진 카드 개수
fruit_counts = {"STRAWBERRY": 0, "BANANA": 0, "LIME": 0, "PLUM": 0}

# 카드 정보를 받아서 과일 수 누적
for _ in range(N):
    S, X = input().split()
    X = int(X)
    fruit_counts[S] += X

# 종을 칠 조건 확인
should_ring_bell = any(count == 5 for count in fruit_counts.values())

# 결과 출력
if should_ring_bell:
    print("YES")
else:
    print("NO")

코드 설명

1. fruit_counts 딕셔너리를 통해 각 과일의 개수를 0으로 초기화합니다.
2. 각 카드를 처리하면서 S와 X의 값을 읽고, fruit_counts의 해당 과일에 X를 더합니다.
3. any(count == 5 for count in fruit_counts.values())로 과일이 정확히 5개가 된 경우를 확인합니다.
4. 조건에 따라 "YES" 또는 "NO"를 출력합니다.

입력 예시

3
STRAWBERRY 2
BANANA 3
STRAWBERRY 3

출력 예시

YES

위 입력에서는 딸기가 총 5개(2 + 3)가 되므로, 한별이는 종을 쳐야 합니다.

🧩 마무리

이번 포스팅에서는 할리갈리 게임에서 종을 쳐야 하는 시점을 자동으로 판별하는 코드를 구현해 보았습니다. 과일 개수를 누적하면서 조건을 확인하는 문제를 통해 딕셔너리와 반복문의 활용을 연습할 수 있었습니다. 앞으로도 다양한 게임과 관련된 문제로 알고리즘을 즐겁게 풀어보아요!