haejun IT

문제 1~N까지 번호 적힌 구슬 중복을 허락하여 M번 뽑아 일렬로 나열하는 방법을 모두 출력 my_solution from itertools import product n, m = map(int, input().split()) lists = [i for i in range(1, n + 1)] cnt = 0 for i in product(lists, repeat=m): print(*i, sep=" ") cnt += 1 print(cnt) 중복을 허락하여 모든 조합을 나열해야하기에 product 사용 soltuion n, m = map(int, input().split()) res = [0] * m cnt = 0 def recursive(level): global cnt if level == m: for j..

문제 포인트 자연수 n과 n개의 원소로 구성된 자연수 집합이 주어짐 자연수 집합을 두개의 부분집합으로 나누었을 때, 두 부분집합의 합이 서로 같은 경우가 존재하면 'YES'출력, 그렇지 않으면 'NO' 출력 둘로 나뉘는 두 부분집합은 서로소 집합 두 부분집합을 합했을 때, 주어진 원래의 집합이 되어야 한다. e.g) {1, 3, 5, 6, 7, 10} -> {1, 3, 5, 7} = {6, 10} 이므로 'YES' 출력 n = int(input()) nums = list(map(int, input().split())) total = sum(nums) def recursive(level, sum): if sum > (total // 2): # 시간 복잡도 줄이기...

문제 포인트 최소힙 자료구조를 구현하라 최소힙 자료구조는 부모가 자식보다 작으면 되고, 같은 형제 레벨에서 순서는 상관이 없다. my_solution tmp = [] while True: num = int(input()) if num == -1: sys.exit() elif num != 0: tmp.append(num) elif num == 0: if tmp: tmp.sort() # 그나마 정렬해서 pop 하는건 5초짜리는 통과함.. print(tmp.pop(0)) # print(tmp.pop(tmp.index(min(tmp)))) # 5초짜리도 case5 타임아웃 else: print(-1) continue list, pop() 사용 (최소힙 자료구조 몰랐음) 이렇게 하니 pop()연산때문에 입력값이 ..

문제 포인트 최소힙을 사용하여 최대힙으로 구현해보자 heapq 자료구조는 무조건 최솟값만을 위한 자료구조이기 때문에 이 자료구조를 사용하여 최솟값을 판별하기 위해서는 약간의 트릭이 필요하다. 바로 -로 push하는 것. 3,4 있으면 3이 더 작은 숫자지만 -3, -4가 있으면 -4가 더 작은 숫자가 된다. 이렇게 push하고 마지막에 pop()하는 단계에서 - 붙여서 pop()하주면 그게 가장 큰 값이다. solution """ 힙큐 자료구조는 무조건 최소힙을 만들어내기 때문에 최대힙을 사용하기 위해서는 - 값으로 push하여 - 값을 붙인 상태로 출력하면 최댓값 출력 가능. """ tmp = [] while True: num = int(input()) if num == -1: sys.exit() el..