Greedy - LV 1

이태휘·2023년 5월 1일
0
post-custom-banner

체육복

set이라는 집합단위를 활용하여 중복제거하는 것을 생각하는게 어려웠다,,,

  • 내 두뇌문제보단 python이라는 언어에 대한 이해도 부족으로 생각함.
  • 그래서 츄라이하다가 구글링함 ,, ^_^

요점
1. 중복 제거를 해야함! => 여벌이 있는 친구가 체육복 잃어버렸다면 자기거 자기가 사용해야하기 때문
2. 잃어버린 친구의 양옆에서 가져올 수 있으면 체육복을 가져오기!

def solution(n, lost, reserve):
    unique_reserve = set(reserve)-set(lost) 
    unique_lost = set(lost)-set(reserve) 
    for i in unique_reserve: 
        if i-1 in unique_lost: 
            unique_lost.remove(i-1) 
            
        elif i+1 in unique_lost: 
            unique_lost.remove(i+1) 
            
    return n - len(unique_lost)
profile
풀스택 개발자 가보자구~
post-custom-banner

0개의 댓글