[Python] 백준 14923 미로탈출 & 프로그래머스 JadenCase 문자열 만들기

백준 14923

# 백준 14923 미로탈출

from collections import deque
n, m = map(int, input().split())
Hx, Hy = map(int, input().split())
Ex, Ey = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]
visited = [[[0] * 2 for _ in range(m)] for _ in range(n)]

d_row = [0, -1, 0, 1]
d_col = [1, 0, -1, 0]

def solution(Hx, Hy):
    queue = deque()
    x, y = Hx-1, Hy-1
    queue.append([x,y, 0])
    visited[x][y][0] = 0
    while queue:
        x, y, w = queue.popleft()
        if x == Ex-1 and y == Ey-1:
            return visited[x][y][w]
        for i in range(4):
            nx = d_row[i] + x
            ny = d_col[i] + y
            if 0<= nx < n and 0 <= ny < m :
                if graph[nx][ny] == 0 and visited[nx][ny][w]==0: # 다음노드가 0이고 아직 방문안했다면
                    queue.append([nx,ny,w])
                    visited[nx][ny][w] = visited[x][y][w]+1
                elif graph[nx][ny] == 1 and w == 0: # 다음 노드가 1(벽)이고 그 벽을 안 부신 상태라면
                    visited[nx][ny][w+1] = visited[x][y][w] + 1
                    queue.append([nx, ny, w+1])

    else:
        return -1

print(solution(Hx, Hy))

 

 

프로그래머스 JadenCase 문자열 만들기

 

내 코드

def solution(s):
    s = s.lower()
    new_list = []
    str_list = s.split(" ")
    for i in str_list:
        if not i[0:1].isdigit():
            a = i[0:1].upper()
            i = "".join([a, i[1:]])
        new_list.append(i)

    return " ".format(" ".join(new_list))

solution("3people unFollowed me")

 

더 쉬운 코드

def solution2(s):
    return ' '.join([word.capitalize() for word in s.split(" ")])

solution2("3people unFollowed me")

 

함수 .capitalize()를 쓰면 한방에 해결된다...ㅎ