알고리즘 문제풀이

백준 10815 숫자 카드 Python 자료구조 실버5

아크몽 2024. 3. 8. 16:44

파이썬에서 딕셔너리를 이용할 수 있음에 감사하자

'''
상근이 숫자 n개
정수 m개 주어짐
주어진 숫자카드를 상근이가 가지고 있는가?

같은수는 적혀있지 않음, 음수 숫자카드도 존재함

'''
n = int(input())
n_arr = list(map(int,input().split()))
m = int(input())
m_arr = list(map(int,input().split()))
dict_arr = {}
for i in m_arr:
    dict_arr[i] = 0
# print(dict_arr)
for j in n_arr:
    if dict_arr.get(j) ==0 :
        dict_arr[j] += 1
# print(dict_arr)
print(*dict_arr.values())

시간초과 (2%) 리스트를 사용해서 시간초과 뜸

 for i in range(m): # 숫자카드
     flag = 1
     for j in range(n): # 상근이 카드
         if m_arr[i] == n_arr[j]:
             print(1, end=' ')
             flag = 0
             break
     if flag:
         print(0,end=' ')