#617240 (C++17) No.1391 ±1 Abs Sum

• 問題: No.1391 ±1 Abs Sum
• ユーザー: 沙耶花
• 提出日時: 2021-02-12 23:11:32
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,594 bytes
• コンパイル時間: 2,196 ms
• コンパイル使用メモリ: 209,676 KB
• 実行使用メモリ: 6,464 KB
• 最終ジャッジ日時: 2023-09-27 06:30:54
• 合計ジャッジ時間: 8,114 ms
• ジャッジサーバーID （参考情報）: judge11 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 4,380 KB
testcase_01	AC	2 ms; 4,380 KB
testcase_02	AC	1 ms; 4,380 KB
testcase_03	AC	2 ms; 4,376 KB
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	RE	-
testcase_09	RE	-
testcase_10	WA	-
testcase_11	RE	-
testcase_12	RE	-
testcase_13	RE	-
testcase_14	RE	-
testcase_15	RE	-
testcase_16	RE	-
testcase_17	RE	-
testcase_18	RE	-
testcase_19	RE	-
testcase_20	RE	-
testcase_21	RE	-
testcase_22	RE	-
testcase_23	RE	-
testcase_24	RE	-
testcase_25	RE	-
testcase_26	RE	-
testcase_27	RE	-
testcase_28	RE	-
testcase_29	RE	-
testcase_30	RE	-
testcase_31	RE	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	AC	1 ms; 4,376 KB
testcase_36	AC	43 ms; 6,348 KB

ソースコード

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/dsu>
#include <atcoder/modint>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define modulo 1000000007
#define mod(mod_x) ((((long long)mod_x+modulo))%modulo)
#define Inf 1000000000000000000
int N,K;
vector<long long> A;

long long f(long long x){
	deque<long long> D0,D1;
	rep(i,N){
		if(i<K)D0.push_back(A[i]);
		else D1.push_back(A[i]);
	}
	
	long long mini = Inf;
	long long cur = 0LL;
	rep(i,N){
		if(A[i]<=x)cur -= x;
		else cur += x;
	}
	rep(i,D0.size()){
		if(D0[i]<=x){
			cur -= D0[i];
			cur += x*2;
		}
		else{
			cur += D0[i];
			cur -= x*2;
		}
	}
	rep(i,D1.size()){
		if(D1[i]<=x){
			cur += D1[i];
		}
		else{
			cur -= D1[i];
		}
	}
	mini = cur;
	
	rep(i,N){
		long long A = D0.front();
		D0.pop_front();
		if(A<=x){
			cur += A;
			cur -= x*2;
			cur += A;
		}
		else{
			cur -= A;
			cur += x*2;
			cur -= A;
		}
		D1.push_back(A);
		
		A = D1.front();
		D1.pop_front();
		
		if(A<=x){
			cur -= A;
			cur -= A;
			cur += x*2;
		}
		else{
			cur += A;
			cur += A;
			cur -= x*2;
		}
		
		mini = min(mini,cur);
		
		
	}
	//if(mini==-89)cout<<x<<endl;
	return mini;
}

int main(){
	
	cin>>N>>K;
	A.resize(N);
	rep(i,N){
		scanf("%lld",&A[i]);
	}
	if(K==0){
		long long ans = 0LL;
		rep(i,N){
			ans += A[i]-A[0];
		}
		cout<<-ans<<endl;
		return 0;
	}
	
	
	long long ans = f(A[0]);
	ans = min(ans,f(A.back()));
	ans = min(ans,f((A[0]+A.back())/2));
	ans = min(ans,f((min(A.back(),A[0]+A.back())/2+1)));
	
	cout<<ans<<endl;
	
	return 0;
}