전체 글10 [백준 1197 / Python(파이썬)] 최소 스패닝 트리 [백준 1197 / Python(파이썬)] 최소 스패닝 트리 문제 풀이 크루스컬 알고리즘(Kruskal Algorithm) 이용 알고리즘을 이용하여 가중치의 크기를 구한 뒤 print해준다. 틀린 풀이: 시간 초과 #최소 스패닝 트리 import sys V, E = map(int, sys.stdin.readline().split()) arr = [list(map(int, sys.stdin.readline().split())) for _ in range(E)] arr.sort(key=lambda x: x[2]) uf = [i for i in range(V+1)] #union-find def kruskal(): global V, E, arr, uf mst = 0 #최소 스패닝 트리의 가중치를 담음 for u.. 2022. 7. 29. [백준 1700 / 파이썬(Python)] 멀티탭 스케줄링 [백준 1700 / 파이썬(Python)] 멀티탭 스케줄링 문제 풀이 이 문제는 그리드 알고리즘이므로 문제를 풀기 위한 기준을 나누어줬다. 1. 멀티탭에 이미 전기용품이 꽂혀있을 때: continue 2. 멀티탭에 구멍이 남아있을 때: 전기용품을 꽂아주고 continue 3. 멀티탭에 구멍이 없어 전기용품을 바꿔줘야할 때: 3-1 남은 순서 중에 계속 써야하는 전기용품이 없을 때: 멀티탭에서 뺄 전기용품의 위치를 바꾸어준 뒤 break 3-2 계속 전기용품을 써야할 때: 이후 순서에서 써야하는 전기용품 중 가장 멀리있는 것으로 위치를 바꾸고, 멀티탭에서 뺄 전기용품의 위치를 바꾸어준다. 틀린 풀이 idx와 farN의 초기화가 진행되지 않아 27%에서 틀렸다. n, k = map(int, input().s.. 2022. 7. 28. [백준 14179/파이썬(Python)] 빗물 [백준 14179/파이썬(Python)] 빗물 문제 풀이 2차원 세계에서 빈공간과 벽을 0과 1로 나누어줌 2차원 세계의 가장 바닥부터 왼쪽에서 오른쪽으로 벽인지 아닌지 검사 temp라는 배열을 만들어준 뒤 왼쪽 벽과 오른쪽 벽에 대한 가로 좌표를 저장 temp의 길이가 총 2가 되면(왼쪽 벽이 나타난 뒤 오른쪽 벽이 나타나면) 벽이 붙어있는지 검사 붙어있지 않으면 빗물이 고이는 부분을 0에서 2로 수정하고 빗물의 양을 count 해줌 # 빗물 h, w = map(int, input().split()) arr = list(map(int, input().split())) m = [[1] * w for _ in range(h)] temp = [] #왼쪽의 벽 가로 좌표와 오른쪽에 존재하는 벽 가로 좌표 저장.. 2022. 7. 18. Train/Validation 나눠주기 yolov5 학습을 위해 train : validation을 8:2로 나누어주기 위한 코드. 경로를 작성해줄 때 glob 라이브러리를 이용하던데 라이브러리를 사용하여 실행하면 마지막 파일 앞 경로 부분이 '/'가 아닌 '\\' 이런 식으로 작성이 되어 경로 오류가 생겼다. 따라서 os.listdir과 os.path.join을 이용하여 경로를 합쳐주었다. import os from pathlib import Path import shutil from sklearn.model_selection import train_test_split #getting list of images image_files = os.listdir("이미지경로") img_path = [] for file in image_files: .. 2022. 4. 3. 이전 1 2 3 다음
