O - [ 그리디 ] 볼링공 고르기 - R

난이도 :   풀이시간 :   시간제한 :   기출 : 

[ 문제유형 ] 문제번호 문제제목

 

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개의 공의 무게가 각각 주어질 때,

 

두 사람이 볼링공을 고르는 경우의 수를 구하는 프로그램을 작성하세요.

 

:: 입력 조건

 

1. 첫째 줄에 볼링공의 개수 N, 공의 최대 무게 M이 공백으로 구분되어 각각 자연수 형태로 주어집니다.

 

   (1 <= N <= 1,000, 1 <= M <= 10)

 

2. 둘째 줄에 각 볼링공의 무게 K가 공백으로 구분되어 순서대로 자연수 형태로 주어집니다. (1 <= K <= M)

 

:: 출력 조건

 

첫째 줄에 두 사람이 볼링공을 고르는 경우의 수를 출력합니다.

 

---

  아이디어  

1. n, m 입력받음

2. m 의 범위가 1-10 이기때문에 11칸짜리 배열을 선언

3. 입력받은 볼링공의 무게를 인덱스로 해서 그 칸의 값을 하나씩 증가시킴

4. 입력이 끝나면 이중반복문을 돌면서

무게가 i인 공을 고르는 경우 * 무게가 i보다 무거운 공을 고를 수 있는 경우 를 결과값 변수에 모두 더해줌

5. 결과값 출력

 

  내   코드  

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <string>

using namespace std;

int main() {

	int n, m;
	int w[11] = { 0, };
	
	cin >> n >> m;

	vector<int> v;

	for (int i = 1; i <= n; i++) {
		int k;
		cin >> k;
		w[k]++;
		v.push_back(k);
	}

	int result = 0;

	for (int i = 1; i < m; i++) {
		for (int j = i + 1; j <= m; j++) {
			result += (w[i] * w[j]);
		}
	}

	cout << result;
}

 

  정답  코드  

#include <bits/stdc++.h>

using namespace std;

int n, m;
// 1부터 10까지의 무게를 담을 수 있는 배열
int arr[11];

int main(void) {
    cin >> n >> m;

    for (int i = 0; i < n; i++) {
        int x;
        cin >> x;
        arr[x] += 1;
    }

    int result = 0;

    // 1부터 m까지의 각 무게에 대하여 처리
    for (int i = 1; i <= m; i++) {
        n -= arr[i]; // 무게가 i인 볼링공의 개수(A가 선택할 수 있는 개수) 제외
        result += arr[i] * n; // B가 선택하는 경우의 수와 곱해주기
    }

    cout << result << '\n';
}

 

  리뷰  

나는 2중 for문을 돌아서 연산을 하나하나 따로 해줬는데

그럴 필요없이 해당 무게의 공 갯수를 전체 공 갯수에서 빼준 뒤에 곱하면 연산이 더 간단해짐