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

jessie117님의 프로필 이미지
jessie117

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

1-L

질문있습니다!

작성

·

89

0

안녕하세요! 강의와는 다른 방식의 로직으로 풀어보았습니다.

v 라는 방문처리 할 수있는 배열을 만들었습니다. 재료 배열을 순회하면서 M에서 해당 값을 뺏을 때 배열에 그 값이 있다면 입력값의 위치와 M-해당값의 위치를 방문 처리 해주면서 결과값을 count 하였습니다.

Test case는 맞다고 뜨는데 제출하면 틀렸다고 뜨네요ㅜㅜ

문제에 맞지 않는 잘못된 코드 일까요...?

http://boj.kr/eb03bcf5031743d9a74c8465cb6a703c

답변 1

0

큰돌님의 프로필 이미지
큰돌
지식공유자

안녕하세요 jessie님 ㅎㅎ

혹시 이런 로직을 구현하려고 하신게 아닌가요?

제가 좀 다듬어 봤습니다.

참고 부탁드립니다.

find() -> map 자료구조 사용 등

#include <bits/stdc++.h>
using namespace std;

int N, M, a[15004], v[15004];

int main() {
    cin >> N;
    cin >> M;
    for (int i = 0; i < N; i++) cin >> a[i];

    map<int, int> mp;
    int ret = 0;

    for (int i = 0; i < N; i++) {
        int idx = M - a[i];
        if (mp[idx] > 0) {
            ret++;
            mp[idx]--;
        } else {
            mp[a[i]]++;
        }
    }

    cout << ret;
    return 0;
}

 

물론 jessie님의 코드처럼 find()를 써서 풀 수도 있습니다.

이경우 탐색범위를 제한하는 로직을 넣어야 합니다.

#include <iostream>
#include <algorithm>
using namespace std;

int N, M, a[15004], v[15004];

int main() {
    cin >> N >> M;
    for (int i = 0; i < N; i++) cin >> a[i];

    int ret = 0;
    for (int i = 0; i < N; i++) {
        if (v[i] == 0) {
            int complement = M - a[i];
            int *pos = find(a + i + 1, a + N, complement); // i 이후의 요소만 검색
            if (pos != a + N && v[pos - a] == 0) {
                v[i] = 1;
                v[pos - a] = 1;
                ret++;
            }
        }
    }

    cout << ret;
    return 0;
}

다만 map 사용을 추천드립니다!(이 문제의 경우 map 사용이 더 빠릅니다.)


imagemap : 12ms, find() : 48ms

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

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

감사합니다.

강사 큰돌 올림.


jessie117님의 프로필 이미지
jessie117

작성한 질문수

질문하기