프로그래머스 입문 Day 21 파이썬

코딩테스트

프로그래머스 입문 Day 21 파이썬

피갓자 2025. 5. 17. 13:00

숨어있는 숫자의 덧셈 (2)

문제 설명

문자열 my_string이 매개변수로 주어집니다.

my_string은 소문자, 대문자, 자연수로만 구성되어있습니다.

my_string안의 자연수들의 합을 return하도록 solution 함수를 완성해주세요.

제한사항

1 ≤ my_string의 길이 ≤ 1,000
1 ≤ my_string 안의 자연수 ≤ 1000
연속된 수는 하나의 숫자로 간주합니다.
000123과 같이 0이 선행하는 경우는 없습니다.
문자열에 자연수가 없는 경우 0을 return 해주세요.

입출력 예

my_string	result
"aAb1B2cC34oOp"	37
"1a2b3c4d123Z"	133

입출력 예 설명

입출력 예 #1

"aAb1B2cC34oOp"안의 자연수는 1, 2, 34 입니다. 따서 1 + 2 + 34 = 37 을 return합니다.

입출력 예 #2

"1a2b3c4d123Z"안의 자연수는 1, 2, 3, 4, 123 입니다. 따라서 1 + 2 + 3 + 4 + 123 = 133 을 return합니다.

solution.py

def solution(my_string):
    answer = 0
    temp = ''
    
    for i in my_string:
        if i.isdigit():
            temp += i
        else:
            if temp:
                answer += int(temp)
                temp = ''
    
    if temp:
        answer += int(temp)
    
    return answer

안전지대

문제 설명

다음 그림과 같이 지뢰가 있는 지역과 지뢰에 인접한 위, 아래, 좌, 우 대각선 칸을 모두 위험지역으로 분류합니다.

지뢰는 2차원 배열 board에 1로 표시되어 있고 board에는 지뢰가 매설 된 지역 1과, 지뢰가 없는 지역 0만 존재합니다.
지뢰가 매설된 지역의 지도 board가 매개변수로 주어질 때, 안전한 지역의 칸 수를 return하도록 solution 함수를 완성해주세요.

제한사항

board는 n * n 배열입니다.
1 ≤ n ≤ 100
지뢰는 1로 표시되어 있습니다.
board에는 지뢰가 있는 지역 1과 지뢰가 없는 지역 0만 존재합니다.

입출력 예

board	result
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 0, 0], [0, 0, 0, 0, 0]]	16
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 1, 0], [0, 0, 0, 0, 0]]	13
[[1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1]]	0

입출력 예 설명

입출력 예 #1

(3, 2)에 지뢰가 있으므로 지뢰가 있는 지역과 지뢰와 인접한 위, 아래, 좌, 우, 대각선 총 8칸은 위험지역입니다. 따라서 16을 return합니다.

입출력 예 #2

(3, 2), (3, 3)에 지뢰가 있으므로 지뢰가 있는 지역과 지뢰와 인접한 위, 아래, 좌, 우, 대각선은 위험지역입니다. 따라서 위험지역을 제외한 칸 수 13을 return합니다.

입출력 예 #3

모든 지역에 지뢰가 있으므로 안전지역은 없습니다. 따라서 0을 return합니다.

문제 분석

의사 코드(pseudo code)
1 n × n 크기의 이차원 배열을 0으로 초기화
2 board의 모든 칸 순회
  a 현재 칸에 지뢰가 있는 경우
  b 인접한 8개 칸의 값 1로 변경(범위 안일 경우)
3 값이 0인 칸의 수 반환

solution.py

def solution(board):
    n = len(board)
    # n × n 배열 초기화
    answer = [[0] * n for _ in range(n)]
    
    # board의 모든 칸 순회
    for i in range(n):
        for j in range(n):
            # 현재 칸에 지뢰가 있는 경우
            if board[i][j] == 1:
                # 지뢰가 있는 칸의 인접한 8방향 검사
                for dx in [-1, 0, 1]:
                    for dy in [-1, 0, 1]:
                        # 인접 칸 좌표 계산
                        ni, nj = i + dx, j + dy
                        # 인접 칸이 board 범위 내에 있는지 확인
                        if 0 <= ni < n and 0 <= nj < n:
                            # 위험지역으로 표시
                            answer[ni][nj] = 1

    # 안전지대 개수 계산
    result = 0
    for i in range(n):
        for j in range(n):
            if answer[i][j] == 0:
                result += 1

    return result

복잡도 분석

n × n 배열의 모든 칸을 순회하는 과정의 시간 복잡도는 O(N^2)입니다.
n × n 크기의 배열이 생성되므로 공간 복잡도는 O(N^2)입니다.
i, j, dx, dy 등의 상수 크기 변수들의 공간 복잡도는 O(1)입니다.
따라서 최종 시간 복잡도와 공간 복잡도는 O(N^2)입니다.

삼각형의 완성조건 (2)

문제 설명

선분 세 개로 삼각형을 만들기 위해서는 다음과 같은 조건을 만족해야 합니다.

가장 긴 변의 길이는 다른 두 변의 길이의 합보다 작아야 합니다.

삼각형의 두 변의 길이가 담긴 배열 sides이 매개변수로 주어집니다. 나머지 한 변이 될 수 있는 정수의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항

sides의 원소는 자연수입니다.
sides의 길이는 2입니다.
1 ≤ sides의 원소 ≤ 1,000

입출력 예

sides	result
[1, 2]	1
[3, 6]	5
[11, 7]	13

입출력 예 설명

입출력 예 #1

두 변이 1, 2 인 경우 삼각형을 완성시키려면 나머지 한 변이 2여야 합니다. 따라서 1을 return합니다.

입출력 예 #2

가장 긴 변이 6인 경우
- 될 수 있는 나머지 한 변은 4, 5, 6 로 3개입니다.
나머지 한 변이 가장 긴 변인 경우
- 될 수 있는 한 변은 7, 8 로 2개입니다.
따라서 3 + 2 = 5를 return합니다.

입출력 예 #3

가장 긴 변이 11인 경우
- 될 수 있는 나머지 한 변은 5, 6, 7, 8, 9, 10, 11 로 7개입니다.
나머지 한 변이 가장 긴 변인 경우
- 될 수 있는 한 변은 12, 13, 14, 15, 16, 17 로 6개입니다.
따라서 7 + 6 = 13을 return합니다.

solution.py

def solution(sides):
    a, b = sorted(sides)

    case_1 = (b - a - 1)
    case_2 = (a + b - 1) - 1

    return case_1 + case_2

def solution(sides):
    a, b = sorted(sides)
    return (a + b - 1) - (abs(a - b) + 1) + 1

def solution(sides):
    return 2 * min(sides) - 1

외계어 사전

문제 설명

PROGRAMMERS-962 행성에 불시착한 우주비행사 머쓱이는 외계행성의 언어를 공부하려고 합니다.

알파벳이 담긴 배열 spell과 외계어 사전 dic이 매개변수로 주어집니다.

spell에 담긴 알파벳을 한번씩만 모두 사용한 단어가 dic에 존재한다면 1, 존재하지 않는다면 2를 return하도록 solution 함수를 완성해주세요.

제한사항

spell과 dic의 원소는 알파벳 소문자로만 이루어져있습니다.
2 ≤ spell의 크기 ≤ 10
spell의 원소의 길이는 1입니다.
1 ≤ dic의 크기 ≤ 10
1 ≤ dic의 원소의 길이 ≤ 10
spell의 원소를 모두 사용해 단어를 만들어야 합니다.
spell의 원소를 모두 사용해 만들 수 있는 단어는 dic에 두 개 이상 존재하지 않습니다.
dic과 spell 모두 중복된 원소를 갖지 않습니다.

입출력 예

spell	dic	result
["p", "o", "s"]	["sod", "eocd", "qixm", "adio", "soo"]	2
["z", "d", "x"]	["def", "dww", "dzx", "loveaw"]	1
["s", "o", "m", "d"]	["moos", "dzx", "smm", "sunmmo", "som"]	2

입출력 예 설명

입출력 예 #1

"p", "o", "s" 를 조합해 만들 수 있는 단어가 dic에 존재하지 않습니다. 따라서 2를 return합니다.

입출력 예 #2

"z", "d", "x" 를 조합해 만들 수 있는 단어 "dzx"가 dic에 존재합니다. 따라서 1을 return합니다.

입출력 예 #3

"s", "o", "m", "d" 를 조합해 만들 수 있는 단어가 dic에 존재하지 않습니다. 따라서 2을 return합니다.

유의사항

입출력 예 #3 에서 "moos", "smm", "som"도 "s", "o", "m", "d" 를 조합해 만들 수 있지만 spell의 원소를 모두 사용해야 하기 때문에 정답이 아닙니다.

solution.py

def solution(spell, dic):
    for i in dic:
        if sorted(i) == sorted(spell):
            return 1
    return 2