77322024-01-10 19:41:04tamasmarkSípálya (55 pont)cpp17Wrong answer 37/5597ms20492 KiB
// sipalya'.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <iostream>
#include <deque>
#include <algorithm>
#include <vector>
#include <queue>
#include <climits>

using namespace std;
struct adat
{
    long long first,second,third;
};
bool operator<(const adat a, const adat b)
{
    return a.third<b.third;
}

int main()
{
    long long n, k;
    cin >> n >> k;
    vector<pair<long long,long long>>x(n + 1);
    priority_queue<pair<long long, long long>>p;
    priority_queue<adat>p2;
    long long s1,s2,s3,mini=LONG_MAX,kulombseg;
    for (int i = 1; i <= n; ++i)
    {
        cin >> x[i].first;

        x[i].second = x[i - 1].second + x[i].first;
        p.push({ x[i].first,i });
        while (!p.empty() && p.top().second <= i - k)
        {
            p.pop();
        }
        if (p.top().first-x[i].first<i-p.top().second)
        {
            p2.push({ x[i].first,i,p.top().first-x[i].first+i-p.top().second});
            
        }
        while (!p2.empty() && p2.top().second <= i - k)
        {
            p2.pop();
        }
        if(!p2.empty())
        {
            if(p.top().first==p2.top().first&&p.top().second==p2.top().second) p2.pop();
            if(p.top().second>p2.top().second) p2.pop();
        }

        if (i >= k)
        {
            x[i].second -= x[i - k].first;
            s1 = (p.top().second - (i - k + 1)) + p.top().first;
            if(!p2.empty()) s2 = p.top().first - ((p2.top().second-1) - p.top().second);
            else s2 = p.top().first - (i - p.top().second);
            if (!p2.empty())
            {
                //cout<<p2.top().first;
                kulombseg = p2.top().first - s2+1;
                s2 = p2.top().first - (i - p2.top().second);
                s1 += kulombseg;
                s3 = (((s1 + s2) * k) / 2)-x[i].second;
                //cout<<s3<<"\n";
            }
            else
            {
                 s3 = (((s1 + s2)*k)/2)-x[i].second;
                 //cout<<s3<<"\n";
            }
            mini = min(mini, s3);
        }
    }
    cout << mini;
    return 0;
}
/*
5 3
5 5 6 3 1
*/
// Run program: Ctrl + F5 or Debug > Start Without Debugging menu
// Debug program: F5 or Debug > Start Debugging menu

// Tips for Getting Started: 
//   1. Use the Solution Explorer window to add/manage files
//   2. Use the Team Explorer window to connect to source control
//   3. Use the Output window to see build output and other messages
//   4. Use the Error List window to view errors
//   5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project
//   6. In the future, to open this project again, go to File > Open > Project and select the .sln file
SubtaskSumTestVerdictTimeMemory
base37/55
1Accepted0/03ms1812 KiB
2Accepted0/03ms2004 KiB
3Accepted2/23ms2240 KiB
4Accepted2/23ms2504 KiB
5Accepted2/23ms2480 KiB
6Accepted2/23ms2756 KiB
7Accepted3/33ms2604 KiB
8Wrong answer0/17ms3528 KiB
9Wrong answer0/17ms3784 KiB
10Wrong answer0/17ms3628 KiB
11Accepted1/17ms3384 KiB
12Accepted1/17ms3548 KiB
13Accepted1/17ms3616 KiB
14Accepted2/27ms3744 KiB
15Accepted2/27ms3808 KiB
16Accepted2/289ms13548 KiB
17Accepted2/282ms17700 KiB
18Wrong answer0/282ms17688 KiB
19Wrong answer0/385ms20492 KiB
20Wrong answer0/286ms18256 KiB
21Accepted2/287ms17948 KiB
22Accepted2/290ms17900 KiB
23Wrong answer0/290ms18244 KiB
24Accepted2/297ms18180 KiB
25Accepted2/289ms18244 KiB
26Accepted2/286ms18488 KiB
27Accepted2/285ms18704 KiB
28Wrong answer0/382ms19060 KiB
29Accepted3/385ms18800 KiB
30Wrong answer0/385ms19276 KiB