Cod sursa(job #2489504)

Utilizator andrei42Oandrei42O andrei42O Data 8 noiembrie 2019 20:43:39
Problema Secventa Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("secventa.in");
ofstream g("secventa.out");
const int N = 500010;
deque <pair<int, int>> dq;
deque <pair<int, int>>::iterator it;
int n, k, x, S, start, s[N], v[N], maxi;
int main()
{
    f >> n >> k;
    for(int i = 1; i <= k; i++)
    {
        f >> v[i];
        dq.push_back({v[i], i});
        s[1] += v[i];
    }
    sort(dq.begin(), dq.end());
    tie(maxi, S) = dq.front();
    start = 1;
    for(int i = k + 1; i <= n; i++)
    {
        f >> v[i];
        s[i - k + 1] = s[i - k] - v[i - k] + v[i];
        while(dq.front().second < i - k + 1 && !dq.empty())
            dq.pop_front();
        int st = 0, dr = dq.size();
        while(st < dr)
        {
            int mi = (st + dr) / 2;
            if(dq[mi].first < v[i])
                st = mi + 1;
            else
                dr = mi;
        }
        if(dq[st].second < i && dq[st].first == v[i])
            st++;
        it = dq.begin() + st;
        dq.insert(it, make_pair(v[i], i));
//        for(int j = 0; j < dq.size(); j++)
//            cout << "(" << dq[j].first <<", " << dq[j].second <<"), ";
//        cout << endl;
        int tempV, id;
        tie(tempV, id) = dq.front();
        if(maxi < tempV)
        {
            maxi = tempV;
            start = i - k + 1;
            S = s[start];
        }
        else if(maxi == tempV && s[i - k + 1] > S)
        {
            S = s[i - k + 1];
            start = i - k + 1;
        }
    }
    g << start << ' ' << start + k - 1 << ' ' << maxi;
    return 0;
}