인프런 커뮤니티 질문&답변
질문있습니다!
작성
·
122
0
안녕하세요! 강의와는 다른 방식의 로직으로 풀어보았습니다.
v 라는 방문처리 할 수있는 배열을 만들었습니다. 재료 배열을 순회하면서 M에서 해당 값을 뺏을 때 배열에 그 값이 있다면 입력값의 위치와 M-해당값의 위치를 방문 처리 해주면서 결과값을 count 하였습니다.
Test case는 맞다고 뜨는데 제출하면 틀렸다고 뜨네요ㅜㅜ
문제에 맞지 않는 잘못된 코드 일까요...?
답변 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 사용이 더 빠릅니다.)
map : 12ms, find() : 48ms
또 질문 있으시면 언제든지 질문 부탁드립니다.
좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)
감사합니다.
강사 큰돌 올림.





