본문 바로가기

Algorithm_이코테

[이코테 DP] 01. 개미전사

728x90
반응형

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
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
#define MAX 1001
int n;
vector<int> ar;
vector<int> dp(MAX);
int main()
{    
    ios::sync_with_stdio(false); // 계산시간 단축 // 문제마다 오류 유무 다름
    cin.tie(NULL); cout.tie(NULL);// 입출력 시간 단축
    cin >> n;
    for (int i = 0; i < n; i++) {
        int num;
        cin >> num;
        ar.push_back(num);
    }
    
    dp[0= ar[0];
    dp[1= max(ar[0], ar[1]);
    for (int i = 2; i < n; i++) {
        dp[i] = max(dp[i - 1], dp[i - 2]+ar[i]);
    }
    cout << dp[n - 1];
    return 0;
}
cs
반응형