結果
問題 | No.1391 ±1 Abs Sum |
ユーザー |
![]() |
提出日時 | 2021-02-12 23:12:08 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,701 bytes |
コンパイル時間 | 1,713 ms |
コンパイル使用メモリ | 169,176 KB |
実行使用メモリ | 6,400 KB |
最終ジャッジ日時 | 2024-07-20 00:44:03 |
合計ジャッジ時間 | 4,841 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 10 WA * 24 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define int long long #define REP(i,m,n) for(int i=(m);i<(n);i++) #define rep(i,n) REP(i,0,n) #define pb push_back #define all(a) a.begin(),a.end() #define rall(c) (c).rbegin(),(c).rend() #define mp make_pair #define endl '\n' #define vec vector<ll> #define mat vector<vector<ll> > #define fi first #define se second #define double long double typedef long long ll; typedef unsigned long long ull; typedef pair<ll,ll> pll; //typedef long double ld; typedef complex<double> Complex; const ll INF=1e9+7; const ll inf=INF*INF; const ll mod=998244353; const ll MAX=140010; signed main(){ ll n,k;cin>>n>>k; vector<ll>a(n); vector<ll>s(n); rep(i,n)cin>>a[i]; for(int i=n-1;i>=0;i--)a[i]-=a[0]; s[0]=a[0]; REP(i,1,n){ s[i]=s[i-1]+a[i]; } if(k==0){ ll ans=0; rep(i,n)ans-=a[i]; cout<<ans<<endl; return 0; }else if(k==n){ ll ans=0; rep(i,n)ans+=a[i]; cout<<ans<<endl; return 0; } ll r=k,l=0; ll now=0; rep(i,n){ if(i<k)now+=a[i]; else now-=a[i]; } ll ans=now; rep(i,n){ while(r<n&&a[r]-a[i]<a[i]-a[l]){ l++; r++; if(l>0)ans=min(ans,-(s[n-1]-s[r-1])+a[i]*(n-r)+s[r-1]-s[i]-a[i]*(r-1-i)+a[i]*(i-l+1)-(s[i]-s[l-1])+s[l-1]-a[i]*l); else ans=min(ans,-s[i]+s[r-1]-s[i]-(s[n-1]-s[r-1])+a[i]*(n-r+i-(r-1-i))); } //if(l>0)ans=min(ans,-(s[n-1]-s[r-1])+a[i]*(n-r)+s[r-1]-s[i]-a[i]*(r-1-i)+a[i]*(i-l+1)-(s[i]-s[l-1])+s[l-1]-a[i]*l); //else ans=min(ans,-s[i]+s[r-1]-s[i]-(s[n-1]-s[r-1])+a[i]*(n-r+i-(r-1-i))); } cout<<ans<<endl; }