Submission #574095

#	Time	Username	Problem	Language	Result	Execution time	Memory
574095		MohammadAghil	K blocks (IZhO14_blocks)	C++14	0 / 100	1 ms	340 KiB

blocks

      #include <bits/stdc++.h>
//   #pragma GCC optimize ("Ofast,unroll-loops")
// #pragma GCC target ("avx2")
    using namespace std;
  typedef long long ll;
   typedef pair<int, int> pp;
    #define er(args ...) cerr << __LINE__ << ": ", err(new istringstream(string(#args)), args), cerr << endl
      #define per(i,r,l) for(int i = (r); i >= (l); i--)
        #define rep(i,l,r) for(int i = (l); i < (r); i++)
           #define all(x) x.begin(), x.end()
              #define sz(x) (int)(x).size()
                  #define pb push_back
                      #define ss second
                           #define ff first
                                   void err(istringstream *iss){}template<typename T,typename ...Args> void err(istringstream *iss,const T &_val, const Args&...args){string _name;*iss>>_name;if(_name.back()==',')_name.pop_back();cerr<<_name<<" = "<<_val<<", ",err(iss,args...);}
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const ll mod = 1e9 + 7, maxn = 1e5 + 1, maxk = 101, lg = 21, inf = ll(1e9) + 5;
ll pw(ll a,ll b,ll md=mod){if(!b)return 1;ll k=pw(a,b>>1ll);return k*k%md*(b&1ll?a:1)%md;}

int dp[maxn][maxk], a[maxn];

int main(){
     cin.tie(0) -> sync_with_stdio(0); 
     int n, k; cin >> n >> k;
     rep(i,1,n+1) cin >> a[i];
     rep(i,1,n+1) dp[i][0] = inf; 
     rep(j,1,k+1){
          vector<pp> stk{{j-1, dp[j-1][j-1]}};
          rep(i,0,j) dp[i][j] = inf;
          rep(i,j,n+1){
               int mn = inf;
               while(sz(stk) && a[stk.back().ff] < a[i]){
                    mn = min(mn, stk.back().ss);
                    stk.pop_back();
               }
               dp[i][j] = mn + a[i];
               if(sz(stk)) dp[i][j] = min(dp[i][j], dp[stk.back().ff][j]);
               stk.pb({i, min(dp[i][j-1], mn)});
          }
     }
     cout << dp[n][k] << '\n';
     return 0; 
}

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