브루트 포스/브루트포스(재귀)

에너지 모으기

4gats 2023. 5. 29. 22:14

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

 

16198번: 에너지 모으기

N개의 에너지 구슬이 일렬로 놓여져 있고, 에너지 구슬을 이용해서 에너지를 모으려고 한다. i번째 에너지 구슬의 무게는 Wi이고, 에너지를 모으는 방법은 다음과 같으며, 반복해서 사용할 수 있

www.acmicpc.net

배열을 지운다? vector를 쓰자..

재귀를 할 때 

복사한 벡터의 값을 지운다.

vector<int> b(e);
b.erase(b.begin() + i);

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
#include <iostream>
#include <vector>
 
using namespace std;
 
int go(vector<int>& e)
{
    int n = e.size();
    if (n == 2)
        return 0;
 
    int ans = 0;
 
    for (int i = 1; i < n - 1; i++)
    {
        int energy = e[i - 1] * e[i + 1];
        //e 벡터가 그대로 b에 들어감
        vector<int> b(e);
        b.erase(b.begin() + i);
        energy += go(b);
        if (ans < energy)
            ans = energy;
    }
    return ans;
}
 
int main()
{
    int n;
    vector<int> e;
    cin >> n;
    int x;
    for (int i = 0; i < n; i++)
    {
        cin >> x;
        e.push_back(x);
    }
 
    cout << go(e) << '\n';
    return 0;
}
 cs