해결된 질문
작성
·
134
0
안녕하세요 큰돌님, 문제를 풀다가 테스트 코드랑 제가 직접 만든 케이스는 잘 돌아가는데 백준에 넣으면 바로 틀려버립니다..ㅎㅎ
저는 이 문제를 visited 배열로 풀었는데요, 벡터에 고유 번호를 담고 만약 m=9라면 먼저 for문을 통해 v의 원소를 하나하나 확인합니다. 먼저 9-2=7이 존재 하는지 확인하고 존재 한다면 갑옷을 만드는 경우가 있는거니까 cnt++을 했습니다. 어디서 반례가 생기는지 모르겠습니다.ㅜㅜ
별로 효율적이지 못한 코드 같긴하지만 한번 검토 부탁드립니다..ㅎㅎ
답변 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점은 제게 큰 힘이 됩니다. :)
감사합니다.
강사 큰돌 올림.