[프로그래머스] Swift - 개인정보 수집 유효기간 (LV.1)

👆문제풀러 가기👆

문제 설명

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

 

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

 

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

 

다음은 오늘 날짜가 2022.05.19일 때의 예시입니다.

약관 종류	유효기간
A	6 달
B	12 달
C	3 달

번호	개인정보 수집 일자	약관 종류
1	2021.05.02	A
2	2021.07.01	B
3	2022.02.19	C
4	2022.02.20	C

• 첫 번째 개인정보는 A약관에 의해 2021년 11월 1일까지 보관 가능하며, 유효기간이 지났으므로 파기해야 할 개인정보입니다.
• 두 번째 개인정보는 B약관에 의해 2022년 6월 28일까지 보관 가능하며, 유효기간이 지나지 않았으므로 아직 보관 가능합니다.
• 세 번째 개인정보는 C약관에 의해 2022년 5월 18일까지 보관 가능하며, 유효기간이 지났으므로 파기해야 할 개인정보입니다.
• 네 번째 개인정보는 C약관에 의해 2022년 5월 19일까지 보관 가능하며, 유효기간이 지나지 않았으므로 아직 보관 가능합니다.

따라서 파기해야 할 개인정보 번호는 [1, 3]입니다.

 

오늘 날짜를 의미하는 문자열 today, 약관의 유효기간을 담은 1차원 문자열 배열 terms와 수집된 개인정보의 정보를 담은 1차원 문자열 배열 privacies가 매개변수로 주어집니다. 이때 파기해야 할 개인정보의 번호를 오름차순으로 1차원 정수 배열에 담아 return 하도록 solution 함수를 완성해 주세요.

 

제한사항

• today는 "YYYY.MM.DD" 형태로 오늘 날짜를 나타냅니다. 
• 1 ≤ terms의 길이 ≤ 20
  • terms의 원소는 "약관 종류 유효기간" 형태의 약관 종류와 유효기간을 공백 하나로 구분한 문자열입니다.
  • 약관 종류는 A~Z중 알파벳 대문자 하나이며, terms 배열에서 약관 종류는 중복되지 않습니다.
  • 유효기간은 개인정보를 보관할 수 있는 달 수를 나타내는 정수이며, 1 이상 100 이하입니다.
• 1 ≤ privacies의 길이 ≤ 100
  • privacies[i]는 i+1번 개인정보의 수집 일자와 약관 종류를 나타냅니다.
  • privacies의 원소는 "날짜 약관 종류" 형태의 날짜와 약관 종류를 공백 하나로 구분한 문자열입니다.
  • 날짜는 "YYYY.MM.DD" 형태의 개인정보가 수집된 날짜를 나타내며, today 이전의 날짜만 주어집니다.
  • privacies의 약관 종류는 항상 terms에 나타난 약관 종류만 주어집니다.
• today와 privacies에 등장하는 날짜의 YYYY는 연도, MM은 월, DD는 일을 나타내며 점(.) 하나로 구분되어 있습니다.
  • 2000 ≤ YYYY ≤ 2022
  • 1 ≤ MM ≤ 12
  • MM이 한 자릿수인 경우 앞에 0이 붙습니다.
  • 1 ≤ DD ≤ 28
  • DD가 한 자릿수인 경우 앞에 0이 붙습니다.
• 파기해야 할 개인정보가 하나 이상 존재하는 입력만 주어집니다.

 

입출력 예제

today	terms	privacies	result
"2022.05.19"	["A 6", "B 12", "C 3"]	["2021.05.02 A", "2021.07.01 B", "2022.02.19 C", "2022.02.20 C"]	[1, 3]
"2020.01.01"	["Z 3", "D 5"]	["2019.01.01 D", "2019.11.15 Z", "2019.08.02 D", "2019.07.01 D", "2018.12.28 Z"]	[1, 4, 5]

 

 

풀이 [ 메모리: 16.4mb, 최대시간:  3.21ms ] 

func solution(_ today:String, _ terms:[String], _ privacies:[String]) -> [Int] {
    let expired = expiredPeriods(of: terms) // 약관에 따른 유효기간
    let dates = calculateExpired(privacies, expired)
    var result = [Int]()

    // 현재 날짜와 유효기간 비교
    for i in 0..<dates.count {
        let (exYear, exMonth, exDay) = (dates[i][0], dates[i][1], dates[i][2]) // 파기일
        let ymd = today.split(separator: ".").compactMap { Int($0) }
        let (year, month, day) = (ymd[0], ymd[1], ymd[2]) // 현재 날짜

        if year > exYear
        || year == exYear && month > exMonth
        || year == exYear && month == exMonth && day > exDay { // 파기일보다 현재 날짜가 크면 인덱스 저장.
            result.append(i+1)
        }
    }

    return result.sorted()
}

func expiredPeriods(of terms: [String]) -> [String: Int] {
    var expired = [String: Int]() // 약관에 따른 유효기간

    // 약관에 따른 유효기간 저장
    for term in terms {
        let tp = term.split(separator: " ").map { String($0) }
        let type = tp[0]
        let period = Int(tp[1])!

        expired[type, default: 0] = period
    }

    return expired
}

func calculateExpired(_ privacies: [String], _ expired: [String: Int]) -> [[Int]] {
    var dates = [[Int]]()

    for privacy in privacies {
        let dt = privacy.split(separator: " ").map { String($0) }
        let date = dt[0].split(separator: ".").compactMap { Int($0) }
        let type = dt[1]

        var (year, month, day) = (date[0], date[1], date[2])

        month += expired[type]!

        if day == 1 { // day가 1일이면, 마감일은 이전 달, 마지막 날임.
            month -= 1
            day = 28
        } else {
            day -= 1
        }

        if month > 12 {
            year += month / 12
            month %= 12
        }

        if month == 0 { // month가 12의 배수 또는 1월이었으면 전년도 12월이 되어야 함.
            year -= 1
            month = 12
        }

        dates.append([year, month, day])
    }
    return dates
}

문제를 받고 생각한 것.

1. 폐기되어있어야 하는 개인정보의 번호를 찾기
2. 유효기간 마감일과 오늘 날짜의 연, 월, 일을 비교해 오늘 날짜가 더 크면 폐기해야하는 개인정보
3. 수집된 날이 1일이면, 마감일은 28일이다. -> 유효기간이 1달이면, month는 그대로이고 day = 28이 되어야 한다.
4. 유효기간은 월단위로 제공되기 때문에, 수집된 월 + 유효기간을 한다.
5. 수집된 월 + 유효기간이 12월을 넘기면 연도를 올리고, 월은 12의 나머지 값이 된다.
6. 월이 0이 되면, 12월이라는 의미이므로, year -1, month = 12를 한다
7. 이 과정을 반복한 뒤, 연, 월, 일을 현재와 비교한다.

라고 생각했는데, 다 풀고나서 다른 사람들의 코드를 보니 굳이 이렇게 할 필요가 없었다는 걸 알았다.

연도는 12개월이고, 1개월은 28일이므로

• 1년 = 12 * 28
• 1개월 = 28

이라고 생각하면 코드가 훨씬 간결해진다.

 

풀이 [ 메모리: 16.4mb, 최대시간:  1.68ms ] 

func solution(_ today:String, _ terms:[String], _ privacies:[String]) -> [Int] {
    let expired = expiredPeriods(of: terms) // 약관에 따른 유효기간
    let dates = calculateExpired(privacies, expired)
    var result = [Int]()

    // 현재 날짜와 유효기간 비교
    for i in 0..<dates.count {
        let ymd = today.split(separator: ".").compactMap { Int($0) }
        let (year, month, day) = (ymd[0] * 12 * 28, ymd[1] * 28, ymd[2]) // 현재 날짜
        let days = year + month + day

        if days >= dates[i] { // 파기일보다 현재 날짜가 크면 인덱스 저장.
            result.append(i+1)
        }
    }

    return result
}

func expiredPeriods(of terms: [String]) -> [String: Int] {
    var expired = [String: Int]() // 약관에 따른 유효기간

    // 약관에 따른 유효기간 저장
    for term in terms {
        let tp = term.split(separator: " ").map { String($0) }
        let type = tp[0]
        let period = Int(tp[1])!

        expired[type, default: 0] = period * 28
    }

    return expired
}

func calculateExpired(_ privacies: [String], _ expired: [String: Int]) -> [Int] {
    var dates = [Int]()

    for privacy in privacies {
        let dt = privacy.split(separator: " ").map { String($0) }
        let date = dt[0].split(separator: ".").compactMap { Int($0) }
        let type = dt[1]

        let (year, month, day) = (date[0] * 12 * 28, date[1] * 28, date[2])

        dates.append(year + month + day + expired[type]!)
    }
    return dates
}

나는 변수를 많이 사용했지만, 변수는 많이 사용하지 않아도 되는 문제이다.