結果

問題 No.1391 ±1 Abs Sum
ユーザー kaagekaage
提出日時 2021-02-14 23:53:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 85 ms / 2,000 ms
コード長 5,605 bytes
コンパイル時間 2,154 ms
コンパイル使用メモリ 155,416 KB
実行使用メモリ 4,996 KB
最終ジャッジ日時 2023-09-29 17:46:04
合計ジャッジ時間 5,326 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 2 ms
4,380 KB
testcase_03 AC 1 ms
4,376 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,376 KB
testcase_09 AC 1 ms
4,380 KB
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 81 ms
4,924 KB
testcase_12 AC 61 ms
4,544 KB
testcase_13 AC 74 ms
4,740 KB
testcase_14 AC 47 ms
4,380 KB
testcase_15 AC 49 ms
4,380 KB
testcase_16 AC 62 ms
4,472 KB
testcase_17 AC 54 ms
4,380 KB
testcase_18 AC 74 ms
4,712 KB
testcase_19 AC 53 ms
4,380 KB
testcase_20 AC 78 ms
4,880 KB
testcase_21 AC 50 ms
4,548 KB
testcase_22 AC 50 ms
4,592 KB
testcase_23 AC 58 ms
4,808 KB
testcase_24 AC 54 ms
4,796 KB
testcase_25 AC 55 ms
4,788 KB
testcase_26 AC 57 ms
4,888 KB
testcase_27 AC 39 ms
4,480 KB
testcase_28 AC 41 ms
4,380 KB
testcase_29 AC 53 ms
4,828 KB
testcase_30 AC 45 ms
4,452 KB
testcase_31 AC 81 ms
4,996 KB
testcase_32 AC 41 ms
4,380 KB
testcase_33 AC 47 ms
4,436 KB
testcase_34 AC 76 ms
4,840 KB
testcase_35 AC 2 ms
4,376 KB
testcase_36 AC 85 ms
4,924 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 2 "library/other/template.hpp"
#define _CRT_SECURE_NO_WARNINGS
#ifdef ONLINE_JUDGE
#pragma GCC target("avx512f")
#elif defined EVAL
#else
#pragma GCC target("avx2")
#endif
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <string.h>
#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <cfloat>
#include <climits>
#include <cmath>
#include <complex>
#include <ctime>
#include <deque>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <list>
#include <map>
#include <memory>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

#define rep(i, n) for (int i = 0; i < (n); i++)
#define REP(i, n) for (int i = 1; i <= (n); i++)
#define all(V) V.begin(), V.end()

using i128 = __int128_t;
using u128 = __uint128_t;
using uint = unsigned int;
using lint = long long;
using ulint = unsigned long long;
using P = std::pair<int, int>;
using LP = std::pair<lint, lint>;

constexpr int INF = INT_MAX / 2;
constexpr lint LINF = LLONG_MAX / 2;
constexpr double eps = DBL_EPSILON;
constexpr double PI = 3.141592653589793238462643383279;

template <class T>
class prique : public std::priority_queue<T, std::vector<T>, std::greater<T>> {
};
template <class F>
inline constexpr decltype(auto) lambda_fix(F&& f) {
	return [f = std::forward<F>(f)](auto&&... args) {
		return f(f, std::forward<decltype(args)>(args)...);
	};
}
template <class T>
constexpr std::vector<T> make_vec(size_t n) {
	return std::vector<T>(n);
}
template <class T, class... Args>
constexpr auto make_vec(size_t n, Args&&... args) {
	return std::vector<decltype(make_vec<T>(args...))>(
		n, make_vec<T>(std::forward<Args>(args)...));
}
template <class T, class U>
constexpr inline bool chmax(T& lhs, const U& rhs) {
	if (lhs < rhs) {
		lhs = rhs;
		return true;
	}
	return false;
}
template <class T, class U>
constexpr inline bool chmin(T& lhs, const U& rhs) {
	if (lhs > rhs) {
		lhs = rhs;
		return true;
	}
	return false;
}
constexpr inline lint gcd(lint a, lint b) {
	while (b) {
		lint c = a;
		a = b;
		b = c % b;
	}
	return a;
}
inline lint lcm(lint a, lint b) { return a / gcd(a, b) * b; }
constexpr bool isprime(lint n) {
	if (n == 1) return false;
	for (int i = 2; i * i <= n; i++) {
		if (n % i == 0) return false;
	}
	return true;
}
template <class T>
constexpr T mypow(T a, lint b) {
	T res(1);
	while (true) {
		if (b & 1) res *= a;
		b >>= 1;
		if (!b) break;
		a *= a;
	}
	return res;
}
constexpr lint modpow(lint a, lint b, lint m) {
	a %= m;
	lint res(1);
	while (b) {
		if (b & 1) {
			res *= a;
			res %= m;
		}
		a *= a;
		a %= m;
		b >>= 1;
	}
	return res;
}
template <class T>
constexpr void printArray(const std::vector<T>& vec, char split = ' ') {
	rep(i, vec.size()) {
		std::cout << vec[i];
		std::cout << (i == (int)vec.size() - 1 ? '\n' : split);
	}
}
template <class InputIter>
constexpr void printArray(InputIter l, InputIter r, char split = ' ') {
	auto rprev = std::prev(r);
	for (auto i = l; i != r; i++) {
		std::cout << *i;
		std::cout << (i == rprev ? '\n' : split);
	}
}
LP extGcd(lint a, lint b) {
	if (b == 0) return {1, 0};
	LP s = extGcd(b, a % b);
	std::swap(s.first, s.second);
	s.second -= a / b * s.first;
	return s;
}
LP ChineseRem(const lint& b1, const lint& m1, const lint& b2, const lint& m2) {
	lint p = extGcd(m1, m2).first;
	lint tmp = (b2 - b1) * p % m2;
	lint r = (b1 + m1 * tmp + m1 * m2) % (m1 * m2);
	return {r, m1 * m2};
}
int LCS(const std::string& a, const std::string& b) {
	auto dp = make_vec<int>(a.size() + 1, b.size() + 1);
	rep(i, a.size()) {
		rep(j, b.size()) {
			chmax(dp[i + 1][j], dp[i][j]);
			chmax(dp[i][j + 1], dp[i][j]);
			if (a[i] == b[j]) chmax(dp[i + 1][j + 1], dp[i][j] + 1);
		}
		chmax(dp[i + 1][b.size()], dp[i][b.size()]);
	}
	rep(j, b.size()) chmax(dp[a.size()][j + 1], dp[a.size()][j]);
	return dp[a.size()][b.size()];
}
template <class T, std::enable_if_t<std::is_convertible<int, T>::value,
									std::nullptr_t> = nullptr>
void compress(std::vector<T>& vec) {
	auto tmp = vec;
	std::sort(all(tmp));
	tmp.erase(std::unique(all(tmp)), tmp.end());
	for (T& i : vec) i = std::lower_bound(all(tmp), i) - tmp.begin();
}
template <class T>
void compress(T* l, T* r) {
	std::vector<T> tmp(l, r);
	std::sort(all(tmp));
	tmp.erase(std::unique(all(tmp)), tmp.end());
	for (auto i = l; i < r; i++) {
		*i = std::lower_bound(all(tmp), *i) - tmp.begin();
	}
}
template <class InputIter>
void compress(InputIter l, InputIter r) {
	std::vector<typename InputIter::value_type> tmp(l, r);
	std::sort(all(tmp));
	tmp.erase(std::unique(all(tmp)), tmp.end());
	for (auto i = l; i < r; i++) {
		*i = std::lower_bound(all(tmp), *i) - tmp.begin();
	}
}
#line 2 "main.cpp"
int N, K;
lint A[200010];
int main() {
	std::cin >> N >> K;
	rep(i, N) std::cin >> A[i];
	lint cnt = 0;
	rep(i, N) {
		if (i < K)
			cnt += A[i] - A[0];
		else {
			cnt -= A[i] - A[0];
		}
	}
	int l = 0, r = N - K;
	lint ans = cnt;
	REP(i, N - 1) {
		cnt += (A[i] - A[i - 1]) * i;
		cnt -= (A[i] - A[i - 1]) * (N - i);
		if (l <= i)
			cnt -= (A[i] - A[i - 1]) * l * 2;
		else
			cnt -= (A[i] - A[i - 1]) * (2 * i - l) * 2;
		if (r <= N - i)
			cnt += (A[i] - A[i - 1]) * r * 2;
		else
			cnt += (A[i] - A[i - 1]) * (2 * (N - i) - r) * 2;
		while (r && std::abs(A[i] - A[l]) > std::abs(A[N - r] - A[i])) {
			cnt += std::abs(A[N - r] - A[i]) * 2;
			cnt -= std::abs(A[i] - A[l]) * 2;
			l++, r--;
		}
		chmin(ans, cnt);
	}
	std::cout << ans << std::endl;
	return 0;
}
0