[프로그래머스 Python] 코딩 기초 트레이닝 Day 5 - 조건문

---

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”로 이루어진 문자열입니다.

코드 작성

def solution(code):
    mode = False
    ret = ''
    for index, char in enumerate(code):
        if char == '1': 
            mode = not mode
        else: 
            ret += char if index % 2 == int(mode) else ''
    return ret if len(ret) else 'EMPTY'

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가 적어도 하나 존재합니다.

코드 작성

def solution(a, d, included):
    new_list = []
    new_list.append(a)
    for i in range(len(included)-1):
        new_list.append(new_list[-1]+d)
        
    answer = 0
    for j, k in zip(new_list, included):
        if k:
            answer += j
    return answer

3. 주사위 게임 2

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

• 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지 다른 숫자는 다르다면 (a + b + c) × (a^2 + b^2 + c^2)점을 얻습니다.
• 세 숫자가 모두 같다면 (a + b + c) × (a^2 + b^2 + c^2) × (a^3 + b^3 + c^3)점을 얻습니다.

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

제한사항

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

코드 작성

def solution(a, b, c):
    if a != b and a != c and b != c:
        return a+b+c
    elif (a == b and a != c) or (a == c and a != b) or (b == c and b != a):
        return (a+b+c) * (a**2 + b**2 + c**2)
    return (a+b+c) * (a**2 + b**2 + c**2) * (a**3 + b**3 + c**3)

4. 원소들의 곱과 합

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

제한사항

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

코드 작성

def solution(num_list):
    multi = num_list[0]
    add = num_list[0]
    for num in num_list[1:]:
        multi *= num
        add += num
    return 1 if multi < add**2 else 0

5. 이어 붙인 수

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

제한사항

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

코드 작성

def solution(num_list):
    even = ''
    odd = ''
    for num in num_list:
        if num % 2 == 0:
            even += str(num)
        else:
            odd += str(num)
    return int(even) + int(odd)

6. 문제 출처

• 코드 처리하기
• 등차수열의 특정한 항만 더하기
• 주사위 게임 2
• 원소들의 곱과 합
• 이어 붙인 수

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.