백준 1920번 수찾기

문제

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

입력

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다. 모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다.

출력

M개의 줄에 답을 출력한다. 존재하면 1을, 존재하지 않으면 0을 출력한다.

 

 

첫번쨰 풀이 

 

import sys

n_num=int(sys.stdin.readline())

n=list(map(int,sys.stdin.readline().split()))

m_num=int(sys.stdin.readline())

m=list(map(int,sys.stdin.readline().split()))

result=[]
for i in range(len(m)):
    if m[i] in n:
        result.append(1)
    else:
        result.append(0)
        
        
for i in result:
    print(i)
    

시간초과 오류 

 

문제를 자세히 읽어보니 정수의 범위가 2^31승 까지였다. 확실히 시간초과가 나올만했던거같다.

 

2번쨰 풀이 

 

딕셔너리를 이용한 풀이

 

import sys
input=sys.stdin.readline
n=int(input().rstrip())
n_list=list(map(int,input().split()))
m=int(input())
m_list=list(map(int,input().split()))
dic={}
for i in n_list:
    dic[i]=1
    
for val in m_list:
    try:
        print(dic[val])
    except KeyError:
        print(0)

 

여기서 KeyError는 dic 안에 해당하는 Key값이 없기떄문에 발생하는오류

 

dic안에 val해당하는 key값이 있다면 ? 1을 그대로 출력

아니라면 0을 출력해주었습니다.

 

다른방법이 없을까 고안을 해보았습니다

 

3번째 풀이 이분탐색

 

 

import sys

input=sys.stdin.readline

n=int(input().rstrip())

n_list=list(map(int,input().split()))

n_list=sorted(n_list)

m=int(input())

m_list=list(map(int,input().split()))

def binary_search(element,some_list,start=0,end=None):

    if end==None:

        end=len(some_list)-1

    if start>end:

        return 0

    mid = (start+end) //2

    if element == some_list[mid]:

        return 1

    elif element < some_list[mid]:

        end = mid-1

    elif element > some_list[mid]:

        start = mid+1

    return binary_search(element,some_list,start,end)

for i in range(len(m_list)):

    print(binary_search(m_list[i],n_list))

 

이분탐색으로 계속 절반으로 쪼개가면서 풀어가는형식입니다.

 

 

 

백준 골드까지 얼마남지 않았네요 열심히 해도록 하겠습니다.!