結果

問題 No.1391 ±1 Abs Sum
ユーザー east1016east1016
提出日時 2021-02-12 22:50:47
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 4,758 bytes
コンパイル時間 2,454 ms
コンパイル使用メモリ 207,884 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-19 23:55:06
合計ジャッジ時間 3,984 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 24 ms
5,376 KB
testcase_12 AC 20 ms
5,376 KB
testcase_13 AC 23 ms
5,376 KB
testcase_14 AC 16 ms
5,376 KB
testcase_15 AC 17 ms
5,376 KB
testcase_16 AC 21 ms
5,376 KB
testcase_17 AC 19 ms
5,376 KB
testcase_18 AC 24 ms
5,376 KB
testcase_19 AC 17 ms
5,376 KB
testcase_20 AC 24 ms
5,376 KB
testcase_21 AC 17 ms
5,376 KB
testcase_22 AC 16 ms
5,376 KB
testcase_23 AC 19 ms
5,376 KB
testcase_24 AC 18 ms
5,376 KB
testcase_25 AC 18 ms
5,376 KB
testcase_26 AC 19 ms
5,376 KB
testcase_27 AC 14 ms
5,376 KB
testcase_28 AC 14 ms
5,376 KB
testcase_29 AC 18 ms
5,376 KB
testcase_30 AC 16 ms
5,376 KB
testcase_31 AC 27 ms
5,376 KB
testcase_32 WA -
testcase_33 WA -
testcase_34 AC 25 ms
5,376 KB
testcase_35 AC 2 ms
5,376 KB
testcase_36 AC 27 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#include <atcoder/all>
//using namespace atcoder;
//using mint = modint;

#include <bits/stdc++.h>
#define int long long
#define sint signed
#define endl "\n" // fflush(stdout);
#define ALL(v) (v).begin(),(v).end()
#define Vi vector<int>
#define VVi vector<Vi>
#define VVVi vector<VVi>
#define Vm vector<mint>
#define VVm vector<Vm>
#define Vs vector<string>
#define Vd vector<double>
#define Vc vector<char>
#define Pii pair<int,int>
#define Pdd pair<double,double>
#define VPii vector<Pii>
#define Tiii tuple<int,int,int>
#define VTiii vector<Tiii>
#define PQi priority_queue<int>
#define PQir priority_queue<int,vector<int>,greater<int>>
#define pb push_back
#define mp make_pair
#define mt make_tuple
#define itos to_string
#define stoi stoll
#define FI first
#define SE second
#define cYES cout<<"YES"<<endl
#define cNO cout<<"NO"<<endl
#define cYes cout<<"Yes"<<endl
#define cNo cout<<"No"<<endl
#define cyes cout<<"yes"<<endl
#define cno cout<<"no"<<endl
#define sortr(v) sort(v,greater<>())
#define rep(i,a,b) for(int i=a;i<b;i++)
#define repeq(i,a,b) for(int i=a;i<=b;i++)
#define repreq(i,a,b) for(int i=a;i>=b;i--)
#define dem(a,b) ((a+b-1)/(b))
#define INF 3000000000000000000 // 3.0*10^18
#define MAX LLONG_MAX
#define PI acos(-1.0L)

using namespace std;

/* debug */
template <typename T>
ostream& operator<<(ostream& os,const vector<T> &V){int N=V.size();
if(N==0){os<<'.';return os;}rep(i,0,N-1){os<<V[i]<<' ';}os<<V[N-1];return os;}
template <typename T>
ostream& operator<<(ostream& os,const vector<vector<T>> &V){
int N=V.size();rep(i,0,N-1)os<<V[i]<<endl;os<<V[N-1];return os;}
template <typename T,typename S>
ostream& operator<<(ostream& os, pair<T,S> const&P){os<<P.FI<<' '<<P.SE;return os;}
//ostream& operator<<(ostream& os, mint const&M){os<<M.val();return os;}

/* useful */
template<typename T>void Vin(vector<T> &v){int n=v.size();rep(i,0,n)cin>>v[i];}
int scomb(int n, int r){if(r<0||r>n)return 0;if((n-r)<r)r=n-r; // nCr
int a=1;for(int i=n;i>n-r;--i){a=a*i;}for(int i=1;i<r+1;++i){a=a/i;}return a;}
int digit_sum(int n){int ret=0; while(n>0){ret+=n%10;n/=10;}return ret;}
int digit(int k,int i){string s = itos(k);return s[s.size()-i]-'0';}
template<typename T>void press(T &v){v.erase(unique(ALL(v)),v.end());}
int SMALLER(Vi &a,int x){return lower_bound(a.begin(),a.end(),x)-a.begin();}
int orSMALLER(Vi &a,int x){return upper_bound(a.begin(),a.end(),x)-a.begin();}
int BIGGER(Vi &a,int x){return a.size()-orSMALLER(a,x);}
int orBIGGER(Vi &a,int x){return a.size()-SMALLER(a,x);}
int COUNT(Vi &a,int x) {return upper_bound(ALL(a),x)-lower_bound(ALL(a),x);}
int maxind(Vi &a){return max_element(ALL(a))-a.begin();}
int minind(Vi &a){return min_element(ALL(a))-a.begin();}
template<typename T>bool chmax(T &a,T b) {if(a<b){a=b;return 1;}return 0;}
template<typename T>bool chmin(T &a,T b) {if(a>b){a=b;return 1;}return 0;}
/*
Vi zip(Vi b){int Z=b.size();
Pii p[Z+10];int a=b.size();Vi l(a);for(int i=0;i<a;i++)
p[i]=mp(b[i],i);sort(p,p+a);int w=0;for(int i=0;i<a;i++)
{if(i&&p[i].first!=p[i-1].first)w++;l[p[i].second]=w;}return l;}
*/
//Vi vis(Vi &v){Vi S(v.size()+1);rep(i,1,S.size())S[i]+=v[i-1]+S[i-1];return S;}

//const int MOD = 998244353;
const int MOD = 1000000007;
const double EPS = 1e-10;

void init(){cin.tie(0);cout.tie(0);ios::sync_with_stdio(0);
cout<<fixed<<setprecision(12);//mint::set_mod(MOD);
}

/************************************ START ************************************/

void sol() {
  int n,K;cin >> n >> K;
  Vi a(n);Vin(a);
  if(n == 1) {
    cout << 0 << endl;
    return;
  }
  if(K == 0) {
    int now = a[0];
    int ans = 0;
    rep(i,0,n) {
      ans -= abs(a[i]-now);
    }
    cout << ans << endl;
    return;
  }
  if(K == n) {
    int now = a[n/2];
    int ans = INF;
    int pre = 0;
    rep(i,0,n) {
      pre += abs(a[i]-now);
    }
    chmin(ans,pre);
    now = a[n/2-1];
    pre = 0;
    rep(i,0,n) {
      pre += abs(a[i]-now);
    }
    chmin(ans,pre);
    cout << ans << endl;
    return;
  }
  int ans = INF;
  int po = n-K;
  int indr = n-po,indl = 0;
  int a0 = a[0];
  int sum = 0;
  int L=0,R=0,Lct=0,Rct=0;
  rep(i,0,n) {
    a[i] = abs(a[i]-a0);
    sum += a[i];
  }
  repreq(i,n-1,n-po) {
    R += a[i];
    Rct++;
  }
  int nw = 0;
  int ll=0,rr=n;
  rep(i,0,n) {
    int sa = abs(nw-a[i]);
    sum -= sa*rr;
    sum += sa*ll;
    rr--;
    ll++;
    while(indr < n && abs(a[indl]-a[i]) > abs(a[indr]-a[i])) {
      R -= a[indr];
      L += a[indl];
      indl++;
      indr++;
      Lct++;
      Rct--;
    }
    int pre = sum;
    int lll,rrr;
    rrr = abs(R - Rct*a[i]);
    lll = abs(L - Lct*a[i]);
    pre -= (rrr+lll)*2;
    chmin(ans,pre);
    nw = a[i];
  }
  //cout << a << endl;
  cout << ans << endl;
}

signed main() {
  init();
  sol();
  return 0;
}
0