[이코테] 그리디_볼링공 고르기 (python)

문제

나의 풀이

1. 조합 사용: 성공

# 이코테_그리디_볼링공 고르기
from itertools import combinations
n, m = map(int, input().split()) # n: 볼링공 개수, m: 공의 최대 무게
ball = list(map(int, input().split()))

count = 0 # 총 경우의 수 = 0으로 초기화

for i in combinations(ball, 2): # 조합
	if i[0] != i[1]: # a와 b의 공 무게가 다를 때
		count += 1
        
print(count)

2. 이중 for문 사용: 성공

n, m = map(int, input().split()) # 공의 개수 n, 공의 최대 무게 m
ball = list(map(int, input().split())) # 공 list

result = 0 # 두 사람이 공을 고르는 경우의 수를 출력할 변수

for i in range(n): # a가 i번째 공을 고르면
	for j in range(1, n - i): # b가 j번째 공을 골랐을 때
		if ball[i] != ball[i + j]: # 두 공의 무게가 다르면
			result += 1 # 경우의 수 += 1
            
print(result)