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

이효민님의 프로필 이미지
이효민

작성한 질문수

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

1-L

1-L1111_1940 주몽

해결된 질문

작성

·

134

0

안녕하세요 큰돌님, 문제를 풀다가 테스트 코드랑 제가 직접 만든 케이스는 잘 돌아가는데 백준에 넣으면 바로 틀려버립니다..ㅎㅎ

저는 이 문제를 visited 배열로 풀었는데요, 벡터에 고유 번호를 담고 만약 m=9라면 먼저 for문을 통해 v의 원소를 하나하나 확인합니다. 먼저 9-2=7이 존재 하는지 확인하고 존재 한다면 갑옷을 만드는 경우가 있는거니까 cnt++을 했습니다. 어디서 반례가 생기는지 모르겠습니다.ㅜㅜ

별로 효율적이지 못한 코드 같긴하지만 한번 검토 부탁드립니다..ㅎㅎ

 

http://boj.kr/2e4c75e8c7d24a1284cc02827162c36c

답변 1

1

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

안녕하세요 효민님 ㅎㅎ

일단 먼저 드는 생각은 visited 배열을 사용한다면 이렇게 하는게 어떤가요?

#include <bits/stdc++.h>
using namespace std;
int visited[200004], cnt; 
int main() {
    int n, m;
    cin >> n >> m;
    vector<int> v(n);   
    for (int i = 0; i < n; i++) {
        cin >> v[i];
    }

    for (int i = 0; i < n; i++) {
        if (visited[i]) continue; 
        for (int j = i + 1; j < n; j++) {
            if (!visited[j] && v[i] + v[j] == m) {
                visited[i] = visited[j] = 1; 
                cnt++;
                break;  
            }
        }
    }

    cout << cnt << "\n";
    return 0;
}

이게 첫번째 생각이고 만약 find를 쓴다고 생각하면

이렇게 하시면 됩니다. n = 1일 때의 경우, 그리고 같은 경우(고유한 번호기 때문에 해당 부분은 일어나지 않으나 그냥 find로 하게 되면 5, 5 이렇게 되는 경우가 생겨버림) 를 고려하는 코드를 추가로 넣어시면 됩니다.

#include <bits/stdc++.h>
using namespace std;
int visited[200004], cnt; 
int main() {
    int n, m;
    cin >> n >> m; 
	if(m > 200000 || n == 1){
        cout << 0 << "\n";
        return 0; 
    }
    vector<int> v(n);   
    for (int i = 0; i < n; i++) {
        cin >> v[i];
    }
    for(int i : v){
        if(visited[i]) continue;  
        if(m - i < 0) continue;
        if(m - i == i) continue;
        auto it = find(v.begin(), v.end(), m - i);
        if(it == v.end()) continue; 
        if(visited[m - i] == 0 && visited[i] == 0){
            visited[i] = visited[m - i] = 1; 
            cnt++; 
        } 
    } 
    cout << cnt << "\n";
    return 0;
}

이렇게 해보시겠어요? ㅎㅎ

 

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

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

감사합니다.

강사 큰돌 올림.

 

이효민님의 프로필 이미지
이효민

작성한 질문수

질문하기