제한 조건
- 풀이 시간 : 30분
- 시간 제한 : 1초
문제
A, B 두 사람이 볼링을 치고 있습니다. 두 사람은 서로 무게가 다른 볼링공을 고르려고 합니다.
볼링공은 총 N개가 있으며 각 볼링공마다 무게가 적혀 있고, 공의 번호는 1번부터 순서대로 부여됩니다. 또한 같은 무게의 공이 여러개 있을 수 있지만, 서로 다른 공으로 간주합니다. 볼링공의 무게는 1부터 M까지의 자연수 형태로 존재합니다.
예를 들어 N이 5이고, M이 3이며 각각의 무게가 차례대로 1, 3, 2, 3, 2일때 각 공의 번호가 차례대로 1번부터 5번까지 부여됩니다. 이때 두 사람이 고를 수 있는 볼링공 번호의 조합을 구하면 다음과 같습니다.
(1번, 2번), (1번, 3번), (1번, 4번), (1번, 5번), (2번, 3번), (2번, 5번), (3번, 4번), (4번, 5번)결과적으로 두 사람이 공을 고르는 경우의 수는 8가지입니다.
N개의 공의 무게가 각각 주어질 때, 두 사람이 볼링공을 고르는 경우의 수를 구하는 프로그램을 작성하세요.
입력 조건
- 첫째 줄에 볼링공의 개수 N, 공의 최대 무게 M이 공백으로 구분되어 각각 자연수 형태로 주어집니다. (1 <= N <= 1,000, 1 <= M <= 10)
- 둘째 줄에 각 볼링공의 무게 K가 공백으로 구분되어 순서대로 자연수 형태로 주어집니다. (1 <= K <= M)
출력 조건
- 첫째 줄에 두 사람이 볼링공을 고르는 경우의 수를 출력합니다.
예시
| 입력 | 출력 |
|---|---|
| 5 3 1 3 2 3 2 | 8 |
| 8 5 1 5 4 3 2 4 5 2 | 25 |
풀이
이 문제에서의 포인트는 서로 다른 무게와 번호의 조합이다.
즉, 처음부터 공을 하나씩 꺼내고 꺼낸 공 번호의 이후의 공들을 꺼내와서 무게가 동일하지 않은 번호들만 카운트하면 된다. 이 것은 2중 for문을 활용하여 간단하게 구현이 가능하다.
시간 복잡도는 연산 횟수가 (n-1) + (n-2) + … + 2 + 1 = n(n-1)/2이므로 O(n^2)가 된다. 공의 갯수가 1000개이므로 제한 시간 안에 풀이가 가능하다.
코드
느리더라도 하나씩 천천히. 하지만 꾸준히