Merry's Dev. Blog

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42842

정답 코드

첫 번째 제출

def solution(brown, yellow):
    answer = []
    yellow_wh = (brown-4)/2
    i=1
    while i <=  yellow_wh//2:
        if i * (yellow_wh-i) == yellow:
            answer.append((yellow_wh-i)+2)
            answer.append(i+2)
            break
        i=i+1
    return answer

• yellow_wh: yellow의 가로 길이+세로 길이
  1. brown = (yellow 가로 길이+세로 길이) * 2 + 4 이므로 중간 합을 기준으로 식을 변형하면 위 코드처럼 됨
• yellow_wh//2 까지 i 를 1씩 더하면서 반복문 돌리기: yellow = i + (yellow_wh-i) 조건에 맞는 쌍을 찾기 위함
• yellow 가로>=세로 길이 이므로 애초에 큰 숫자(yellow_wh - i)를 먼저 append 하면 정답 나옴 

두 번째 제출 (다른 사람 풀이 참고)

def solution(brown, yellow):
    yellow_wh = (brown-4)/2
    i=1
    while i <=  yellow_wh//2:
        if i * (yellow_wh-i) == yellow:
            return [(yellow_wh-i)+2, i+2]
        i=i+1

• answer 배열에 append 하지 않고, 바로 배열 형태로 return 하면 코드가 더욱 짧아짐

문제 및 레퍼런스

https://school.programmers.co.kr/learn/courses/30/lessons/42862

https://namhandong.tistory.com/152

https://iambeginnerdeveloper.tistory.com/107

정답 코드

def solution(n, lost, reserve):
    
    answer = 0  #1번
    new_reserve = set(reserve)-set(lost)  #2번
    new_lost = set(lost)-set(reserve) 
    
    for i in new_reserve:  #2번
        if i-1 in new_lost: 
            new_lost.remove(i-1)  #3번
        elif i+1 in new_lost: 
            new_lost.remove(i+1) 
            
    answer = n - len(new_lost)  #1번
    
    return answer

배운 점

0. 변수 설정

• 문풀 전: 내가 정한 규칙에 따라 입력값을 항상 재설정 
  • 문제점: 레퍼런스를 보게 될 경우, 변수 달라 항상 고생. 또한 대문자로 변수 썼으므로 고속으로 코딩하기에 불편
• 결론: 프로그래머스처럼 입력값 변수가 정해져 있는 경우 그냥 그거 쓰자 

1. 리턴 값 초기화의 중요성 

• 문풀 전: 리턴 값을 초기화 하지 않고, 바로 answer = n - len(new_lost) 식으로 값 할당 
  • 문제점: 코드를 맞게 작성하였는데도, 심지어 정답 코드들과 초기화 부분을 제외하고 코드가 동일한데도  계속 일부 3-5개 케이스를 통과하지 못함 
  • 원인: 이전에 문풀 했을 때 할당되었던 answer이 누적될 수 있음
• 결론: 항상 리턴 값은 초기화하고 시작하는 것을 습관화하자

2. 배열 or 집합 값의 직접 탐색 

1. 직접 탐색
   • 문풀 전: 항상 for i in range(len(lost)) 식으로 탐색하고 싶은 배열의 길이의 범위를 지정하여 탐색
     
     • 문제점: lost[i] 식으로 한 번 더 그 값을 지정해주어야 하는 번거로움 발생. 또한 이번 문제처럼 lost 도 탐색해야 하는 경우 무조건 이중 for문 사용하여 배열 값에 접근해야함 
   • 결론: 배열 값에 바로 접근하자. for i in lost
2. 집합 값 탐색
   • 문풀 전: 집합을 생성하고 집합을 탐색하는 것을 몰랐음
     • 문제점: 이번 문제처럼 공통 원소를 제거하는 set(lost) - set(reserve) 같은 코드를 사용할 수 없음
   • 결론: 배열, 집합 양자 왔다 갔다 하자
     • 배열을 집합으로 만들기: set(lost) ( 결과: {2, 4} )
     • 차집합 (= 공통 원소 제거): set(lost) - set(reserve)
     • 집합도 len(lost), lost.remove(i) 처럼 배열에서 사용하는 메서드 사용 가능

3. remove() 메서드

• 문풀 전: 배열에서 원소 제거할 때 무조건 pop() 메서드 사용
  • 문제점: 메서드 특성 상 스택큐 문제에서 뒤의 원소부터 제거한다는 느낌이 강함 (실제 특정 값만 제거할 수 있는데도). 따라서 부담스러움
• 결론
  • remove() 메서드를 사용하자.
  • 다만 배열 or 집합 순회하면서 remove() 메서드를 사용할 경우, 순회하는 대상이 달라져 원하는 결과가 나오지 않을 가능성 있음 
    -> lost[:] 식으로 리스트의 전체를 슬라이싱하는 기법 사용하자. 이를 통해 리스트의 모든 요소를 복사하여 새로운 리스트를 생성하고, 이를 순회하면서 각 요소에 접근 가능 

참고 문제

• 백준 2864 5와 6의 차이

A, B = map(str, input().split()) #str로 입력

mini = int(A.replace('6', '5')) + int(B.replace('6', '5')) #replace함수
maxi = int(A.replace('5', '6')) + int(B.replace('5', '6'))
print(mini, maxi)

• 접근 오류 
  • ''.replace() 식으로 gpt한테 이상한 함수 사용법 받아와서 적용해보다가 틀림  
  • 문제의 핵심 로직 생각해낸 것은 맞았지만, replace() 함수를 사용해본 적 없어서 괜히 함수 써야할 것 같은 강박 관념에 틀림
  • input을 str로 받아놓고, A.replace(6, 5) 식으로 int로 함
  • print 할 때 굳이 print(mini, ' ', maxi) 안 하고 위처럼 해도 알아서 공백 넣어 숫자 2개 출력해줌 
  •  

참고 문제

• 백준 5585 거스름돈

M = 1000 - int(input())  // input 받아 한 번에 M 정의
coins = [500,100,50,10,5,1]  // 반복되는 로직의 대상을 coins 배열에 넣기
ans = 0
for coin in coins:
    ans += n//coin
    M %= coin  // M 업데이트
print(ans)

참고 문제 

• 백준 11034 캥거루 세 마리2

while True:  # 무한 루프 돌겠다
    try:
        A, B, C = map(int, input().split())
        jumpA = B - A - 1
        jumpB = C - B - 1
        ans = max(jumpA, jumpB)
        if ans == 0:
            print(0)
        else:
            print(ans)
    except EOFError:  # input 공백일 때 읽어들이려고 할 때 발생하는 에러, break 처리
        break

참고 문제 

• 백준 10162 전자레인지
• 백준 2720 세탁소 사장 동혁 

T = int(input())
C = []
for i in range(T):
    C = list(map(int, input().split()))
    
    # 값 초기화
    Q = D = N = P = 0
    
    # 리스트의 첫 번째 값만 활용하고 바로 답 출력
    Q = C[0] // 25
    D = C[0] % 25 // 10
    N = (C[0] % 25) % 10 // 5
    P = ((C[0] % 25) % 10) % 5 // 1
    print(Q, D, N, P)