[프로그래머스] 개인정보수집 유효기간

[Lv.1] 정답율 40%

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

프로그래머스

 

[풀이]

- datetime 모듈을 이용해 문자열 날짜를 컴퓨터가 이해하는 날짜 포맷으로 변경했다
- dateutil.relativedelta를 이용해 현재 날짜에서 개월 또는 일자 단위로 빼거나 더하는 법을 배웠다

 

[코드]

from datetime import datetime
from dateutil.relativedelta import relativedelta

def solution(today, terms, privacies):
    date_format = '%Y.%m.%d'
    today = datetime.strptime(today, date_format) # datetime으로 변환

    # 약관 종류별 유효기간을 해쉬테이블로 정리
    terms_dict = {}
    for idx, term in enumerate(terms) :
        terms_dict[term.split(' ')[0]] = term.split(' ')[1]

    answer = []
    for privacy_idx, privacy in enumerate(privacies) :
        term_start = privacy[:10] # 약관 시작일 추출
        term_start = datetime.strptime(term_start, date_format) # datetime으로 변환
        term_key = privacy[-1] # 약관 종류 추출
        term_validate = int(terms_dict[term_key]) # 약관 유효기간 추출

        if term_start + relativedelta(months=term_validate) - relativedelta(days=1) < today :
            answer.append(privacy_idx+1)

    return answer