Submission #705172

#	Time	Username	Problem	Language	Result	Execution time	Memory
705172		Alihan_8	수열 (APIO14_sequence)	C++17	28 / 100	2085 ms	32640 KiB

sequence

#include <bits/stdc++.h>
// include <ext/pb_ds/assoc_container.hpp>
// include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
using namespace std;
#define all(x) x.begin(), x.end()
#define pb push_back
// define ordered_set tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update>
#define mpr make_pair
#define ln '\n'
void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
#define int long long
signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int n, k; cin >> n >> k;
    vector <int> p(n);
    for ( auto &i: p ) cin >> i;
    vector <vector<int>> dp(n+1, vector <int> (k+1));
    vector <int> pref(n+1);
    for ( int i = 1; i <= n; i++ ){
        pref[i] = pref[i-1]+p[i-1];
    }
    auto chmax = [&](int &l, int r){
        auto flag = l < r;
        if ( l < r ) swap(l, r);
        return flag;
    };
    vector <vector<int>> par(n+1, vector <int> (k+1));
    for ( int i = 1; i < n; i++ ){
        for ( int j = 1; j <= k; j++ ){
            int &res = dp[i][j];
            for ( int L = 0; L < i; L++ ){
                auto flag = chmax(res, dp[L][j-1]+(pref[i]-pref[L])*(pref[n]-pref[i]));
                if ( flag ) par[i][j] = L;
            }
        }
    }
    int Mx = 0, it = -1;
    for ( int i = 1; i <= n; i++ ){
        if ( chmax(Mx, dp[i][k]) ) it = i;
    }
    cout << Mx << ln;
    while ( k > 0 ){
        cout << it << ' ';
        it = par[it][k--];
    }

    cout << '\n';
}
/*
7 3
4 1 3 4 0 2 3
*/

Compilation message (stderr)

sequence.cpp: In function 'void IO(std::string)':
sequence.cpp:11:29: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                      ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
sequence.cpp:11:70: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                                                               ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

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