Swift 프로그래머스 수포자

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건

• 시험은 최대 10,000 문제로 구성되어있습니다.
• 문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

입출력 예

입력	출력
[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

핵심:

1.1,2,3의 각 패턴을 알아낸다.

1 -> 5로 나눴을 때 나머지 5일 경우엔 5

2 ->홀수인 경우엔 2 짝수일경우엔 8로 나눴을 때 나머지가 각각 2,4,6,8

3 -> 5로 나눴을 때 나머지가 1,2,3,4 5일 경우엔 0

2.패턴을 이용하여 answers어레이 값에 맞게 값을 넣어주고 count변수를 생성해 answers와 각 배열을 비교해준다. 만약 값이 같을 땐 count 숫자를 +1해준다.

3.모든 경우의 수를 알아내 비교한다. -> 총 10가지

a,b,c 모두 같을 때 1

a,b,c 중 가장 큰 수가 있을 때 3

3개 중 두개가 같고 그 두수가 더 클때 3

3개 중 두개가 같고 그 두수가 더 작을 때 3

func solution(_ answers:[Int]) -> [Int] {
    var a = [Int]()
    var b = [Int]()
    var c = [Int]()
    var acount = 0
    var bcount = 0
    var ccount = 0
    for number in 1...answers.count{
        if number%5 != 0{
            a.append(number%5)
        }else{
            a.append(5)
        }
        if number%2 != 0 {
            b.append(2)
        }else if number%8 == 2{
            b.append(1)
        }else if number%8 == 4{
            b.append(3)
        }else if number%8 == 6{
            b.append(4)
        }else if number%8 == 0{
            b.append(5)
        }
        if number%5 == 1{
            c.append(3)
            c.append(3)
        }else if number%5 == 2 {
            c.append(1)
            c.append(1)
        }else if number%5 == 3 {
            c.append(2)
            c.append(2)
        }else if number%5 == 4 {
            c.append(4)
            c.append(4)
        }else if number%5 == 0 {
            c.append(5)
            c.append(5)
        }
        if a[number-1] == answers[number-1] {
                   acount += 1
               }
               if b[number-1] == answers[number-1]{
                   bcount += 1
               }
               if c[number-1] == answers[number-1]{
                   ccount += 1
               }
    }
    if acount > bcount && acount > ccount{
        return [1]
    }else if bcount > acount && bcount > ccount{
        return [2]
    }else if ccount > acount && ccount > bcount{
        return [3]
    }else if acount == bcount && bcount > ccount{
        return [1,2]
    }else if acount == ccount && ccount > bcount{
        return [1,3]
    }else if ccount == bcount && bcount > acount{
        return [2,3]
    }else if acount == bcount && bcount < ccount{
        return [3]
    }else if acount == ccount && bcount > ccount{
        return [2]
    }else if bcount == ccount && acount > ccount{
        return [1]
    }else {
        return [1,2,3]
    }
    
}

다른 사람의 풀이 중 가장 간결한 것

핵심 :

1.a,b,c에 패턴을 반복되는 구간대로 각 각 저장해놨다. 딕셔너리를 사용해서 1,2,3을 분류

2.answers.enumerated()라는 메소드를 이용해서  i,v를 각 각 키와 밸류로 값을 줌 ex)answers[i] = v 와 같이 값이 나오게 함.

3.answer = (a:[1,2,3,4,5]) 이런 소괄호 안에 딕셔너리 형태로 넣어주면 answer.key값을 넣어주면 배열을 알 수 있음

4.a 5개 b는 8개 c는 10개가 각 각 반복이므로 index를 %5 %8 %10해주면 각 배열의 같은 값이 나온다. 여기서 answers의 밸류값과 같다면 point의 밸류값을 1씩 더해줘라

5.sorted{ $0.key < $1.key }를 이용해 point의 키(1,2,3) 중 숫자가 적은 순으로 소팅하고 , filter{ $0.value == point.values.max()}를 이용해 point안의 value가 최대값인 걸 필터,그리고 map{$0.key} sorting된 순으로 filter에 통과된 point의 key들을 배열로 담아라

 

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    let answer = (
        a: [1, 2, 3, 4, 5], // index % 5
        b: [2, 1, 2, 3, 2, 4, 2, 5], // index % 8
        c: [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] // index % 10
    )
    var point = [1:0, 2:0, 3:0]

    for (i, v) in answers.enumerated() {
        if v == answer.a[i % 5] { point[1] = point[1]! + 1 }
        if v == answer.b[i % 8] { point[2] = point[2]! + 1 }
        if v == answer.c[i % 10] { point[3] = point[3]! + 1 }
    }

    return point.sorted{ $0.key < $1.key }.filter{ $0.value == point.values.max() }.map{ $0.key }
}