Submission #949602

#	Time	Username	Problem	Language	Result	Execution time	Memory
949602		Baizho	K blocks (IZhO14_blocks)	C++14	100 / 100	177 ms	50628 KiB

blocks

#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
  
#define ordered_set tree<int, null_type,less_equal<int>, rb_tree_tag,tree_order_statistics_node_update>
 
// #pragma GCC optimize("Ofast,unroll-loops,fast-math")
// #pragma GCC target("popcnt,avx")
 
 
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<ll,ll> pll;
 
#define sz size()
#define ff first
#define ss second
#define all(a) a.begin(),a.end()
#define pb push_back
 
const int mod = ll(1e9)+7; //(b + (a%b)) % b (to mod -1%(10^9+7) correctly in c++ its -1 but its suppose to be 10^9+6
const ll MOD = 998244353;  // (a%mod)*(binpow(b,mod-2,mod) = (a/b)%mod
const int N = ll(1e5)+100;
const int M = ll(2e5) + 100;
const long long inf = 5e18;
//const long double eps = 1e-15L;
 
ll lcm(ll a, ll b) { return (a / __gcd(a,b))*b; }
 
ll binpow(ll a, ll b, ll m) { ll res=1; a %= m; while(b>0){ if(b&1)res=(res * a) % m; a=(a * a) % m; b/=2; } return res%m;}
 
void Freopen(string Key){ freopen((Key+".in").c_str(), "r", stdin); freopen((Key+".out").c_str(), "w", stdout); }
 
int n, k, a[N], dp[N][101], lo[N], st[30][N];
 
int get(int l, int r) {
    int j = lo[r - l + 1];
    return max(st[j][l], st[j][r - (1 << j) + 1]);
}
 
void Baizho() {
    for(int i = 2; i < N; i ++) lo[i] = lo[i / 2] + 1;
    cin>>n>>k;
    for(int i = 1; i <= n; i ++) {
        cin>>a[i];
        st[0][i] = a[i];
    }
    for(int j = 1; j <= 17; j ++) {
        for(int i = 1; i + (1 << j) - 1 <= n; i ++) {
            st[j][i] = max(st[j - 1][i], st[j - 1][i + (1 << j - 1)]);
        }
    }
    for(int i = 0; i <= n; i ++) for(int j = 0; j <= k; j ++) dp[i][j] = 2e9;
    dp[0][0] = 0;
     for(int kk = 1; kk <= k; kk ++) {
        vector<int> v1, v2;
        for(int i = kk; i <= n; i ++) {
            int cur = dp[i - 1][kk - 1];
            while(!v2.empty() && v2.back() < a[i]) {
                cur = min(cur, v1.back());
                v1.pop_back();
                v2.pop_back();
            } 
            if(v1.empty() || cur + a[i] < v1.back() + v2.back()) {
                v1.pb(cur), v2.pb(a[i]);
            }
            dp[i][kk] = v1.back() + v2.back();
        }
    }
    cout<<dp[n][k]<<"\n";
}
 
signed main() {     
//  Freopen("nondec");
    ios_base::sync_with_stdio(false);   
    cin.tie(0);cout.tie(0); 
    // precalc();
    
    int ttt = 1;
   // cin>>ttt;
 
    for(int i=1; i<=ttt; i++) {Baizho(); }
}

Compilation message (stderr)

blocks.cpp: In function 'void Baizho()':
blocks.cpp:53:64: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
   53 |             st[j][i] = max(st[j - 1][i], st[j - 1][i + (1 << j - 1)]);
      |                                                              ~~^~~
blocks.cpp: In function 'void Freopen(std::string)':
blocks.cpp:35:34: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   35 | void Freopen(string Key){ freopen((Key+".in").c_str(), "r", stdin); freopen((Key+".out").c_str(), "w", stdout); }
      |                           ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
blocks.cpp:35:76: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   35 | void Freopen(string Key){ freopen((Key+".in").c_str(), "r", stdin); freopen((Key+".out").c_str(), "w", stdout); }
      |                                                                     ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 14 / 14
• Subtask 2: 18 / 18
• Subtask 3: 21 / 21
• Subtask 4: 47 / 47