백준[10816] - 숫자 카드 2

 

 

 

문제

10816번: 숫자 카드 2

 

 

 

작성한 코드

• 처음 작성한 코드

import sys
input = sys.stdin.readline

N = int(input())
N_list = list(map(int, input().split()))
N_list.sort()

M = int(input())
M_list = list(map(int, input().split()))

for card in M_list:
    lt = 0
    rt = N-1
    count = 0
    isExit = False
    
    while lt<=rt:
        mid = (lt+rt)//2
        
        if card == N_list[mid]:
            isExit = True
            print(N_list.count(card), end=' ')
            break
        elif card > N_list[mid]:
            lt = mid + 1
        else:
            rt = mid - 1

    if not isExit:
        print(0, end=' ')

- 이전 문제(수 찾기)와 동일하게 코드를 작성했는데 시간 초과가 나버렸다.

 

• 다른 사람 풀이

import sys
from collections import Counter
input = sys.stdin.readline

N = int(input())
N_list = list(map(int, input().split()))
N_list.sort()

M = int(input())
M_list = list(map(int, input().split()))

counter = Counter(N_list)

for card in M_list:
    print(counter[card], end=' ')

- 해당 문제를 시간초과 내지 않고 푸는 방법 : dictionary 사용 / Counter 사용 (나는 Counter를 사용해서 풀어보았다.)

- Coutner 함수 : 리스트 내부 원소 갯수를 count 해서 dictionary 형태로 만들어주는 함수

 

 

cf.) counter를 출력해보면 이렇게 나오게 된다.

Counter({10: 3, -10: 2, 3: 2, 2: 1, 6: 1, 7: 1})