프로그래머스 기초문제

[프로그래머스 기초문제] Day 5

Devleoper_yh 2025. 1. 9. 00:10
반응형

1. 코드 처리하기

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

mode는 0과 1이 있으며, idx를 0 부터 code의 길이 - 1 까지 1씩 키워나가면서 code[idx]의 값에 따라 다음과 같이 행동합니다.

  • 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"

입출력 예 설명

입출력 예 #1

  • code의 각 인덱스 i에 따라 다음과 같이 mode와 ret가 변합니다.
i code[i] mode ret
0 "a" 0 "a"
1 "b" 0 "a"
2 "c" 0 "ac"
3 "1" 1 "ac"
4 "a" 1 "ac"
5 "b" 1 "acb"
6 "c" 1 "acb"
7 "1" 0 "acb"
8 "a" 0 "acba"
9 "b" 0 "acba"
10 "c" 0 "acbac"

따라서 "acbac"를 return 합니다. 

1. 기본 구조

func solution(_ code: String) -> String {
    var mode = 0 // mode는 처음에 0으로 시작
    var ret = "" // 결과 문자열
    return ret
}

 

2. code를 읽어야하기 때문에 순회하도록 for문을 사용한다.

func solution(_ code: String) -> String {
    var mode = 0 // mode는 처음에 0으로 시작
    var ret = "" // 결과 문자열
    let characters = Array(code) // 문자열을 배열로 변환

    for (idx, char) in characters.enumerated() {
        // 여기서 char와 idx를 처리
    }

    return ret
}

 

3. 문자가 "1" 이면 mode를 바꾼다.

func solution(_ code: String) -> String {
    var mode = 0 // mode는 처음에 0으로 시작
    var ret = "" // 결과 문자열
    let characters = Array(code) // 문자열을 배열로 변환

    for (idx, char) in characters.enumerated() {
    if char == "1" {
        mode = 1 - mode // mode를 반대로 변경
    } else {
        // mode에 따라 동작 처리
    }

    return ret
}

 

4. mode 값에 따라 ret에 조건 문자 더하기

func solution(_ code: String) -> String {
    var mode = 0 // mode는 처음에 0으로 시작
    var ret = "" // 결과 문자열
    let characters = Array(code) // 문자열을 배열로 변환

    for (idx, char) in characters.enumerated() {
    if char == "1" {
        mode = 1 - mode // mode를 반대로 변경
    } else {
        // mode에 따라 동작 처리
        if mode == 0 && idx % 2 == 0 {
            ret.append(char)
        } else if mode == 1 && idx % 2 == 1 {
            ret.append(char)
        }
    }

    return ret
}

 

5. ret이 빈 문자열일 경우 "EMPTY" return

func solution(_ code: String) -> String {
    var mode = 0 // mode는 처음에 0으로 시작
    var ret = "" // 결과 문자열
    let characters = Array(code) // 문자열을 배열로 변환

    for (idx, char) in characters.enumerated() {
    if char == "1" {
        mode = 1 - mode // mode를 반대로 변경
    } else {
        // mode에 따라 동작 처리
        if mode == 0 && idx % 2 == 0 {
            ret.append(char)
        } else if mode == 1 && idx % 2 == 1 {
            ret.append(char)
        }
    }

    return ret.isEmpty ? "EMPTY" : ret
}

 

enumerated() 참고 

https://developer.apple.com/documentation/swift/array/enumerated()

 

enumerated() | Apple Developer Documentation

Returns a sequence of pairs (n, x), where n represents a consecutive integer starting at zero and x represents an element of the sequence.

developer.apple.com


2. 등차수열의 특정한 항만 더하기

두 정수 a, d와 길이가 n인 boolean 배열 included가 주어집니다. 첫째항이 a, 공차가 d인 등차수열에서 included[i]가 i + 1항을 의미할 때, 이 등차수열의 1항부터 n항까지 included가 true인 항들만 더한 값을 return 하는 solution 함수를 작성해 주세요.

제한사항

  • 1 ≤ a ≤ 100
  • 1 ≤ d ≤ 100
  • 1 ≤ included의 길이 ≤ 100
  • included에는 true가 적어도 하나 존재합니다.

입출력 예

a d included result
3 4 [true, false, false, true, true] 37
7 1 [false, false, false, true, false, false, false] 10

입출력 예 설명

입출력 예 #1

  • 예제 1번은 a와 d가 각각 3, 4이고 included의 길이가 5입니다. 이를 표로 나타내면 다음과 같습니다.

||1항|2항|3항|4항|5항|

|-|-|---|---|---|---|

|등차수열|3|7|11|15|19|

|included|true|false|false|true|true|

따라서 true에 해당하는 1항, 4항, 5항을 더한 3 + 15 + 19 = 37을 return 합니다.

입출력 예 #2

  • 예제 2번은 a와 d가 각각 7, 1이고 included의 길이가 7입니다. 이를 표로 나타내면 다음과 같습니다.

||1항|2항|3항|4항|5항|6항|7항|

|-|-|---|---|---|---|---|---|

|등차수열|7|8|9|10|11|12|13|

|included|false|false|false|true|false|false|false|

따라서 4항만 true이므로 10을 return 합니다.

 

1. 배열 순회, 조건이 true의 경우만 계산

func solution(_ a: Int, _ d: Int, _ included: [Bool]) -> Int {
    var sum = 0

    for (index, isIncluded) in included.enumerated() {
        // index와 해당 값(isIncluded)을 가져옴
    }

    return sum
}

 

2. 등차수열 계산

let s1 = a + index * d

 

3. isIncluded if 조건

func solution(_ a: Int, _ d: Int, _ included: [Bool]) -> Int {
    var sum = 0

    for (index, isIncluded) in included.enumerated() {
        if isIncluded {
   			let s1 = a + index * d // // 등차수열의 (index + 1)번째 항 계산
   			sum += s1
		}
    }

    return sum
}

3. 주사위 게임 2

1부터 6까지 숫자가 적힌 주사위가 세 개 있습니다. 세 주사위를 굴렸을 때 나온 숫자를 각각 a, b, c라고 했을 때 얻는 점수는 다음과 같습니다.

  • 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
  • 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a2 + b2 + c2 )점을 얻습니다.
  • 세 숫자가 모두 같다면 (a + b + c) × (a2 + b2 + c2 ) × (a3 + b3 + c3 )점을 얻습니다.

세 정수 a, b, c가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

 

제한사항

  • a, b, c는 1이상 6이하의 정수입니다.

입출력 예

a b c result
2 6 1 9
5 3 3 473
4 4 4 110592

입출력 예 설명

입출력 예 #1

  • 예제 1번에서 세 주사위 숫자가 모두 다르므로 2 + 6 + 1 = 9점을 얻습니다. 따라서 9를 return 합니다.

입출력 예 #2

  • 예제 2번에서 두 주사위 숫자만 같으므로 (5 + 3 + 3) × (52 + 32 + 32 ) = 11 × 43 = 473점을 얻습니다. 따라서 473을 return 합니다.

입출력 예 #3

  • 예제 3번에서 세 주사위 숫자가 모두 같으므로 (4 + 4 + 4) × (42 + 42 + 42 ) × (43 + 43 + 43 ) = 12 × 48 × 192 = 110,592점을 얻습니다. 따라서 110592를 return 합니다.

제출 답안

func solution(_ a: Int, _ b: Int, _ c: Int) -> Int {
    let sum = a + b + c
    let s1 = a * a + b * b + c * c
    let s2 = a * a * a + b * b * b + c * c * c

    if a == b && b == c {
        // 세 숫자가 모두 같음
        return sum * s1 * s2
    } else if a == b || b == c || a == c {
        // 두 숫자만 같음
        return sum * s1
    } else {
        // 모두 다름
        return sum
    }
}

4. 원소들의 곱과 합

정수가 담긴 리스트 num_list가 주어질 때, 모든 원소들의 곱이 모든 원소들의 합의 제곱보다 작으면 1을 크면 0을 return하도록 solution 함수를 완성해주세요.

제한사항

  • 2 ≤ num_list의 길이 ≤ 10
  • 1 ≤ num_list의 원소 ≤ 9

입출력 예

num_list result
[3, 4, 5, 2, 1] 1
[5, 7, 8, 3] 0

입출력 예 설명

입출력 예 #1

  • 모든 원소의 곱은 120, 합의 제곱은 225이므로 1을 return합니다.

입출력 예 #2

  • 모든 원소의 곱은 840, 합의 제곱은 529이므로 0을 return합니다.

제출 답안

func solution(_ num_list: [Int]) -> Int {
    let s1 = num_list.reduce(1, *) // 모든 원소의 곱
    let s2 = num_list.reduce(0, +) // 모든 원소의 합
    let s3 = s2 * s2 // 합의 제곱

    return s1 < s3 ? 1 : 0
}

5. 이어 붙인 수

정수가 담긴 리스트 num_list가 주어집니다. num_list의 홀수만 순서대로 이어 붙인 수와 짝수만 순서대로 이어 붙인 수의 합을 return하도록 solution 함수를 완성해주세요.

제한사항

  • 2 ≤ num_list의 길이 ≤ 10
  • 1 ≤ num_list의 원소 ≤ 9
  • num_list에는 적어도 한 개씩의 짝수와 홀수가 있습니다.

입출력 예

num_list result
[3, 4, 5, 2, 1] 393
[5, 7, 8, 3] 581

입출력 예 설명

입출력 예 #1

  • 홀수만 이어 붙인 수는 351이고 짝수만 이어 붙인 수는 42입니다. 두 수의 합은 393입니다.

입출력 예 #2

  • 홀수만 이어 붙인 수는 573이고 짝수만 이어 붙인 수는 8입니다. 두 수의 합은 581입니다.

chatGPT 참고..

func solution(_ num_list: [Int]) -> Int {
    let oddString = num_list.filter { $0 % 2 != 0 }.map { String($0) }.joined()
    let evenString = num_list.filter { $0 % 2 == 0 }.map { String($0) }.joined()

    let oddNumber = Int(oddString) ?? 0
    let evenNumber = Int(evenString) ?? 0

    return oddNumber + evenNumber
}


1.	홀수 필터링:
	•	filter { $0 % 2 != 0 }를 사용해 리스트에서 홀수만 추출합니다.
	•	map { String($0) }로 각 숫자를 문자열로 변환합니다.
	•	joined()를 사용해 문자열로 이어붙입니다.
2.	짝수 필터링:
	•	filter { $0 % 2 == 0 }를 사용해 리스트에서 짝수만 추출합니다.
	•	이후 처리 방식은 홀수와 동일합니다.
3.	문자열을 정수로 변환:
	•	Int(oddString) 및 Int(evenString)을 사용해 문자열을 정수로 변환합니다.
	•	변환이 실패할 경우, 디폴트 값으로 0을 사용합니다.
4.	결과 계산:
	•	홀수와 짝수로 만들어진 두 정수를 더한 값을 반환합니다.

반응형