코딩테스트 연습

[파이썬] 백준: 1차원 배열 연습문제

콩콩(๓° ˘ °๓)♡ 2023. 1. 24. 16:25

#10807 개수 세기

N = int(input())
N_list = list(map(int, input().split()))
v = int(input())
print(N_list.count(v))

count 함수 사용.

 

#10871 X보다 작은 수 

N, X = map(int, input().split())
A = list(map(int, input().split()))
for i in range(N):
    if A[i] < X:
        print(A[i],end = " ")

 

단순히 X보다 작은 A의 원소를 출력하려면 아래와 같이 할 수 있다.

for i in A:
    if i  <  X:
        print(i, end=" ")

하지만 전체 원소개수 N이 주어졌기 때문에 N회 반복하는 for문을 이용하여 답을 출력한다.

 

#2562 최댓값

A라는 빈 리스트를 만들어 입력값을 A의 원소로 append  gownsek.

A = []
for _ in range(9):
    i = int(input())
    A.append(i)
    
print(max(A))
print(A.index(max(A))+1)

for 문을 list comprehension으로 작성하면 다음과 같다.

이렇게 하면 append를 사용하지 않아도 된다.

A=[int(input()) for _ in range(9)]

A.sort() 함수 사용하여 오름차순으로 정렬한 뒤 0번, -1번 인덱스의 값을 출력할 수 도 있다.

A.sort()
print(A[0], A[-1])

 

#5597

org= [i for i in range(1,31)]
for _ in range(28):
    done = int(input())
    org.remove(done)
    
print(min(org))
print(max(org))

remove함수를 이용해 org 리스트에서 과제를 제출한 done 원소를 삭제해준다.