Submission #469215

#	Time	Username	Problem	Language	Result	Execution time	Memory
469215		AmirrezaM	Liteh and Newfiteh (INOI20_litehfiteh)	C++17	100 / 100	316 ms	142724 KiB

Main

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;

#define vc vector
#define pb push_back
#define fr first
#define sc second

const int MAXN = 1 << 17 , lg = 19 , inf = 1e8;

int n , a[MAXN] , dp[MAXN] , pd[MAXN][lg][lg];

int main(){
  cin >> n;
  for(int i = 0 ; i < n ; i++) cin >> a[i];
  for(int i = 0 ; i < n ; i++){
	if(a[i] > lg){
	  cout << -1 << endl;
	  return 0;
	}
  }
  for(int i = 0 ; i < n ; i++){
	for(int d = 0 ; d < lg ; d++){
	  if(a[i] > d + 1 or a[i] < d) pd[i][0][d] = inf;
	  else pd[i][0][d] = a[i] - d;
	}
  }
  for(int j = 1 ; j < lg ; j++){
	for(int i = 0 ; i + (1 << j) <= n ; i++){
	  for(int d = 0 ; d < lg-1 ; d++){
		pd[i][j][d] = min(pd[i][j-1][d] + pd[i+(1<<(j-1))][j-1][d] ,
						  pd[i][j-1][d+1] + pd[i+(1<<(j-1))][j-1][d+1] + 1);
		pd[i][j][d] = min(pd[i][j][d] , inf);
	  }
	}
  }
  for(int i = n-1 ; i >= 0 ; i--){
	dp[i] = inf;
	for(int j = 0 ; (1 << j) + i <= n ; j++){
	  dp[i] = min(dp[i] , pd[i][j][0] + dp[i+(1<<j)]);
	}
  }
  if(dp[0] < inf) cout << dp[0] << endl;
  else cout << -1 << endl;
}

• Subtask 1: 30 / 30
• Subtask 2: 70 / 70