볼링공 고르기

문제

두 사람은 서로 무게가 다른 볼링공을 고르려고 한다. 볼링공은 총 N개가 있으며 각 볼링공마다 무게가 적혀 있고, 공의 번호는 1번부터 순서대로 부여된다. 또한 같은 무게의 공이 여러 개 있을 수 있지만, 서로 다른 공으로 간주한다. 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재한다.

N개의 공의 무게가 각각 주어질 때, 두 사람이 볼링공을 고르는 경우의 수를 구하라.

 

입력

첫 째줄 -> 볼링공의 개수 N(1 ~ 1000), 공의 최대 무게 M(1 ~ 10)
둘 째줄 -> 각 볼링공의 무게 K (1 ~ M)

 

출력

두 사람이 볼링공을 고르는 경우의 수

 

풀이

1. 각 무게 별로 볼링공의 수가 몇 개인지 count하는 배열을 만든다.

ex) 볼링공 5개의 무게가 각각 [1 3 2 3 2]로 주어졌을 때

- 무게가 1인 볼링공 : 1개

- 무게가 2인 볼링공 : 2개

- 무게가 3인 볼링공 : 3개

2. A가 특정한 무게의 볼링공을 선택했을 때, 이어서 B가 볼링공을 선택하는 경우를 차례대로 계산한다.

- 남은 총 볼링공 개수(N) - A가 선택한 무게의 볼링공 개수 = B가 선택하는 경우의 수

- 경우의 수 += A가 선택한 무게의 볼링공 개수 * B가 선택하는 경우의 수

n,m=map(int,input().split())
data=list(map(int,input().split()))
counter=[0]*11
result=0
for x in data:
  counter[x]+=1
for i in range(1,m+1):
  n-=counter[i]
  result+=counter[i]*n
print(result)