Submission #1014455

#	Time	Username	Problem	Language	Result	Execution time	Memory
1014455		rayan_bd	Stove (JOI18_stove)	C++17	20 / 100	1072 ms	196664 KiB

stove

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> pbds; 


#define getar(ar,n) for(ll i=0;i<n;++i) cin>>ar[i]
#define show(n) cout<<n<<'\n'
#define all(v) v.begin(), v.end()
#define br cout<<"\n"
#define pb push_back
#define nl '\n'
#define yes cout<<"YES\n"
#define no cout<<"NO\n"
#define ret return
#define ll long long
#define ld long double
#define sza(x) ((int)x.size())

const int mxN = 5005;
const ll MOD = 1e9 + 7;
const ll INF = 1e18;
const ld EPS = 1e-9;

ll ar[mxN];
ll dp[mxN][mxN];
ll n;

ll f(ll idx,ll prev,ll k){
	if(idx==n) return 0;
	if(k<=0) return INF;
	//if(dp[idx][k]!=-1) return dp[idx][k];
	ll cut=ar[idx]+1-prev+f(idx+1,ar[idx+1],k-1);
	ll notCut=f(idx+1,prev,k);
	if(idx==n-1){
		notCut+=ar[idx]+1-prev;
	}
	return dp[idx][k]=min(cut,notCut);
}

void solve(ll tc) {
    ll k;cin>>n>>k;
    memset(dp,-1,sizeof(dp));
    getar(ar,n);
    sort(ar,ar+n);
    cout<<f(0,ar[0],k)<<nl;
}

int main() {
    


    ios_base::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    solve(0);
    return 0;
}

• Subtask 1: 20 / 20
• Subtask 2: 0 / 30
• Subtask 3: 0 / 50