프로그래머스) 개인정보 수집 유효기간 -python

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

프로그래머스

 

문제

고객의 약관 동의를 얻어서 수집된 1~n번으로 분류되는 개인정보 n개가 있습니다. 약관 종류는 여러 가지 있으며 각 약관마다 개인정보 보관 유효기간이 정해져 있습니다. 당신은 각 개인정보가 어떤 약관으로 수집됐는지 알고 있습니다. 수집된 개인정보는 유효기간 전까지만 보관 가능하며, 유효기간이 지났다면 반드시 파기해야 합니다.

예를 들어, A라는 약관의 유효기간이 12 달이고, 2021년 1월 5일에 수집된 개인정보가 A약관으로 수집되었다면 해당 개인정보는 2022년 1월 4일까지 보관 가능하며 2022년 1월 5일부터 파기해야 할 개인정보입니다.
당신은 오늘 날짜로 파기해야 할 개인정보 번호들을 구하려 합니다.

모든 달은 28일까지 있다고 가정합니다.

즉 오늘 날짜와 주어진 날짜를 비교해 파기하는 정보들을 구하는 문제다.

사실 이 문제는 난이도 1인만큼 라이브러리를 활용한다면 그리 어렵지 않은 문제다.

단지 나같은 경우엔 python이 주언어가 아닌지라 라이브러리 이용에 애를 좀 먹었다.

 

import datetime
from dateutil.relativedelta import relativedelta

def solution(today, terms, privacies):
    answer = []
    format = '%Y-%m-%d'
    dict = {}
    today = today.replace('.','-')
    dt_datetime = datetime.datetime.strptime(today,format) #today
    
    
    for i in terms:
        a,b = i.split()
        dict[a] = int(b)
        #term를 dict에 저장
    
    for i in range(len(privacies)):
        a , b= privacies[i].split()
        
        p_day = a.replace('.','-')
        p = datetime.datetime.strptime(p_day,format)
        
        plus = p + relativedelta(months=dict[b])
        
        if plus <= dt_datetime:
            answer.append(i+1)
        
    
    
    return answer