백준 기초Python(반복문, 1차원 배열)

반복문

15552번

빠른 A+B

import sys
 
a = int(input())
for i in range(a):
        a,b = map(int, sys.stdin.readline().split())
        print(a+b)

 

=> for문을 통해 여러번 입력받을 경우 input은 너무 시간이 오래 걸리기 때문에 sys.stdin.readline사용!

 

 

1차원 배열

10807번

1차원 배열 안에 v가 몇개 있는지 출력하기

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

 

 

10871번

1차원 배열안에 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 = ' ')

 

 

2562번

1차원 배열안의 최댓값과 그 위치 찾기(9개 입력받음)

A = []
for i in range(9):
    b = int(input())
    A.append(b)

print(max(A))
print(A.index(max(A))+1)

 

 

10810번

바구니에 공 넣기( 한바구니에 한개만 들어감. 이미 있는 바구니에 다시 넣으면 교체)

N,M = map(int,input().split())
A = [0]*N

for i in range(M):
    a, b, c = map(int,input().split())
    for j in range(a-1,b):
        A[j] = c

for i in range(N):
    print(A[i], end= ' ')

=> A[i]를 바로 쓰기 위해 A의 크기를 미리 지정(A = [0]*N)

 

 

10813번 

바구니안의 공 바꾸기(바구니 순서대로 공 배치 후 선택한 2개의 바구니속의 공 교체)

N,M = map(int,input().split())
A = [0]*N

for i in range(N):
    A[i] = i+1
    
for i in range(M):
    a, b = map(int,input().split())
    c = A[a-1]
    A[a-1] = A[b-1]
    A[b-1] = c
 
for i in range(N):
    print(A[i], end= ' ')