꾸준한 연습장

[Lv.2 / 프로그래머스 / 파이썬] 교점에 별 만들기

김세진 — Tue, 1 Jul 2025 09:15:44 +0900

프로그래머스

SW개발자를 위한 평가, 교육의 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

주어진 직선들의 교점을 구해, 그 자리에 별을 찍어 출력해야 한다.

참고 사항을 확인하여 교점을 구하는 방법을 알 수 있다.

정수로 표현되는 교점에만 별을 찍어야 함에 유의해야 한다.

우선 모든 교점을 다 구한 다음, 교점 좌표의 x,y별 min, max값을 구해 그 크기만큼의 배열을 생성한다.

이제 구한 교점들을 순회하며 좌표를 배열의 인덱스에 맞게끔 민맥싱하여 별을 찍어주면 된다.

def solution(line):
    N = len(line)
    cand = set()
    
    min_x = min_y = float('inf')
    max_x = max_y = -float('inf')
    
    # 모든 직선을 서로 비교
    for i in range(N):
        for j in range(i+1, N):
            point = get_xy(line[i], line[j])
            if not point:
                continue
            
            x, y = point
            min_x = min(min_x, x); max_x = max(max_x, x)
            min_y = min(min_y, y); max_y = max(max_y, y)
            
            cand.add(point)
    
    # 필요한 넓이만큼 좌표 평면 생성
    board = [["."] * (max_x - min_x+ 1) for _ in range(max_y - min_y  + 1)]
    
    # 별 찍기
    for x, y in cand:
        board[max_y - y][x - min_x] = '*'
    
    return [''.join(row) for row in board]
            

def get_xy(l1, l2):
    a, b, e = l1
    c, d, f = l2
    
    # 두 직선이 평행 또는 일치인 경우
    if a*d - b*c == 0:
        return None
    
    x = (b*f - e*d) / (a*d - b*c)
    y = (e*c - a*f) / (a*d - b*c)
    
    # 나누어 떨어지지 않는 경우
    if x != int(x) or y != int(y):
        return None
    
    return (int(x), int(y))

[Lv.2 / 프로그래머스 / 파이썬] 모음사전

김세진 — Fri, 13 Jun 2025 17:31:27 +0900

프로그래머스

SW개발자를 위한 평가, 교육의 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

사전 정렬 규칙에 따라 단어를 순서대로 만들었을 때, 주어진 단어의 번호를 출력해야 한다.

다음 단어를 만드는 법은 다음과 같다.

1. 'A'를 붙인다.

2. 끝 글자를 다음 순서의 모음으로 교체한다.

만약 끝 글자가 U일 경우, 그 다음 순서의 모음을 교체해야 하므로, U를 제거한 뒤 다음 순서로 넘어가면 된다.

def solution(word):
    alpha = ['A', 'E', 'I', 'O', 'U']
    
    cur_word = 'A'
    cnt = 1
    
    while cur_word != word:
        cnt += 1
        
        # 현재 단어 길이가 5보다 작으면 'A'를 붙임
        if len(cur_word) < 5:
            cur_word += 'A'
            continue
        
        # 끝 글자가 'U' 면 계속 제거
        while cur_word[-1] == 'U':
            cur_word = cur_word[:-1]
        
        # 끝 글자를 다음 순서의 글자로 바꿈
        cur_idx = alpha.index(cur_word[-1])
        cur_word = cur_word[:-1] + alpha[(cur_idx+1)%5]
        
    return cnt

[Lv.2 / 프로그래머스 / 파이썬] 무인도 여행

김세진 — Fri, 30 May 2025 17:02:18 +0900

프로그래머스

SW개발자를 위한 평가, 교육의 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

연결된 땅들의 크기를 계산하여 각 섬의 크기를 오름차순으로 출력하는 문제이다.

DFS 및 BFS를 사용하여 해결할 수 있다.

작성된 코드의 알고리즘 순서는 다음과 같다.

맵을 탐색하며 X가 아닌 부분(섬의 일부)을 찾는다.
섬의 일부를 발견한 경우, BFS로 섬 전체를 탐색한다.
1. 섬을 방문한 좌표의 경우 X로 표시로 체크하여 재방문을 막는다.
섬 탐색이 끝나면 현재 섬의 크기를 저장한 다음, 맵을 이어서 탐색하며 다른 섬들을 찾는다.
맵 탐색이 전부 끝나면, 저장된 섬의 크기를 오름차순 정렬한 다음 반환한다.

def solution(maps):
    board = [list(m) for m in maps]
    move = [(0, 1), (1, 0), (0, -1), (-1, 0)]
    N, M = len(board), len(board[0])
    res = []
    
    # 맵 탐색
    for i in range(N):
        for j in range(M):
            if board[i][j] == 'X':
                continue
            
            # 섬의 일부를 발견했으므로 섬 탐색
            q = [(i, j)]
            size = 0
            while q:
                x, y = q.pop()
                
                # 방문 체크
                if board[x][y] == 'X':
                    continue
                size += int(board[x][y])
                board[x][y] = 'X'
                
                # 4방향 탐색
                for a, b in move:
                    dx, dy = x+a, y+b
                    if dx < 0 or dx >= N or dy < 0 or dy >= M or board[dx][dy] == 'X':
                        continue
                    
                    q.append((dx, dy))
            else:
                res.append(size)
    
    return sorted(res) if res else [-1]

[Lv.2 / 프로그래머스 / 파이썬] 지게차와 크레인 (2025 프로그래머스 코드챌린지 1차 예선)

김세진 — Fri, 9 May 2025 14:46:31 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

requests를 순회하며 처리하는 시뮬레이션에서 BFS를 수행하는 문제이다.

BFS를 가장자리부터 진행하도록 하기 위해 빈 값으로 storage를 래핑해주었다.

그리고 주어진 요청을 순회하며 BFS 혹은 전체 삭제를 진행한다.

순회가 종료되면, 남은 블럭의 개수를 리턴한다.

BFS를 진행할 때, 현재 요청 타입에 맞는 블럭과 만났을 때 즉시 삭제하지 않도록 따로 삭제 예정 리스트를 구성한 다음 처리해야 한다. 그렇지 않으면 같은 타입의 블럭 뭉치를 만났을 때, 연결된 모든 부분을 삭제하게 된다.

def solution(storage, requests):
    # 빈 값으로 storage 래핑
    empty_row = ["-" * len(storage[0])]
    storage = empty_row + storage + empty_row
    storage = [list("-" + row + "-") for row in storage]
    
    N, M = len(storage), len(storage[0])
    move = [(1,0), (0,1), (-1,0), (0,-1)]
    
    # 요청 순회
    for req in requests:
        t = req[0]
        
        # 알파벳 2개 요청인 경우 전체 삭제
        if len(req) == 2:
            for i in range(N):
                for j in range(M):
                    if storage[i][j] == t:
                        storage[i][j] = "-"
        else: # 알파벳 1개 요청인 경우, 바깥에서 접근 가능한 부분만 삭제
            visited = [[False] * M for _ in range(N)]
            q = [(0, 0)]
            # BFS
            del_list = []
            while q:
                # 방문 체크
                x, y = q.pop()
                if visited[x][y]:
                    continue
                visited[x][y] = True
                
                # 탐색
                for a, b in move:
                    dx, dy = x+a, y+b
                    # index out of range 방지
                    if dx >= N or dx < 0 or dy >= M or dy < 0:
                        continue
                
                    # 타입 비교 후 제거 목록에 추가
                    if storage[dx][dy] == t:
                        del_list.append((dx, dy))
                    
                    # 다음 탐색 후보지 추가
                    if visited[dx][dy] or storage[dx][dy] != "-":
                        continue
                    
                    q.append((dx, dy))
            
            # 컨테이너 제거
            for x, y in del_list:
                storage[x][y] = "-"
                
    # 결과 반환
    result = 0
    for i in range(N):
        for j in range(M):
            if storage[i][j] != "-":
                result += 1
    return result

[Lv.2 / 프로그래머스 / 파이썬] 귤 고르기

김세진 — Fri, 2 May 2025 17:28:58 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

귤의 크기별로 그 개수를 센 딕셔너리를 구성한다.
그리고 개수를 기준으로 내림차순 정렬한다.
개수가 가장 많은 종류부터 선택하여 k에서 선택한 종류의 개수를 뺀다.
k가 남았다면 3번을, k가 0이거나 음수가 되면 목표치를 달성한 것이므로 지금까지 선택한 종류의 수를 return 한다.

def solution(k, tangerine):
    # 딕셔너리 구성
    d = dict()
    for i in tangerine:
        if i in d:
            d[i] += 1
        else:
            d[i] = 1
    
    # 내림차순 정렬
    lst = [i[1] for i in sorted(d.items(), key=lambda x:-x[1])]
    
    # 개수가 많은 귤 부터 상자에 담음
    res = 0
    for i in lst:
        res += 1 # 종류 + 1
        k -= i # 귤의 개수만큼 k를 감소
        if k <= 0: # k를 달성했으므로 res return
            return res

[Lv.2 / 프로그래머스 / 파이썬] 비밀 코드 해독 (2025 프로그래머스 코드챌린지 1차 예선)

김세진 — Fri, 11 Apr 2025 18:22:48 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

1부터 n까지 5개의 숫자로 만들 수 있는 정수 조합들을 모두 탐색하며 해결해야 하는 문제이다.
조합을 만든 뒤, q랑 비교해보며 ans 와 같은 결과가 나오는 조합이면 비밀 코드에 해당된다.

itertools를 활용하면 조합을 보다 쉽게 생성할 수 있다.

import itertools

def solution(n, q, ans):
    # 배열의 전체 조합
    arrs = itertools.combinations([i for i in range(1, n+1)], 5)
    
    # 배열을 순회하며 탐색
    res = 0
    for arr in arrs:
        # 현재 조합과 q를 비교하며 ans와 일치하는지 판별
        for i in range(len(q)):
            cur_q = q[i]
            cnt = 0
            for num in arr:
                if num in cur_q:
                    cnt += 1
            if ans[i] != cnt:
                break
        else:
            res += 1
    return res

[Lv.1 / 프로그래머스 / 파이썬] 유연근무제 (2025 프로그래머스 코드챌린지 1차 예선)

김세진 — Mon, 24 Mar 2025 14:13:39 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

시간 체크를 쉽게 하기 위해 모든 시간을 분으로 바꿔 계산하도록 한다.

또한 요일 체크를 쉽게 하기 위해 나머지 연산을 사용하면 좋다.

def solution(schedules, timelogs, startday):
    result = 0
    
    # 직원 순회
    for i in range(len(schedules)):
        schedule = convert(schedules[i]) + 10
        timelog = timelogs[i]
        cur_day = startday - 1
        
        # 출근 기록 순회
        for log in timelog:
            cur_day += 1
            
            # 토, 일은 제외
            if cur_day % 7 in [0, 6]:
                continue
            
            # 지각 체크
            if convert(log) > schedule:
                break
        else:
            result += 1
        
    return result
            
def convert(time):
    time = str(time)
    h, m = time[:-2], time[-2:]
    return int(h)*60 + int(m)

[Lv.1 / 프로그래머스 / 파이썬] 택배 상자 꺼내기 (2025 프로그래머스 코드챌린지 2차 예선)

김세진 — Mon, 17 Feb 2025 01:23:36 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

주어진 조건대로 상자를 쌓은 뒤, 해당 지점에서부터 몇 층의 박스가 쌓여있는지 확인하면 된다.

def solution(n, w, num):
    # 박스 정리
    MAX_ROW = n//w+1
    boxes = []
    for row in range(MAX_ROW):
        boxes.append([row*w + i if row*w + i <= n else 0 for i in range(1, w+1)][::-1 if row%2==1 else 1])
    
    # 박스 탐색
    for r in range(MAX_ROW):
        for c in range(w):
            if boxes[r][c] != num:
                continue
            
            # 박스 발견
            res = 0
            for i in range(MAX_ROW):
                res += 1
                r+=1
                if r >= MAX_ROW:
                    return res
                if boxes[r][c] == 0:
                    return res

[Lv.2 / 프로그래머스 / 파이썬] 충돌위험 찾기 (PCCP 기출문제 3번)

김세진 — Tue, 11 Feb 2025 01:14:32 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

입출력 예 #1

입출력 예 #2

입출력 예 #3

딕셔너리를 이용하여 해결 가능한 시뮬레이션 구현 문제이다.
문제 풀이 시 특별하게 필요한 알고리즘은 존재하지 않으나, 주어진 조건에 맞게 정확하게 구현해야 한다.

def solution(points, routes):
    # 번호별로 좌표를 map에 담음
    point_map = dict()
    for i in range(len(points)):
        point_map[i+1] = points[i]
    
    robots = []
    for route in routes:
        start_x, start_y = point_map.get(route[0])
        robots.append((start_x, start_y, route[::-1]))
    
    cnt = 0
    while robots:
        tmp_robots = []
        robots_pos = dict()
        for robot in robots:
            x, y, route = robot
            # 로봇 충돌 계산을 위해 현재 위치의 로봇 count 추가
            if (x, y) in robots_pos:
                robots_pos[(x, y)] += 1
            else:
                robots_pos[(x, y)] = 1
            
            # 현재 위치가 목표에 도달한 경우, 목표 경로 제거
            if [x, y] == point_map.get(route[-1]):
                route.pop()
            
            # 다음 경로가 남은 경우
            if route:
                to_x, to_y = point_map.get(route[-1])
                x, y = move(x, y, to_x, to_y)
                tmp_robots.append((x, y, route))
        
        # 충돌 위험 계산
        cnt += sum([1 if i >= 2 else 0 for i in robots_pos.values()])
        robots = tmp_robots
    
    return cnt
    
    
def move(x, y, to_x, to_y):
    if x > to_x:
        return (x-1, y)
    elif x < to_x:
        return (x+1, y)
    
    if y > to_y:
        return (x, y-1)
    elif y < to_y:
        return (x, y+1)
    
    return (x, y)

[Lv.1 / 프로그래머스 / 파이썬] 데이터 분석 (PCCE 기출문제 10번)

김세진 — Thu, 30 Jan 2025 20:25:09 +0900

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

리스트에 담겨 있는 자료를 조건에 따라 적절히 필터링하고, 정렬하여 출력하는 문제이다.

sort 함수의 key 파라미터로 람다 함수를 넘겨 보다 쉽게 정렬할 수 있다.

def solution(data, ext, val_ext, sort_by):
    idx = ["code", "date", "maximum", "remain"]
    data = [d for d in data if d[idx.index(ext)] < val_ext]
    data.sort(key = lambda x: x[idx.index(sort_by)])
    return data