Cod sursa(job #299177)

Utilizator AnDrEwBoYA Andrei AnDrEwBoY Data 6 aprilie 2009 16:51:30
Problema Secventa 2 Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;

const char iname[] = "secv2.in";
const char oname[] = "secv2.out";

const int MAXN = 50010;

#define FOR(i, a, b)  for (int i = (a); i <= (b); ++ i)
#define Max(a, b)  ((a) > (b) ? (a) : (b))

int S[MAXN], n,k;

int main(void) {
    ifstream in(iname);
    in >> n >> k;
    FOR (i, 1, n)  in >> S[i];

    // In rezolvarea de mai jos, S[i] retine suma tuturor valorile intre pozitiile 1 .. i
    // best[i] = Max(S[i] - S[j]) cu j < i =>
    // best[i] = S[i] - Min(S[j]) cu j < i
    int bestSum = -int(2e9), min = 0, idx, beg, end;
    FOR (i, 1, n) {
        S[i] += S[i - 1];
        if (bestSum < S[i] - min && i-idx+1 >= k)
            bestSum = S[i] - min, beg = idx + 1, end = i;
        if (min > S[i])
            min = S[i], idx = i;
    }
    ofstream out(oname);
    out << beg << " " << end << " " << bestSum;

    in.close(), out.close();
    return 0;
}