結果

問題 No.1391 ±1 Abs Sum
コンテスト
ユーザー chocorusk
提出日時 2021-02-12 21:49:47
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 118 ms / 2,000 ms
コード長 1,151 bytes
コンパイル時間 1,550 ms
コンパイル使用メモリ 123,328 KB
最終ジャッジ日時 2025-01-18 18:21:36
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <stack>
#define popcount __builtin_popcount
using namespace std;
typedef long long int ll;
typedef pair<int, int> P;
ll a[200020];
ll s[200020];
int main()
{
    int n, k; cin>>n>>k;
    for(int i=0; i<n; i++){
        cin>>a[i];
        s[i+1]=s[i]+a[i];
    }
    ll ans=0;
    for(int i=0; i<n; i++){
        if(i<k) ans+=abs(a[0]-a[i]);
        else ans-=abs(a[0]-a[i]);
    }
    int l=0;
    for(int i=1; i<n; i++){
        ll t=a[i]*(i-l)-(s[i]-s[l])+s[l+k]-s[i]-a[i]*(l+k-i);
        while(l+k<n && t>a[i]*(i-l-1)-(s[i]-s[l+1])+s[l+k+1]-s[i]-a[i]*(l+k+1-i)){
            l++;
            t=a[i]*(i-l)-(s[i]-s[l])+s[l+k]-s[i]-a[i]*(l+k-i);
        }
        ans=min(ans, t*2-(a[i]*i-s[i]-a[i]*(n-i)+(s[n]-s[i])));
    }
    cout<<ans<<endl;
    return 0;
}
0