IT를 품은 기계공학도

[ 삼성 SW역량 평가 14888 ] 연산자 끼워넣기 (C++) 본문

코딩 테스트

[ 삼성 SW역량 평가 14888 ] 연산자 끼워넣기 (C++)

아름돌이 2019. 11. 23. 00:30

[ 관련 지식 ]

1. DFS

https://twpower.github.io/73-how-to-implement-dfs-and-bfs-in-cpp

 

[ 난이도 ]

 

[ 문제 링크 ]

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

 

[ 문제 풀이 ]

문제는 간단하다. 첫째줄에 N에 숫자의 개수를 주어지게 되고 둘째 줄에 N개의 숫자 A가 주어지게 된다.

그리고 그다음 줄에는 (+,-,*./)의 개수가 각각 주어진다. 숫자의 순서는 바뀌지 않고 연산자를 적절히 끼워 맞춰 최댓값과 최솟값을 출력하는 문제이며 연산 순서는 우리가 알고 있는 순서가 아닌 앞에서부터 차례대로 연산한다.

예시로 다음과 같다.

입력예)

2

5 6

0 0 1 0

 

출력 예)

 

30

30

 

나는 이문제를 풀기 위해 DFS를 사용하였다. 코드는 간단하다.

 

[ 소스 코드 ]

 
Coded by 공돌학사,2019.11.23
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// 헤더 파일을 불러옵니다.
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
// 숫자의 최대수는 12입니다.
#define Max_Size 12
 
// N : 숫자의 수 , A : 수들
int N;
int A[Max_Size];
 
// 연산자의 개수
int _Plus, _Minus, _Multiply, _Divide;
 
// 이범위내에서 답이 산출됩니다.
int Max = -1000000000;
int Min = 1000000000;
 
// 입력을 받아옵니다.
void Load_Intput()
{
    cin >> N;
    for (int i = 1; i <= N; ++i)
    {
        cin >> A[i];
    }
    cin >> _Plus;
    cin >> _Minus;
    cin >> _Multiply;
    cin >> _Divide;
    
}
 
// DFS를 선언합니다.
void DFS(int P, int M, int Mu, int Di, int cnt, int sum)
{
    // 연산자가 더이상 없을 경우 Max값을 갱신합니다.
    if (cnt == N)
    {
        Max = max(Max, sum);
        Min = min(Min, sum);
    }
 
    // 각각의 경우 연산자 횟수를 줄이고 cnt를 늘리고 Sum에 새로운 Sum을 넣어주며 깊이탐색을 합니다
    if (P > 0) DFS(P - 1, M, Mu, Di, cnt + 1, sum + A[cnt+1]);
    if (M > 0) DFS(P, M-1, Mu, Di, cnt + 1, sum - A[cnt + 1]);
    if (Mu > 0) DFS(P, M, Mu-1, Di, cnt + 1, sum * A[cnt + 1]);
    if (Di > 0) DFS(P, M, Mu, Di-1, cnt + 1, sum / A[cnt + 1]);
}
 
int main()
{
    // 인풋을 불러옵니다.
    Load_Intput();
 
    // DFS 로 모든 경우를 완전탐색합니다.
    DFS(_Plus, _Minus, _Multiply, _Divide, 1, A[1]);
 
    // 결과를 출력합니다.
    cout << Max << endl << Min << endl;
 
    // 시스템을 정상종료 합니다.
    return 0;
}
cs

 

Comments