Cod sursa(job #2391075)

Utilizator eduardcadarCadar Eduard eduardcadar Data 28 martie 2019 17:28:17
Problema Secventa 2 Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <fstream>
using namespace std;
ifstream f("secv2.in");
ofstream g("secv2.out");
int n, k, a[50001], dp[50001], s[50001];
int main()
{
    f >> n >> k;
    for (int i = 1; i <= n; ++i) {
        f >> a[i];
        s[i] = s[i - 1] + a[i];
    }
    int imax = 1, jmax = k, i;
    for (i = 1; i <= k; ++i) dp[i] = dp[i - 1] + a[i];
    for (i = k + 1; i <= n; ++i) {
        if (a[i] > 0 && dp[i - 1] >= 0 && jmax == i - 1) {
            dp[i] = dp[i - 1] + a[i];
            jmax = i;
        }
        else if (s[i] - s[i - k] > dp[i - 1]) {
            imax = i - k + 1, jmax = i;
            dp[i] = s[i] - s[i - k];
        }
        else if (s[i] - s[imax - 1] > dp[i - 1]) {
            jmax = i, dp[i] = s[i] - s[imax - 1];
        }
        else dp[i] = dp[i - 1];
    }
    g << imax << ' ' << jmax << ' ' << dp[n] << endl;
    return 0;
}