[Silver IV] 숫자 카드 2 - 10816

문제 링크

성능 요약

메모리: 115648 KB, 시간: 848 ms

분류

이분 탐색, 자료 구조, 해시를 사용한 집합과 맵, 정렬

문제 설명

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 몇 개 가지고 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 몇 개 가지고 있는 숫자 카드인지 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

출력

첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 몇 개 가지고 있는지를 공백으로 구분해 출력한다.

풀이

n = int(input())
cards = list(map(int, input().split()))
m = int(input())
targets = list(map(int, input().split()))

cnt = {}
for card in cards:
    if card in cnt:
        cnt[card] += 1
    else:
        cnt[card] = 1
        
for target in targets:
    answer = cnt.get(target)
    if answer == None:
        print(0, end=" ")
    else:
        print(answer, end=" ")

설명

• 가지고 있는 카드의 숫자를 key로, 개수를 value로 하는 딕셔너리 cnt를 만든다.
• 개수를 구하고자 하는 카드의 숫자(target)를 키로 get 하여 value를 가져온다.
  -> 없으면 0, 있으면 값 출력

추가 - 파이썬 딕셔너리 내장함수 / 메서드

• 내장함수
  • len(dict) : 딕셔너리 총 요소 길이
  • str(dict) : 사전을 문자열로 반환
  • type(variable) : 변수의 형태
• 메서드
  • dict.key() : 사전의 키 목록
  • dict.values() : 사전의 값 목록
  • dict.items() : 사전의 (key,value) 튜플 목록
  • dict.clear() : 사전의 모든 요소 제거
  • dict.copy() : 사전 복사
  • dict.fromkeys() : seq, value로 사전 생성
  • dict.get(key, default=None) : 키에 저장된 value 불러오기
  • dict.setdefault(key, default=None) : dict.get()과 유사하며,해당 key가 딕셔너리 내에 존재하면 value를 반환한다. 만약 존재하지 않으면 None을 반환한다.
  • dict.uptate(dict2) : 기존 딕셔너리에 dict2 추가

추가2 - 딕셔너리 기본값 처리

https://www.daleseo.com/python-collections-defaultdict/