본문 바로가기

파이썬45

[자료구조] 힙(heapq) 힙(heapq)이란? 완전 이진 트리의 일종으로 우선순위 큐를 위하여 만들어진 자료구조이다.  최소 힙(min heap) 루트 노드가 가장 작은 값을 가진다. 따라서 값이 작은 데이터가 우선적으로 제거된다. 즉, 부모 노드가 항상 자식 노드보다 작다. 최대 힙(max heap) 루트 노드가 가장 큰 값을 가진다. 따라서 값이 큰 데이터가 우선적으로 제거된다. 즉, 부모 노드가 항상 자식 노드보다 크다.파이썬에서 heapq 모듈을 사용하여 구현할 수 있다.import heapqq= []a = [4,9,6,11,8]for i in range(len(a)): heapq.heappush(q,a[i])print(q)#q = [4, 8, 6, 11, 9]for i in range(len(a)): prin.. 2024. 6. 24.
[Algorithm] 파이썬 정렬, lambda 사용 우선 파이썬에서 sort()와 sorted() 함수를 통해 정렬을 할 수 있다.sort() 원본 자체를 정렬시켜 준다.a = [1,5,2,7]a.sort() print(a)#a = [1,2,5,7] sorted() 원본을 변형시키지 않고 새로운 list를 반환한다.a = [1,5,2,7]b = sorted(a)print(a)print(b)#a = [1,5,2,7]#b = [1,2,5,7] reverse 정렬 기능은 기본적으로 오름차순으로 제공한다. 내림차순으로 정렬하고 싶은 경우, reverse 파라미터를 통해 정렬할 수 있다.a = [1,5,2,7]a.sort(reverse=True)print(a)#a = [7,5,2,1] key 정렬 조건을 설정할 함수를 넣을 수 있다.a = ['love','i','.. 2024. 6. 20.
[Algorithm] 백준 1931 회의실 배정 | 파이썬 ※ 1931 회의실 배정https://www.acmicpc.net/problem/1931 문제 해결 TIP처음에는 회의 시간이 짧은 순서대로 배정을 하려 했지만, 그렇게 되면 시각 시간과 끝나는 시간을 고려하지 못하여 비효율적이다. 회의를 최대한 많이 하기 위해서는 최대한 회의가 빨리 끝나야하므로 끝나는 시간을 오름차순으로 정렬하여 배정해야한다.  전체 코드N = int(input())data = []count = 1for i in range(N): a,b = map(int,input().split()) data.append([a,b])data.sort(key=lambda x:(x[1],x[0]))end = data[0][1]for i in range(1,N): if data[i][0]>.. 2024. 6. 20.
[Algorithm] 백준 11047 동전 0 | 파이썬 ※ 11047 동전 0https://www.acmicpc.net/problem/11047  문제 해결 TIP거스름돈 문제와 로직은 비슷하다. 가지고 있는 동전의 종류를 리스트에 내림차순으로 담은 후, 하나씩 나눠가며 총액에서 뺀다. 전체 코드N, K = map(int,input().split())data = []result = 0for i in range(N): data.append(int(input()))data.sort(reverse=True)for i in data: result += K//i K = K - (K//i)*iprint(result) 2024. 6. 20.
[Algorithm] 백준 2839 설탕 배달 | 파이썬 ※ 2839 설탕 배달https://www.acmicpc.net/problem/2839문제 해결 TIP나올 수 있는 경우의 수는 총 4가지이다. 5로 나누어 떨어지는 경우, 3으로 나누어 떨어지는 경우, 3과 5의 조합으로 나누어 떨어지는 경우, 나누어 떨어지지 않는 경우. 처음에는 모든 경우의 수를 if else 문으로 하려 했지만, 조금 더 효율적인 방법을 생각해보았다. 우선 N을 5로 나누어보고 나누어 떨어지지 않는다면 3만큼 한번씩 뺴본 후 다시 5로 나누어보는 방법이다. 이렇게 로직을 짜면 모든 경우를 만족할 수 있게 된다. 전체 코드N = int(input())result = 0while N>=0: if N%5==0: result += N//5 print(resu.. 2024. 6. 20.
[Algorithm] 백준 11399 ATM | 파이썬 ※ 11399 ATMhttps://www.acmicpc.net/problem/11399 문제 해결 TIP최소  시간의 합을 구하기 위해서는 우선 돈을 뽑는 시간 리스트를 오름차순으로 정렬해야 한다. ex) 1, 2, 3, 3, 4그렇다면 1은 5명 모두에게 해당되는 시간이므로 전체시간에 1*5을 더해주고, 2는 4명에게 해당되는 시간이므로 2*4, 3*3, 3*2, 4*1을 순차적으로 더해주면 된다. 전체 코드N = int(input())data = list(map(int,input().split()))data.sort()result = 0for i in range(len(data)): result += data[i]*(N-i)print(result) 2024. 6. 20.
[Algorithm] Backtracking(백트래킹) 순열 및 조합 구현하기 itertools 우선 순열과 조합의 차이를 알아보자. 순열 : 원소들의 순서를 고려하여 나열하는 것 (원소의 순서가 다르면 다른 순열로 취급) 조합 : 원소들의 순서를 고려하지 않고 나열하는 것 (선택된 원소의 순서는 고려되지 않음) 파이썬 라이브러리 중 하나인 itertools를 사용하면 조합과 순열을 쉽게 구할 수 있다. from itertools import permutations, combinations arr = [1,2,3] k = 2 npr = list(permutations(arr, 2)) # [(1, 2), (1, 3), (2, 1), (2, 3), (3, 1), (3, 2)] ncr = list(combinations(arr, 2)) # [(1, 2), (1, 3), (2, 3)] 그러나.. 라이브러리.. 2024. 4. 13.
[Algorithm] 프로그래머스 타겟 넘버 | 파이썬 ※ 타겟 넘버 https://school.programmers.co.kr/learn/courses/30/lessons/43165?language=python3 프로그래머스 코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요. programmers.co.kr 문제 해결 TIP DFS로 구현한다. 숫자 한 개에 대해 음수, 양수 2가지 경우의 수를 각각 뻗어나가며 모든 경우의 수 중에, 마지막 결과가 타겟과 같을 때 방법 하나를 추가하는 식으로 진행한다. 전체 코드 answer = 0 def dfs(numbers, target, result, idx): global answer # 깊이가 같고 마지막 결과가 .. 2024. 3. 26.
[Algorithm] 백준 1260 DFS와 BFS | 파이썬 ※ 1260 DFS와 BFS https://www.acmicpc.net/problem/1260 1260번: DFS와 BFS 첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사 www.acmicpc.net 문제 해결 TIP DFS는 스택으로, BFS는 큐로 구현한다. 문제 해결 풀이 그래프 리스트에 각 노드가 연결된 정보를 표현한다. 전체 코드 from collections import deque N, M, V = map(int,input().split()) graph=[[] for _ in range(N+1)] for i in ra.. 2024. 1. 17.