[백준] 202999번 3대 측정 파이썬

202999번 3대 측정 파이썬 풀이

 

[시도1] pypy3 출력 초과

n, k, l = map(int, input().split())

arr = [list(map(int, input().split())) for _ in range(n)]

count = 0
cnt = ''

for i in range(0, n):
  sum = 0
  sum = arr[i][0] + arr[i][1] + arr[i][2]
  if sum >= k and arr[i][0] >= l and arr[i][1] >= l and arr[i][2] >= l :
    count = count + 1
    cnt = cnt + str(i)
    
print(count)  
for i in range(0,len(cnt)):
  for j in range(0,3):
    print(arr[int(cnt[i])][j] , end = ' ')

 

[시도2] python3 시간 초과 (코드는 시도1과 동일 python3은 정답이 나오는지 해봄)

n, k, l = map(int, input().split())

arr = [list(map(int, input().split())) for _ in range(n)]

count = 0
cnt = ''

for i in range(0, n):
  sum = 0
  sum = arr[i][0] + arr[i][1] + arr[i][2]
  if sum >= k and arr[i][0] >= l and arr[i][1] >= l and arr[i][2] >= l :
    count = count + 1
    cnt = cnt + str(i)
    
print(count)  
for i in range(0,len(cnt)):
  for j in range(0,3):
    print(arr[int(cnt[i])][j] , end = ' ')

 

=> 좀 더 효율적으로 코드를 작성하는 방식을 생각해봐야한다.

 

www.acmicpc.net/problem/20299

20299번: 3대 측정