[백준 2422] 한윤정이 이탈리아에 가서 아이스크림을 사먹는데

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

https://www.acmicpc.net/problem/2422

Tip

•

처음엔 교집합으로 풀었는데, 시간 초과

실패 코드

•

모든 조합을 구하기 위해 iternals의 combinations 사용

◦

구한 조합을 완전 탐색하면서 금지된 조합을 포함하는지 확인

풀이 소스 코드

from itertools import combinations
import sys
si = sys.stdin.readline
n, m = map(int, si().split())
array = [i for i in range(1, n + 1)]
ice = [[False for _ in range(n + 1)] for _ in range(n + 1)]
cnt = 0
for _ in range(m):
a, b = map(int, si().split())
# 이렇게 True로 초기화해주면 조건문에 더 간편하게 쓸 수 있음
ice[a][b] = True
ice[b][a] = True
ice_comb = list(combinations(array, 3))
for i in ice_comb:
if ice[i[0]][i[1]] or ice[i[0]][i[2]] or ice[i[1]][i[2]]:
continue
cnt += 1
print(cnt)
Python