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

안녕하세요 ㅎㅎ

        if (mat_num == num_found) break;

        auto it_found = mat_nums.find(M-mat_num);
        if (it_found != mat_nums.end())
        {
            cnt++;
            num_found = *it_found;

            //cout << "   found num : " << num_found;
        }

제 생각에 이부분은

정렬되어있는 set에서

1 2 3 4 5 6 7

이렇게 되어있을 때

3부분에서 5를 선택했다면

그 다음 5가 나왔을 때 break를 걸어서 전체 탐색을 막으려고 했던 것 같은데요.

괜찮은 로직인 거 같지만... 반례는 다음과 같습니다.

10 10 1 2 3 4 5 6 8 10 12 14

정답 : 2

gh님 코드 : 5

set을 이용하신다면 이렇게 고치시는 것은 어떨까요?

#include <iostream>
#include <string>
#include <set>

using namespace std;

int main() {
    int N, M;
    cin >> N >> M;

    set<int> st;

    for (int i = 0; i < N; i++) {
        int num;
        cin >> num;
        st.insert(num);
    }

    int cnt = 0;

    for (auto mat_num : st) {
        int complement = M - mat_num; 
        if (st.count(mat_num) && st.count(complement)) {
            cnt++;
        }
    } 
    cout << cnt / 2;
    return 0;
}

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

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

감사합니다.

강사 큰돌 올림.