인프런 커뮤니티 질문&답변

안녕하세요 ㅎㅎ

	for (int i : ret)
		cout << i << "\n";

이 코드만 이렇게 고치시면 됩니다. endl보다는 "\n"을 쓰는게 좋습니다.

이렇게 하시면 시간초과가 뜨지 않습니다.

해당 부분은 교안내의 다음 부분 참고 부탁드립니다.

endl보다는 “\n”을 써라.

코드리뷰는 다음과 같습니다.

1. 필요없는 배열

	vec2 = vector<int>(M, 0);
	for (int i = 0; i < M; ++i)
	{
		cin >> vec2[i];
	}

사실 vec2는 필요가 없습니다.

해당 변수를 받아서 이분탐색만 하면 되기 때문에 temp라는 변수로 하면 vector 한개에 대한 공간복잡도를 줄 일 수 있습니다.

vector같은 경우 이렇게도 입력을 받을 수 있습니다.

다만 stark님이 하신것도 맞으니 참고 삼아 알아두시면 좋습니다.

2. 입력시 이렇게도.

    vector<int> vec(N);
    for (int& value : vec)
    {
        cin >> value;
    }

나머지 부분은 너무나도 훌륭하게 잘 짜셨습니다. ㅎㅎ

다듬은 코드는 다음과 같습니다.

#include <iostream>
#include <vector>
#include <queue>
#include <stack>
#include <algorithm>
#include <memory>
using namespace std;

int N, M;
vector<int> vec;
vector<int> vec2;
vector<int> ret;

void bs(int s, int e, int v)
{
	if (s > e)
	{
		ret.push_back(0);
		return;
	}

	int mid = (s + e) / 2;
	int midVal = vec[mid];

	if (midVal == v)
	{
		ret.push_back(1);
		return;
	}
	
	if (midVal > v)
		bs(s, mid - 1, v);
	else if (midVal < v)
		bs(mid + 1, e, v);
}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> N;
	vec = vector<int>(N, 0);
	for (int i = 0; i < N; ++i)
	{
		cin >> vec[i];
	}

	cin >> M;
	sort(vec.begin(), vec.end()); 
	for (int i = 0; i < M; ++i)
	{
        int temp;
		cin >> temp;
        bs(0, vec.size() - 1, temp);
	} 

	for (int i : ret)
		cout << i << "\n";

	return 0;
}

참고로 제가 강의 외의 문제는 답변드리지 않습니다 ㅠㅠ 이번만 답변드렸어요.. ㅎㅎ 참고부탁드립니다.

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.