[Algorithm] 프로그래머스 실패율 Level1 (Swift)

코딩테스트 연습 - 실패율

문제 풀이

• 푸는 데 정말 오래 걸린 문제... 대충 머릿속으로 큰 그림을 그리고, while 문으로 처리하려고 했다.
• 첫번째 생각
  • 먼저 입력으로 오는 배열을 오름차순으로 정렬해야지!
  • 그 배열은 오름차순으로 정렬되어있으니, 앞에서부터 차례대로 체크하면 되겠다!
  • [1, 2, 2, 2, 3, 3, 4, 6]을 예시로 들자. 1단계에 머물러있는 사람은 배열의 0번째 인덱스 1명이다. 2단계에 머물러있는 사람은 배열의 1, 2, 3번째 인덱스의 3명이다. 3단계는 배열의 4, 5번째 인덱스의 2명이다. 4단계는 배열의 6번째 인덱스의 1명이다. 5단계는 아무도 없다. 6단계는 이미 5단계를 통과했다는 뜻.
  • 위의 경우를 while 문에서 하나하나 처리하려니 머리가 터지고.. 하나의 예제를 처리하면 다른 예제가 터지고를 반복. 계속 이럴바에는 for문으로 처리하자..! ⇒ 두번째 생각
• 두번째 생각
  • 이중 for문을 이용해서 하나하나 처리하자
  • 스테이지 수만큼 for문을 돌리는데, 각 스테이지 별로 또 for문을 돌린다.
  • N = 5, stages = [1, 2, 2, 2, 3, 3, 4, 6]
  • currentStageUserCount = 해당 스테이지에 아직 머물러 있는 유저 수
  • userCount = 스테이지를 플레이한 유저 수 (아직 해당 스테이지를 진행 중인 유저도 포함)
  • 실패율 = currentStageUserCount / userCount
  
  

  스테이지	스테이지 배열	currentStageUserCount	userCount	실패율 (currentStageUserCount / userCount)
  스테이지 1 ⇒ 배열에서 1만 체크	[1, 2, 2, 2, 3, 3, 4, 6]	1	8	1/8
  스테이지 2 ⇒ 배열에서 2만 체크	[1, 2, 2, 2, 3, 3, 4, 6]	3	8 - 1 = 7	3/7
  스테이지 3 ⇒ 배열에서 3만 체크	[1, 2, 2, 2, 3, 3, 4, 6]	2	7 - 3 = 4	2/4
  스테이지 4 ⇒ 배열에서 4만 체크	[1, 2, 2, 2, 3, 3, 4, 6]	1	4 - 2 = 2	1/2
  스테이지 5 ⇒ 배열에서 5만 체크	[1, 2, 2, 2, 3, 3, 4, 6]	0	1	0/1 = 0

• 위와 같이 표를 만들고, for 문을 통해서 구현하였다. 그리고 제출하니 테스트 7을 제외한 모든 문제를 통과..! 테스트 7은 런타임 에러가 발생했다.
• 다른 분들의 질문을 찾아봤는데, 0으로 나누는 경우를 조심하라는 말이 있어서 그 부분을 따로 예외처리해주었다. 그랬더니 정답..!
  

  // 수정 전
  result[resultIndex].1 = Double(currentStageUserCount) / Double(userCount)
  
  
  // 수정 후
  if currentStageUserCount == 0 {
      result[resultIndex].1 = 0
  } else {
      result[resultIndex].1 = Double(currentStageUserCount) / Double(userCount)
  }

• 고작 Level1이지만,, 알린이인 나는 이걸 푸는데 많은 시간이 걸렸다..😭 그래도 스스로 풀었다는 것에 의의를 두기로!!!!! 이렇게 꾸준히 하다보면 언젠간 늘겠지,,🥺 알고리즘 매일매일 뿌셔뿌셔

소스 코드 (Swift)

//
//  실패율.swift
//  Programmers
//
//  Created by 안상희 on 2022/02/01.
//

import Foundation

func solution(_ N:Int, _ stages:[Int]) -> [Int] {
    var result = [(Int, Double)]()
    for i in 0..<N {
        result.append((i + 1, 0.0))
    }
    
    let userState = stages.sorted() // 1 2 2 2 3 3 4 6
    var userCount = stages.count // 스테이지를 플레이한 유저 수 (아직 해당 스테이지를 진행 중인 유저도 포함)
    var currentStageUserCount = 0 // 해당 스테이지에 아직 머물러 있는 유저 수
    var stateIndex = 0 // userState 배열에서 사용할 인덱스
    var resultIndex = 0 // result 배열에서 사용할 인덱스
    
    for i in 1...N { // 스테이지 체크
        for j in stateIndex..<stages.count {
            if userState[j] == i {
                currentStageUserCount += 1
                stateIndex += 1
            }
        }
        
        if currentStageUserCount == 0 {
            result[resultIndex].1 = 0
        } else {
            result[resultIndex].1 = Double(currentStageUserCount) / Double(userCount)
        }
        
        userCount -= currentStageUserCount // 유저 수 구하기
        resultIndex += 1 // 다음 스테이지
        currentStageUserCount = 0 // 초기화
    }

    return result.sorted { $0.1 > $1.1 }.map { $0.0 }
}