제출 #645791

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
645791		kith14	Safety (NOI18_safety)	C++17	4 / 100	2 ms	724 KiB

safety

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define db double
#define pairll pair<ll,ll>
#define lpairll pair<ll,pairll>

#define repp(i,a,b) for (ll i = a; i <= b; i++)
#define repz(i,a,b) for (ll i = a; i < b; i++)
#define repm(i,a,b) for (ll i = a; i >= b; i--)
#define fr first
#define sc second
#define mp make_pair
#define pb push_back

const ll N = 5e3+5, MOD = 1e9+7;
ll tc = 1, n, m, ar[N], br[N], dp[2][N];
string s, s1, s2, ye = "YA", no = "TIDAK";

void input(){
  cin >> n >> m;
  repp(i,1,n) cin >> ar[i];
}

void solve(){
  ll lm = 5000;
  repp(i,0,lm){
    dp[n%2][i] = abs(i-ar[n]);
  }

  repm(i,n-1,1){
    ll cur = (i)%2, pre = (i+1)%2;
    deque<ll> dq;
    repp(j,0,m){
      while(dq.size() && dp[pre][j] < dp[pre][dq.back()]){
        dq.pop_back();
      }
      dq.push_back(j);
    } 
    repp(j,0,lm){
      ll btl = max(0LL,j-m), btr = min(j+m,lm);
      while(dq.size() && dq.front() < btl) dq.pop_front();
      while(dq.size() && dp[pre][btr] < dp[pre][dq.back()]){
        dq.pop_back();
      }
      dq.push_back(btr);
      dp[cur][j] = dp[pre][dq.front()]+abs(ar[i]-j);
    }
  }

  ll ans = 1e18;
  repp(i,0,lm){
    ans = min(ans,dp[1][i]);
  }
  cout << ans << endl;
}

int main(){
  ios_base::sync_with_stdio(0);
  cin.tie(NULL);
  cout.tie(NULL);
  //cin >> tc;
  while(tc--){
    input();
    solve();
  }
}

• Subtask 1: 0 / 3
• Subtask 2: 4 / 4
• Subtask 3: 0 / 9
• Subtask 4: 0 / 5
• Subtask 5: 0 / 6
• Subtask 6: 0 / 11
• Subtask 7: 0 / 11
• Subtask 8: 0 / 22
• Subtask 9: 0 / 29