백준 2309번 C++ 풀이

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

2309번: 일곱 난쟁이

시간 제한	메모리 제한	solved.ac 티어
2초	128MB	브론즈 2

문제

왕비를 피해 일곱 난쟁이들과 함께 평화롭게 생활하고 있던 백설공주에게 위기가 찾아왔다. 일과를 마치고 돌아온 난쟁이가 일곱 명이 아닌 아홉 명이었던 것이다.

아홉 명의 난쟁이는 모두 자신이 "백설 공주와 일곱 난쟁이"의 주인공이라고 주장했다. 뛰어난 수학적 직관력을 가지고 있던 백설공주는, 다행스럽게도 일곱 난쟁이의 키의 합이 100이 됨을 기억해 냈다.

아홉 난쟁이의 키가 주어졌을 때, 백설공주를 도와 일곱 난쟁이를 찾는 프로그램을 작성하시오.

입력

아홉 개의 줄에 걸쳐 난쟁이들의 키가 주어진다. 주어지는 키는 100을 넘지 않는 자연수이며, 아홉 난쟁이의 키는 모두 다르며, 가능한 정답이 여러 가지인 경우에는 아무거나 출력한다.

출력

일곱 난쟁이의 키를 오름차순으로 출력한다. 일곱 난쟁이를 찾을 수 없는 경우는 없다.

---

처음에는 누적합으로 생각을 했다가, 두명의 난쟁이를 찾아야 하기 때문에 부적절하다고 판단했다. 조건을 자세히 보니, 입력은 무조건 9개이며, 시간 제한이 2초이기 때문에, 그냥 브루트포스로 접근하기로 했다.

 

일곱 난쟁이의 키의 합이 100이고, 아홉 난쟁이중에서 일곱명의 난쟁이를 찾아야 하므로, 키의 총합은 무조건 100을 넘는다. 따라서 난쟁이들의 키를 모두 입력받고, 총합에서 난쟁이들의 키를 입력받은 배열에서 하나씩 키의 총합에서 뺀 후에, 100을 넘은 값이 키 배열에 있는지 판단하여 문제를 풀었다.

 

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;

array<int, 9> tall;

// 특정 원소가 들어있는지, 들어있다면 어디에 있는지 return한다. 만약 들어있지 않으면 -1을 return한다.
// except_index = 이 위치의 원소값은 검사하지 않는다. (그 위치는 이미 sum_tall에서 이미 뺀 값이기 때문)
int elem_in_arr(int target, int except_index) { 
    int result = -1;
    for(int i = 0; i < 9; i++) {
        if (i != except_index && tall[i] == target)
            result = i;
    }
    return result;
}

int main () {
    // fast i/o
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int sum_tall = 0;
    int temp_sum; // 한명을 뺐을 때, 키의 총합
    int exceed; // temp_sum - 100, 100을 초과한 만큼의 값. 이 값을 함수를 통해 찾는다.
    int is_in; // elem_in_arr함수값을 받기 위한 변수

    for(int& x : tall) { // 입력
        cin >> x;
        sum_tall += x;
    }

    sort(tall.begin(), tall.end()); // 오름차순 출력을 위한 정렬
    
    for(int i = 0; i < 9; i++) {
        temp_sum = sum_tall - tall[i]; // i번째 키를 빼고
        exceed = temp_sum - 100; // 100을 초과한 만큼의 값을 저장
        is_in = elem_in_arr(exceed, i); // 함수 실행
        
        if (is_in != -1){ // 찾았다면 두명의 난쟁이를 지운다.
            tall[i] = -1;
            tall[is_in] = -1;
            break;
        }
    }
    for(int& x : tall) if (x != -1) cout << x << endl; // 출력
}