Submission #977168

#	Time	Username	Problem	Language	Result	Execution time	Memory
977168		thegamercoder19	Split the sequence (APIO14_sequence)	C++14	0 / 100	396 ms	25816 KiB

sequence

/*
Coded by thegamercoder
Happy Competition and Hacking
*/


#define _CRT_SECURE_NO_WARNINGS

#include <bits/stdc++.h>
#define M_PI       3.14159265358979323846
#define FILER 0
using ll = long long;
using ull = unsigned long long;
using ld = long double;
const ll MOD = pow(10, 9) + 7;
const ll INFL = 0x3f3f3f3f3f3f3f3f;
const ull INFUL = 0x3f3f3f3f3f3f3f3f;
const ll INFT = 0x3f3f3f3f;
const ll MAX = 401;
const ll MODD = 998244353;
const ll BL = 1;
const double EPS = 1e-10;
#define V vector
#define pll pair<ll, ll>
#define pull2 pair<ull,ull>
#define MS multiset
#define M map
#define Q queue
#define PQ priority_queue
#define IOF ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define FOR(typ,i,a,b,c) for(typ i = a; i < b; i += c)
#define FORR(typ,i,a,b,c) for(typ i = a; i > b; i -= c)
#define FORA(a,i) for(auto i : a)
#define FORAE(a,i) for(auto &i : a)
#define all(v) v.begin(),v.end()
#define sorta(a) sort(all(a))
#define sortd(a) sort(all(a), greater<ll>())
#define setp(x) setprecision(x)<<fixed
#define RET return
#define log(a,b) log(b)/log(a)
#define WH(s) while(s)
#define WHI(t) WH(t--)
#define YES cout<<"YES"<<endl;
#define NO cout<<"NO"<<endl;
#define Yes cout<<"Yes"<<endl;
#define No cout<<"No"<<endl;
#define YESNO(s) cout<<(s?"YES":"NO")<<endl;
#define YesNo(s) cout<<(s?"Yes":"No")<<endl;
#define TYP 0
using namespace std;
V<ll> a, pf, res,  dp, ndp;
V<V<int>> p;
V<V<ll>> fdp;
ll n, k;

struct line
{
    mutable ll k, b, p, i;
    bool operator<(const line& o) const {
        RET k < o.k;
    }
    bool operator<(ll x)const
    {
        RET p < x;
    }
};
struct lineCont : multiset<line, less<>>
{
    ll div(ll a, ll b) { RET a / b - ((a ^ b) < 0 && (a % b)); }
    bool sect(iterator a, iterator b)
    {
        if (b == end()) { a->p = INFL; RET 0; }
        if (a->k == b->k)  a->p = a->b > b->b ? INFL : -INFL;
        else a->p = div(b->b - a->b, a->k - b->k);
        RET a->p >= b->p;
    }
    void add(ll k, ll b, ll id)
    {
        auto z = insert({ k,b,0,id }), y = z++, x = y;
        WH(sect(y, z)) z = erase(z);
        if (x == begin())RET;
        if (sect(--x, y))sect(x, y = erase(y));
        WH((y = x) != begin() && (--x)->p >= y->p) sect(x, erase(y));
    }
    pll query(ll x)
    {
        auto l = *lower_bound(x);
        RET{ l.k * x + l.b, l.i };
    }
};


void solve()
{
    cin >> n >> k;
    a.resize(n + 1); pf.resize(n + 1);
    FOR(ll, i, 1, n + 1, 1)cin >> a[i], pf[i] = pf[i - 1] + a[i];
    dp.resize(n+1); // dp[i][j-1]
    p.resize(n + 1, V<int>(k + 1)); fdp.resize(n + 1, V<ll>(k + 1));
    lineCont cht;
    //  pf[c] (m) * pf[i] (x) + (dp[c][j-1] - pf[c] * pf[c]) (c)
    FOR(ll, j, 1, k + 1, 1)
    {
        ndp.assign(n + 1, 0);
        FOR(ll, i, 1, n + 1, 1)

        {
            if (i > 1)
            {
                auto l = cht.query(pf[i]);
                ndp[i] = l.first;
                fdp[i][j] = l.first;
                p[i][j] = l.second;
            }
            cht.add(pf[i], fdp[i][j - 1] - pf[i] * pf[i], i);
        }
        FOR(ll, i, 1, n + 1, 1)
        {
            //if (fdp[i][j] != ndp[i]) cout << "NOT CORRECT (dp[i][j]) " << i << ' ' << j << endl;
            //else if (fdp[i][j - 1] != dp[i])cout << "NOT CORRECT (dp[i][j]) " << i << ' ' << j << endl;
            //else cout << "CORRECT (dp[i][j] and dp[i][j-1]) " << i << ' ' << j << endl;
        }
        swap(ndp, dp);
    }
        

    assert(dp[n] == fdp[n][k]);
    cout << dp[n] << endl;
    ll j = k, i = n;
    WH(j)
    {
        cout << p[i][j] << ' ';
        j--;
        i = p[i][j];
    }
    cout << endl;

}
void init()
{

    if (FILER)
    {
        freopen("pump.in", "r", stdin);
        freopen("pump.out", "w", stdout);
    }
}

void handle()
{
    init();
    ll t = 1;
    if (TYP)cin >> t;
    WHI(t)solve();
}

int main()
{
    IOF
        handle();
    RET 0;
}

Compilation message (stderr)

sequence.cpp: In function 'void init()':
sequence.cpp:144:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  144 |         freopen("pump.in", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~
sequence.cpp:145:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
  145 |         freopen("pump.out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 0 / 11
• Subtask 3: 0 / 11
• Subtask 4: 0 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29