[python] 프로그래머스 코테 - level1. 개인정보 수집 유효기간

문제

고객의 약관 동의를 얻어서 수집된 1~n번으로 분류되는 개인정보 n개가 있습니다. 약관 종류는 여러 가지 있으며 각 약관마다 개인정보 보관 유효기간이 정해져 있습니다. 당신은 각 개인정보가 어떤 약관으로 수집됐는지 알고 있습니다. 수집된 개인정보는 유효기간 전까지만 보관 가능하며, 유효기간이 지났다면 반드시 파기해야 합니다.
예를 들어, A라는 약관의 유효기간이 12 달이고, 2021년 1월 5일에 수집된 개인정보가 A약관으로 수집되었다면 해당 개인정보는 2022년 1월 4일까지 보관 가능하며 2022년 1월 5일부터 파기해야 할 개인정보입니다.
당신은 오늘 날짜로 파기해야 할 개인정보 번호들을 구하려 합니다.
모든 달은 28일까지 있다고 가정합니다.
오늘 날짜를 의미하는 문자열 today, 약관의 유효기간을 담은 1차원 문자열 배열 terms와 수집된 개인정보의 정보를 담은 1차원 문자열 배열 privacies가 매개변수로 주어집니다. 이때 파기해야 할 개인정보의 번호를 오름차순으로 1차원 정수 배열에 담아 return 하도록 solution 함수를 완성해 주세요.

풀이

해결!

제한 사항에 term의 길이가 20 이하라고 되어 있어서 20달까지로 밖에 생각을 안했었다. 문제를 잘 읽어보니 길이가 20 이하였던 것이다.

즉, 24개월 일 때를 처리 안해줘서 한 문제가 틀렸던 것이다.

solution("2022.02.28", ["A 24", "B 12"], ["2020.02.01 A", "2020.01.01 B"])

만약 위의 예시라면 2022.01.28, 2020.12.28 까지 보관 가능하다.

난 mon라는 딕셔너리를 생성하여 term의 값을 넣어줬다.

ex) term = ['A' '24', 'B' '12'] -> mon = {'A': '24', 'B': '24'}

고려해야 할 사항

01월 01일일 때 또는 01일일 때

유효기간이 24개월 이상일 때

풀이

from datetime import date
def solution(today, terms, privacies):
    answer = []
    mon = {}
    today_time = date(int(today[:4]), int(today[5:7]),int(today[8:10]))

    for sep in terms:
        ter, mont = sep.split(' ') # term의 값들을 띄어쓰기로 분리 후 ter, mont 에 저장
        mon[ter] = mont # 딕셔너리에 키, 값으로 저장
    
    for j,i in enumerate(privacies,1):
        dateday, term = i.split(' ')
        year = int(dateday[:4])

        month = int(dateday[5:7]) + int(mon[term])  # 수집된 정보의 달 + 해당 유효기간
        monthPer = int(month / 12) # 전체 달이 12이므로 12로 나눈 몫을 구하여 년도에 더해줌
        year = year + monthPer 
        if int(dateday[8:10]) == 1: # 수집된 정보의 일이 01이라면
            if month % 12 == 0: 
                month = month % 12 -1 
                year -=1
            else:
                month = month % 12 -1
        else:
            month = month % 12

        if month == 0:
            month = 12
            year -=1
        elif month < 0:
            month = 12 + month

        day = dateday[8:10]
        if  day == '01': 
            day = 28
            
        else:
            day = int(day) - 1
        
        old_time = date(year,month,day)
        if today_time > old_time:
            answer.append(j)

    return answer