[Python] 백준1978 소수 찾기

22 Dec 2022 in Study on Algorithm

백준1978 소수 찾기 링크

문제

예제 입력

---

코드

n = int(input())
arr = list(map(int, input().split()))
res = 0

for num in arr:
    cnt = 0
    for i in range(1, num+1):
        if num % i == 0:
            cnt += 1
        if cnt > 2:
            break
                
    if cnt == 2:
        res += 1
    
print(res)

설명

파이썬을 통해서 사용자로부터 입력받아 1과 자기 자신외에 나눠지는 값을 찾아서 값이 존재한다면 최종 값에서 제외합니다.

---

정리

list(filter(lambda x: x > 0, [1, -5])) -> 람다는 더이상 사용필요x
map 함수 list(map(lambda, a)) : 데이터를 나눔
reduce 함수 반복가능 자료형(list, tuple, range = iter() 반복자로 만들어 next()함수로 호출가능)에 적용 : 데이터 묶고 줄임(처리)
from functools import reduce
n = reduce(lambda x,y:x+y,range(1, 101))
print(n)

a = [1, 2, 3, 4]
print([x**2 for x in a])
print([x**2 for x in a if x % 2 == 0])
print([int(x) for x in input("입력").split()])
print([j**2 for i in a for j in [1, 2, 3]])
print([x**2 for x in a if x % 2 == 0  if x % 2 == 1])

---

결과