[beakjoon] 수 찾기

김민서·2024년 1월 5일
0

알고리즘 문제풀이

목록 보기
5/47

N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때, 이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.

예제 입력
5
4 1 5 2 3
5
1 3 7 9 5

첫번째 줄에는 그 다음 줄에서 입력받을 배열의 크기를 입력받고 그 다음줄에 배열 A를 입력받는다.
세번째 줄에서도 마찬가지로 그 다음 줄에서 입력받을 배열의 크기를 입력받고 그 다음줄에 배열 nums를 입력받는다.

n = int(input())
a = list(map(int, input().split())
m = int(input())
nums = list(map(int, input().split()))

answer = []

for x in nums: 		
	if x in a:
    	answer.append(1)
    else:
    	answer.append(0)
        
for i in answer:
	print(i)
    

설명:
(x)
nums = [1, 3, 7, 9, 5]
a = [4, 1, 5, 2, 3]

nums 안에 있는 값(x)이 a 안에 존재하는지 확인
nums를 순회하면서 x들을 하나씩 확인
만약 x가 a 안에 있다면
결과 리스트에 1을 담고,
없다면 결과 리스트에 0을 담는다.
결과 리스트를 차례로 출력한다.

0개의 댓글