[Programmers로 코딩테스트 준비] 조이스틱 - Lv.2_Day2👎🏻

아직 이해 ❌

일자: 2024년 08월 28,30일

알고리즘: 탐욕법

 

 

 

문제 설명

조이스틱으로 알파벳 이름을 완성하세요. 맨 처음엔 A로만 이루어져 있습니다.
ex) 완성해야 하는 이름이 세 글자면 AAA, 네 글자면 AAAA

조이스틱을 각 방향으로 움직이면 아래와 같습니다.

▲ - 다음 알파벳
▼ - 이전 알파벳 (A에서 아래쪽으로 이동하면 Z로)
◀ - 커서를 왼쪽으로 이동 (첫 번째 위치에서 왼쪽으로 이동하면 마지막 문자에 커서)
▶ - 커서를 오른쪽으로 이동 (마지막 위치에서 오른쪽으로 이동하면 첫 번째 문자에 커서)
예를 들어 아래의 방법으로 "JAZ"를 만들 수 있습니다.

- 첫 번째 위치에서 조이스틱을 위로 9번 조작하여 J를 완성합니다.
- 조이스틱을 왼쪽으로 1번 조작하여 커서를 마지막 문자 위치로 이동시킵니다.
- 마지막 위치에서 조이스틱을 아래로 1번 조작하여 Z를 완성합니다.
따라서 11번 이동시켜 "JAZ"를 만들 수 있고, 이때가 최소 이동입니다.
만들고자 하는 이름 name이 매개변수로 주어질 때, 이름에 대해 조이스틱 조작 횟수의 최솟값을 return 하도록 solution 함수를 만드세요.

제한 사항

• name은 알파벳 대문자로만 이루어져 있습니다.
• name의 길이는 1 이상 20 이하입니다.

 

입출력 예

name	return
"JEROEN"	56
"JAN"	23

 

 

접근 방식

• ord로 몇 번째인지 계산
• 순행할지 역행할지 조건

 

def solution(name):
    length = len(name)
    alpha_list = []
    for alpha in name:
        alpha_list.append(_ord(alpha))
    print(alpha_list)
    
    if 0 not in alpha_list:
        answer = sum(alpha_list) + length - 1

 

• 알파벳 번호 리스트를 생성
• name에 "A"가 없을 경우 리스트 sum + length-1
• "A"가 있는 경우 순행/역행 결정 과정이 관건

 

def _ord(num):
    point = ord('A')
    half = (ord('Z') + point) // 2
    if ord(num) >= half:
        result = ord('Z') - ord(num) + 1
    else:
        result = ord(num) - point
    return result

def _cnt(alpha_list, i):
    try:
        front = alpha_list[i:].index(0)
        end = alpha_list[i:][::-1].index(0)
        end_index = len(alpha_list) - end - 1
        if front == end_index: # 앞으로 가나 뒤로 가나 같은 거리
            print('same', front)
            return front
        else:
            _cnt(alpha_list, i+1)
    except:
        return i

• 이리저리 꼬여버림....

 

 

From Claude

def solution(name):
    # 알파벳 변경에 필요한 최소 조작 횟수를 계산하는 함수
    def alphabet_to_num(char):
        return min(ord(char) - ord('A'), ord('Z') - ord(char) + 1)
    
    # 전체 알파벳 변경에 필요한 조작 횟수
    answer = 0
    for char in name:
        answer += alphabet_to_num(char)
    
    # 커서 이동에 필요한 최소 조작 횟수 계산
    min_move = len(name) - 1
    for i in range(len(name)):
        next_i = i + 1
        while next_i < len(name) and name[next_i] == 'A':
            next_i += 1
        min_move = min(min_move, i + i + len(name) - next_i)
    
    return answer + min_move

• 움직여야 하는 횟수: min((목표-시작), (끝+1)-목표)

 

반례 -  "BBBBAAAABA" 기댓값: 12, 결과값: 13

 

 

수정된 코드

def solution(name):
    def alphabet_to_num(char):
        return min(ord(char) - ord('A'), ord('Z') - ord(char) + 1)
    
    # 알파벳 변경에 필요한 조작 횟수
    answer = 0
    for char in name:
        answer += alphabet_to_num(char)
    
    # 커서 이동에 필요한 최소 조작 횟수 계산
    min_move = len(name) - 1
    for i in range(len(name)):
        next_i = i + 1
        while next_i < len(name) and name[next_i] == 'A':
            next_i += 1
        
        # 왼쪽으로 돌아가는 경우와 오른쪽으로 계속 가는 경우 중 최소값 선택
        min_move = min(min_move, 2 * i + len(name) - next_i, i + 2 * (len(name) - next_i))
    
    return answer + min_move

• min_move에서 비교하는 갯수 추가됨
  • min_move: 이전까지의 최소 이동 횟수
  • 2 * i + len(name) - next_i: 현재 위치까지 오른쪽으로 갔다가 다시 왼쪽으로 돌아가는 경우
  • i + 2 * (len(name) - next_i): 현재 위치까지 오른쪽으로 간 후, 왼쪽 끝으로 이동하여 나머지를 처리하는 경우

 

접근방식 참고

https://school.programmers.co.kr/questions/75691