77332024-01-10 19:45:17tamasmarkSípálya (55 pont)cpp17Wrong answer 37/55104ms20312 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)
{
    if(a.third==b.third) return a.second<b.second;
    else 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/03ms1836 KiB
2Accepted0/03ms1916 KiB
3Accepted2/23ms1976 KiB
4Accepted2/23ms2248 KiB
5Accepted2/23ms2204 KiB
6Accepted2/23ms2204 KiB
7Accepted3/33ms2336 KiB
8Wrong answer0/17ms3260 KiB
9Wrong answer0/17ms3520 KiB
10Wrong answer0/17ms3472 KiB
11Accepted1/17ms3052 KiB
12Accepted1/17ms3252 KiB
13Accepted1/17ms3272 KiB
14Accepted2/27ms3276 KiB
15Accepted2/27ms3076 KiB
16Accepted2/297ms12860 KiB
17Accepted2/293ms17276 KiB
18Wrong answer0/293ms17328 KiB
19Wrong answer0/397ms20312 KiB
20Wrong answer0/296ms18088 KiB
21Accepted2/294ms17504 KiB
22Accepted2/298ms17632 KiB
23Wrong answer0/298ms17552 KiB
24Accepted2/2104ms17784 KiB
25Accepted2/296ms17856 KiB
26Accepted2/294ms17944 KiB
27Accepted2/293ms18052 KiB
28Wrong answer0/392ms18464 KiB
29Accepted3/392ms18284 KiB
30Wrong answer0/393ms18828 KiB