백준문제풀이/DFS

14888번-연산자 끼워넣기

반응형

문제

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

 

14888번: 연산자 끼워넣기

첫째 줄에 수의 개수 N(2 ≤ N ≤ 11)가 주어진다. 둘째 줄에는 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 100) 셋째 줄에는 합이 N-1인 4개의 정수가 주어지는데, 차례대로 덧셈(+)의 개수, 뺄셈(-)의 개수, 

www.acmicpc.net


접근방법

1) 접근 사고

가능한 모든 연산의 경우를 탐색해줘야 하는 문제였습니다. DFS 알고리즘을 통해 모든 경우의 수를 구해주었습니다.

 

2) 시간 복잡도

O(V + E)

 

3) 배운 점

매개변수를 적절하게 사용하면 코드의 라인이 절약될 수 있다는걸 느낀 문제입니다.

 

4) PS


정답 코드

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
//20210904
#include<bits/stdc++.h>
#define fastio ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define pii pair<int,int>
#define mp(X,Y) make_pair(X,Y)
#define mt(X,Y) make_tuple(X,Y)
#define mtt(X,Y,Z) make_tuple(X,Y,Z)
#define ll long long
#define sz(v) (int)(v).size()
 
using namespace std;
const int MAX = 12;
int arr[MAX];
int op[4];
int n ;
int maxValue = INT_MIN;
int minValue = INT_MAX;
 
int main(void)
{
    fastio;
    cin >> n;
    for(int i = 0; i < n; i++)
        cin >> arr[i];
    for(int i = 0; i < 4; i++)
        cin >> op[i];
 
    //각각의 연산자 갯수가 남아있다면 DFS 탐색을 한다. 
    auto DFS = [&](int plus, int minus, int multiply, int divide, int level, int sum, auto &&DFS) -> void{
        //연산을 다했다면 최대값, 최소값 갱신
        if(level == n){
            maxValue = max(maxValue, sum);
            minValue = min(minValue, sum);
            return ;
        }
 
        if(plus > 0)
            DFS(plus -1, minus, multiply, divide, level + 1, sum + arr[level], DFS);
        if(minus > 0)
            DFS(plus, minus - 1, multiply, divide, level + 1, sum - arr[level], DFS);
        if(multiply > 0)
            DFS(plus, minus, multiply - 1, divide, level + 1, sum * arr[level], DFS);
        if(divide > 0)
            DFS(plus, minus, multiply, divide - 1, level + 1, sum / arr[level], DFS);
    };
 
    DFS(op[0], op[1], op[2], op[3], 1, arr[0], DFS);
    cout << maxValue << "\n";
    cout << minValue << "\n";
    return 0;
}
cs
반응형

'백준문제풀이 > DFS' 카테고리의 다른 글

11725번-트리의 부모 찾기  (0) 2022.04.15
2606번-양  (0) 2022.04.14
11724번-연결 요소의 개수  (0) 2022.04.14