본문 바로가기
알고리즘 문제 풀이/백준

백준 2422 한윤정이 이탈리아에 가서 아이스크림을 사먹는데 c++ (완전탐색)

by 옹구스투스 2021. 9. 23.
반응형

문제 출처 : https://www.acmicpc.net/problem/2422

 

2422번: 한윤정이 이탈리아에 가서 아이스크림을 사먹는데

첫째 줄에 정수 N과 M이 주어진다. N은 아이스크림 종류의 수이고, M은 섞어먹으면 안 되는 조합의 개수이다. 아래 M개의 줄에는 섞어먹으면 안 되는 조합의 번호가 주어진다. 같은 조합은 두 번

www.acmicpc.net

문제

한윤정과 친구들은 이탈리아로 방학 여행을 갔다. 이탈리아는 덥다. 윤정이와 친구들은 아이스크림을 사먹기로 했다. 아이스크림 가게에는 N종류의 아이스크림이 있다. 모든 아이스크림은 1부터 N까지 번호가 매겨져있다. 어떤 종류의 아이스크림을 함께먹으면, 맛이 아주 형편없어진다. 따라서 윤정이는 이러한 경우를 피하면서 아이스크림을 3가지 선택하려고 한다. 이때, 선택하는 방법이 몇 가지인지 구하려고 한다.

입력

첫째 줄에 정수 N과 M이 주어진다. N은 아이스크림 종류의 수이고, M은 섞어먹으면 안 되는 조합의 개수이다. 아래 M개의 줄에는 섞어먹으면 안 되는 조합의 번호가 주어진다. 같은 조합은 두 번 이상 나오지 않는다. (1 ≤ N ≤ 200, 0 ≤ M ≤ 10,000)

출력

첫째 줄에, 가능한 방법이 총 몇 개 있는지 출력한다.

힌트

5개의 아이스크림과 3가지 섞어먹으면 안 되는 조합이 있다. 1번은 2번 3번과 섞으면 안되고, 3번은 4번과 섞으면 안 된다. 따라서 (1 4 5), (2 3 5), (2 4 5)와 같이 3가지 방법이 있다.

알고리즘 분류

풀이

n개의 아이스크림 중에서 3가지를 섞어서 먹는 방법의 개수를 찾는데, 이중 섞어먹으면 안 되는 조합이 주어진다.

nope[][]의 2차원 배열에 섞어먹으면 안 되는 조합을 저장하고, 모든 조합에 대해 이를 조합하면 되는 간단한 문제이다.

 

코드

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int n, m;
bool nope[201][201];
int ans;
vector<int> result;
void combination(int size,int idx) {
	if (size == 3) {
		if (nope[result[0]][result[1]]) {
			return;
		}
		else if (nope[result[0]][result[2]]) {
			return;
		}
		ans++;
		return;
	}
	for (int i = idx; i <= n; i++) {
		if (nope[idx][i]) continue;
		if (idx == i) continue;
		result.push_back(i);
		combination(size + 1, i);
		result.pop_back();
	}
}
int main() {

	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	// (1 ≤ N ≤ 200, 0 ≤ M ≤ 10,000)
	cin >> n >> m;
	for (int i = 0; i < m; i++) {
		int from, to;
		cin >> from >> to;
		nope[from][to] = true;
		nope[to][from] = true;
	}
	combination(0,0);
	cout << ans;
	return 0;
}
반응형

댓글