Cod sursa(job #2815641)

Utilizator toma_ariciuAriciu Toma toma_ariciu Data 9 decembrie 2021 23:04:04
Problema Oo Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream fin("oo.in");
ofstream fout("oo.out");

const int maxN = 100005;
int dp[maxN], v[maxN];
int n, ans;

int main()
{
    fin >> n;
    for(int i = 1; i <= n; i++)
        fin >> v[i];
    /// caz 1 - iau 1 si 2 => nu pot sa iau n si 3
    dp[1] = v[1];
    dp[2] = v[1] + v[2];
    for(int i = 3; i < n; i++)
        dp[i] = max(dp[i - 1], dp[i - 3] + v[i] + v[i - 1]);
    ans = max(ans, dp[n - 1]);
    /// caz 2 - iau 1 si n => nu pot sa iau n - 1 si 2
    dp[1] = v[1] + v[n];
    dp[2] = v[1] + v[n]; /// sa nu faca urat la dp[-1]
    for(int i = 3; i < n - 1; i++)
        dp[i] = max(dp[i - 1], dp[i - 3] + v[i] + v[i - 1]);
    ans = max(ans, dp[n - 2]);
    /// caz 3 - iau 2 si 3 => nu pot sa iau 1
    dp[1] = 0;
    dp[2] = v[2];
    dp[3] = v[2] + v[3];
    for(int i = 4; i <= n; i++)
        dp[i] = max(dp[i - 1], dp[i - 3] + v[1] + v[i - 1]);
    ans = max(ans, dp[n]);
    fout << ans;
    return 0;
}