해결된 질문2024-09-06T13:29:35.782Z

한유태

작성한 질문수 79

1

안녕하세요 유태님 ㅎㅎ

이거 시간초과 나오지 않나요?

그리고..

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

입력만 받아야 합니다. 굳이 mp에 넣을 필요는 없습니다. 해당 vecB를 기반으로 mp에 있냐 없냐를 확인하면 되기 때문입니다.

 

시간초과는 어떻게 해결할까요?

이문제는 시간이 타이트한 문제입니다. 이 경우 cin을 사용하지 않고 scanf 등을 사용하면 풀리기도 하는데요. 이렇게 하시면 됩니다.

#include <iostream>
#include <vector>
#include <map>

using namespace std;

int T;
int N, M;
map<int, int> mp;

int main()
{
    scanf("%d", &T);

    while (T--)
    {
        mp.clear(); 

        scanf("%d", &N);
        vector<int> vecA(N);
        for (int i = 0; i < N; i++)
        {
            scanf("%d", &vecA[i]);
            mp[vecA[i]] = 1;
        }

        scanf("%d", &M);
        vector<int> vecB(M);
        for (int i = 0; i < M; i++)
        {
            scanf("%d", &vecB[i]);
        }

        for (int i = 0; i < M; i++)
        {
            if (mp[vecB[i]] == 1) 
                printf("1\n");
            else 
                printf("0\n");
        }
    }

    return 0;
}

 

감사합니다.