본문 바로가기

Programming/Algorithm (C++)

[백준 5567] c++ :: 결혼식

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다. 

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
#include <stack>
#include <string>

using namespace std;
int n;
int m;
int map[10001][2] = { 0 };
int list[501] = { 0 };
int result = 0;

int main()
{
	cin >> n;
	cin >> m;

	for (int i = 0; i < m; i++) { //map 채워 넣기
		cin >> map[i][0] >> map[i][1];
	}
	for (int i = 0; i < m; i++) {
		if (map[i][0] == 1) {  //만약 맨 처음 1이면
			int check = map[i][1]; //그 1 옆 숫자를 check로 받아서
			list[check] = 1; //list배열에 check해당하는 번째 배열 1로 채우기
			for (int j = 0; j < m; j++) { //만약 이미 채워 진 상태라면(친구의 친구라면)
				if (map[j][0] == check) //맨처음 아이가 check상태인 친구라면
					list[map[j][1]] = 1; //그 옆에 해당하는 번째 친구에도 1로 채우고
				else if (map[j][1] == check) //반대로 되어 있는 상황에도 체크
					list[map[j][0]] = 1;
			}
		}
	}


	for (int i = 2; i <= n; i++) { //1인건 빼고 부터니까
		if (list[i] == 1) { //1로 채워져있으면 result를 증가
			//printf("%d i\n", i);
			result++;
		}
	}

	cout << result << "\n";
	return 0;

}

'Programming > Algorithm (C++)' 카테고리의 다른 글

[백준 1167] c++ :: 트리의 지름  (1) 2024.03.18
[백준 1260] c++ :: DFS와 BFS  (0) 2024.03.17
[백준 3085] c++ :: 사탕 게임  (0) 2021.04.07
rush02 정리  (0) 2021.03.16
c08  (0) 2021.03.11