[JAVA] 프로그래머스 코딩테스트 - 개인정보 수집 유효기간

문제 설명
고객의 약관 동의를 얻어서 수집된 1~n번으로 분류되는 개인정보 n개가 있습니다. 약관 종류는 여러 가지 있으며 각 약관마다 개인정보 보관 유효기간이 정해져 있습니다. 당신은 각 개인정보가 어떤 약관으로 수집됐는지 알고 있습니다. 수집된 개인정보는 유효기간 전까지만 보관 가능하며, 유효기간이 지났다면 반드시 파기해야 합니다.예를 들어, A라는 약관의 유효기간이 12 달이고, 2021년 1월 5일에 수집된 개인정보가 A약관으로 수집되었다면 해당 개인정보는 2022년 1월 4일까지 보관 가능하며 2022년 1월 5일부터 파기해야 할 개인정보입니다.당신은 오늘 날짜로 파기해야 할 개인정보 번호들을 구하려 합니다.모든 달은 28일까지 있다고 가정합니다.

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

프로그래머스

 

 

풀이

- terms 배열에서 필요한 정보(유효기간)를 뽑아 써야 하므로 HashMap으로 만들어주었다.

- 두 날짜를 비교해야 하므로 LocalDate로 파싱했다.

- LocalDate의 isAfter()로 날짜를 비교했다.

- ArrayList로 만료된 날짜의 번호를 추가하고, 제출 전에 Array에 담아 제출했다.

import java.util.HashMap;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;

class Solution {
    public int[] solution(String today, String[] terms, String[] privacies) {
        int[] answer = {};
        
        DateTimeFormatter formatter = 
            DateTimeFormatter.ofPattern("yyyy.MM.dd");
        //날짜 계산을 위해 LocalDate로 파싱한다
        LocalDate formatToday = LocalDate.parse(today, formatter);
            
        //terms를 hashMap에 담는다
        HashMap<String, Integer> termsMap = new HashMap<String, Integer>();
        for (String term : terms) {
            String key = term.split(" ")[0];
            int value = Integer.parseInt(term.split(" ")[1]);
            termsMap.put(key, value);
        }
        
        ArrayList<Integer> al = new ArrayList<Integer>();
        //privacies로 termsMap에서 유효기간을 get
        for (int i=0; i<privacies.length; i++) {
            String code = privacies[i].split(" ")[1];
            String infoDate = privacies[i].split(" ")[0];
            int validMonth = termsMap.get(code);
            
            // 만료 날짜 구하기
            LocalDate validDate = LocalDate.parse(infoDate, formatter)
                .plusMonths(validMonth).minusDays(1);
            
            //만료 날짜와 오늘 날짜 비교
            if(formatToday.isAfter(validDate)) {
                al.add(i+1);
            }
        }
        
        int idx = 0;
        answer = new int[al.size()];
        for (int item : al) {
            answer[idx++] = item;
        }
        return answer;
    }
}