A - Frog 1

A - Frog 1

N個の足場があります。

N個の足場にはH個のコストがあり、N_iのコストはH_iになります。

足場 i にいるとき、足場 i+1 または i+2 へジャンプできます。

このとき、ジャンプ先の足場を j とすると、その時、|H_i - H_j|のコストがかかります。

2番目の足場のコストは、

dp2 = min(dp2, dp1 - |H1 - H2|)

が成り立ちます。3番目の足場のコストは、

dp3 = min(dp3, dp2 - |H2 - H3|)
dp3 = min(dp3, dp1 - |H1 - H3|)

が成り立ちます。よって漸化式は、

dp_i = min(dp_i, dp_i-1 + |H_i - H_i-1|) (i-1>=0の時)
dp_i = min(dp_i, dp_i-2 + |H_i - H_i-2|) (i-2>=0の時)

これをコーディングします。

#include <bits/stdc++.h>
 
#define rep(i,n) for(int i=0; i<(n); ++i)
#define repx(i,x,n) for(int i=x; i<(n); ++i)
#define fixed_setprecision(n) fixed << setprecision((n))
#define execution_time(ti) printf("Execution Time: %.4lf sec\n", 1.0 * (clock() - ti) / CLOCKS_PER_SEC);
#define pai 3.1415926535897932384
#define NUM_MAX 2e18
#define NUM_MIN -1e9
 
using namespace std;
using ll = long long;
using P = pair<int,int>;
template<class T> inline bool chmax(T& a, T b){ if(a<b){ a=b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b){ if(a>b){ a=b; return 1; } return 0; }

int main(){
    int N;
    cin >> N;
    vector<int> H(N, 0);
    repx(i, 0, N) cin >> H[i];

    vector<int> dp(N, 1000000009);
    dp[0] = 0;
    for(int i=0; i<N; i++){
        if(i-1>=0) dp[i] = min(dp[i], dp[i-1] + abs(H[i] - H[i-1]));
        if(i-2>=0) dp[i] = min(dp[i], dp[i-2] + abs(H[i] - H[i-2]));
    }
    cout << dp[N-1] << endl;

    return 0;
}