-
[백준] 백준에서 NumPy 사용 안되나요?Language/Python 2022. 2. 17. 10:00728x90
ㅠㅠ 넘파이로 코드 짰는데 런타임 에러라니
# 마인크래프트 # 땅은 이중 리스트로 한다(0,0) ~ (m,n) # 땅의 높이의 평균을 구하여 반올림(integer): 이상적인 땅의 높이(narashi) 그 높이로 정렬한다. # (narashi - 각각의 땅 높이[i,j]) = 필요한 흙 블럭(damand[i,j]) # demand가 가지고 있는 흙 블럭보다 많으면 narashi 1 빼고, 다시 demand 구함 # demand가 가지고 있는 흙 블럭보다 같거나 적으면 narashi를 구했다 이제 작업 시간(time)을 구하자 # (narashi - 각각의 땅 높이)가 0이면 즉 작업이 필요 없다면 (time += 0) # (narashi - 각각의 땅 높이)가 0보다 크면 즉 블럭을 쌓아야 하면 (time += 1) # (narashi - 각각의 땅 높이)가 0보다 작으면 즉 블럭을 없애야 하면 (time += 2) # time, narashi 출력 import numpy as np def P18111(): n, m, b = map(int, input().split()) height = np.array(list(list(map(int, input().split())) for _ in range(n))) narashi = np.around(height.mean()).astype(int) print("나라시 1", narashi) while True: demand = narashi - height if demand.sum() > b: narashi -= 1 else: break time = 0 # if demand < 0: # time += 2 # elif demand > 0: # time += 1 # else: # pass for i in range(n): for j in range(m): if demand[i][j] < 0: time += 2 elif demand[i][j] > 0: time += 1 else: pass print(time, narashi) if __name__ == '__main__': P18111()
아까우니 블로그에라도 올립니다 ㅠㅠ
728x90'Language > Python' 카테고리의 다른 글
[Python] 설치된 패키지 목록 만들기 | requirements.txt (0) 2022.04.25 [Python] 한 줄로 된 숫자를 리스트로 입력받는 법 (0) 2022.02.04 [Python] 리스트를 요소의 길이 별로 정렬 | 내장 함수 sorted()의 key 매개변수 활용과 딕셔너리 컴프리핸션 | 백준 1181번 해설 (0) 2022.01.28 [Python] 여러 줄 한번에 입력하여 리스트로 만들기 (0) 2022.01.27 [Python] sys.stdin.readline()과 input()의 차이 | 백준 2751번 (0) 2022.01.26