Submission #636622

#	Time	Username	Problem	Language	Result	Execution time	Memory
636622		stoyan_malinin	Peru (RMI20_peru)	C++14	Compilation error	0 ms	0 KiB

peru

#include "peru.h"
#ifdef _LOCAL_
    #include "grader.cpp"
#endif // _LOCAL_

#include <cassert>
#include <vector>
#include <deque>
#include <set>

using namespace std;

int rmq(int l, int r, const int *a)
{
    int maxVal = a[l];
    for(int i = l+1;i<=r;i++) maxVal = max(maxVal, a[i]);

    return maxVal;
}

int convertToAns(const vector<long long> &dp)
{
    long long p23 = 1, ans = 0;
    const long long mod = 1e9 + 7;

    for(int i = dp.size()-1;i>=0;i--)
    {
        ans = (ans + (dp[i]%mod)*p23)%mod;
        p23 = (p23*23)%mod;
    }

    return ans;
}

int solve(int n, int k, int* a)
{
    vector <long long> dp(n);

    dp[0] = a[0];
    deque <int> dq = {0};
    deque <pair <long long, int>> values;

    long long prefMaxVal = a[0];
    for(int i = 1;i<n;i++)
    {
        dp[i] = 1e18;
        prefMaxVal = max(prefMaxVal, (long long)a[i]);
        if(i<k) dp[i] = min(dp[i], prefMaxVal);

        while(dq.empty()==false && i-dq.front()>=k)
        {
            int j = dq.front();
            dq.pop_front();

            if(dq.empty()==false)
            {
                if(values.empty()==false && values.front()==make_pair(dp[j]+a[dq.front()], j)) values.pop_front();
            }
        }

        while(dq.empty()==false && a[dq.back()]<=a[i])
        {
            if(dq.size()>=2)
            {
                if(values.empty()==false && values.back()==make_pair(dp[*prev(prev(dq.end()))] + a[dq.back()], *prev(prev(dq.end()))))
                    values.pop_back();

            }
            dq.pop_back();
        }

        if(dq.empty()==false)
        {
            while(values.empty()==false && values.back().first>=dp[dq.back()]+a[i])
                values.pop_back();

            values.emplace_back(dp[dq.back()]+a[i], dq.back());
        }
        dq.push_back(i);

        int last = max(i-k, 0);
        if(dq.front()>last) dp[i] = min(dp[i], dp[last] + a[dq.front()]);
        if(values.empty()==false) dp[i] = min(dp[i], values.front().first);

        assert(values.size()<=dq.size()-1);

        cout << "dq: ";for(int x: dq) cout << x << " ";
        cout << '\n';
        cout << "values: ";for(auto item: values) cout << "(" << item.first << ", " << item.second << ") ";
        cout << '\n';
        cout << "dp[" << i << "] = " << dp[i] << '\n';

    }

    return convertToAns(dp);
}

Compilation message (stderr)

peru.cpp: In function 'int solve(int, int, int*)':
peru.cpp:87:9: error: 'cout' was not declared in this scope
   87 |         cout << "dq: ";for(int x: dq) cout << x << " ";
      |         ^~~~
peru.cpp:10:1: note: 'std::cout' is defined in header '<iostream>'; did you forget to '#include <iostream>'?
    9 | #include <set>
  +++ |+#include <iostream>
   10 |