알고리즘 문제/BOJ
5567번 결혼식
parkit
2018. 12. 24. 00:24
728x90
반응형
https://www.acmicpc.net/problem/5567
계속 틀렸습니다가 뜨길래 대회(일본정보올림피아드 예선) 테스트 케이스를
통해 이유를 발견하였다.
46번 째 줄 코드(방문 처리)를 꼭 작성해줘야 한다.
방문 처리를 안 하면, two vector에 중복된 원소가 들어간다.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | #include <iostream> #include <queue> #include <stack> #include <cstdio> #include <vector> #include <cstring> #include <string> #include <math.h> #include <algorithm> using namespace std; vector<int> v[10001], one, two; bool visit[10001] = { false, }; int main(void) { int n = 0, m = 0, a = 0, b = 0; scanf("%d %d", &n, &m); for (int i = 0; i < m; i++) { scanf("%d %d", &a, &b); v[a].push_back(b); v[b].push_back(a); } visit[1] = true; for (auto next : v[1]) { visit[next] = true; one.push_back(next); } for (auto vertex : one) { for (auto next : v[vertex]) { if (visit[next]) continue; visit[next] = true; two.push_back(next); } } printf("%d\n", one.size() + two.size()); printf("\n"); return 0; } | cs |
728x90
반응형