[Lv. 0] 코드 처리하기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181932

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

문자열 code가 주어집니다.
code를 앞에서부터 읽으면서 만약 문자가 "1"이면 mode를 바꿉니다. mode에 따라 code를 읽어가면서 문자열 ret을 만들어냅니다.

• mode가 0일 때
  • code[idx]가 "1"이 아니면 idx가 짝수일 때만 ret의 code[idx]를 추가합니다.
  • code[idx]가 "1"이면 mode를 0에서 1로 바꿉니다.
• mode가 1일 때
  • code[idx]가 "1"이 아니면 idx가 홀수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
  • code[idx]가 "1"이면 mode를 1에서 0으로 바꿉니다.

문자열 code를 통해 만들어진 문자열 ret를 return 하는 solution 함수를 완성해 주세요.

단, 시작할 때 mode는 0이며, return 하려는 ret가 만약 빈 문자열이라면 대신 "EMPTY"를 return 합니다.

 

제한 사항 : 

• 1 ≤ code의 길이 ≤ 100,000
  • code는 알파벳 소문자 또는 "1"로 이루어진 문자열입니다.

입출력 예 :

code	result
"abc1abc1abc"	"acbac"

풀이 : 

class Solution {
    fun solution(code: String): String {
        var answer: String = ""
        var mode = false

        for (i in code.indices) {
            if (!mode) {
                if (code[i] == '1') {
                    mode = true
                } else {
                    if (i % 2 == 0) {
                        answer += code[i]
                    }
                }
            } else {
                if (code[i] == '1') {
                    mode = false
                } else {
                    if (i % 2 == 1) {
                        answer += code[i]
                    }
                }
            }
        }

        return answer.ifEmpty { "EMPTY" }
    }
}