結果
問題 | No.1391 ±1 Abs Sum |
ユーザー |
|
提出日時 | 2021-02-12 22:07:59 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 27 ms / 2,000 ms |
コード長 | 3,827 bytes |
コンパイル時間 | 1,950 ms |
コンパイル使用メモリ | 197,672 KB |
最終ジャッジ日時 | 2025-01-18 18:37:51 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
//@formatter:off#include<bits/stdc++.h>#define overload4(_1,_2,_3,_4,name,...) name#define rep1(i,n) for (ll i = 0; i < ll(n); ++i)#define rep2(i,s,n) for (ll i = ll(s); i < ll(n); ++i)#define rep3(i,s,n,d) for(ll i = ll(s); i < ll(n); i+=d)#define rep(...) overload4(__VA_ARGS__,rep3,rep2,rep1)(__VA_ARGS__)#define rrep(i,n) for (ll i = ll(n)-1; i >= 0; i--)#define all(a) a.begin(),a.end()#define rall(a) a.rbegin(),a.rend()#define popcount(x) __builtin_popcount(x)#define pb push_back#define eb emplace_back#ifdef __LOCAL#define debug(...) { cout << #__VA_ARGS__; cout << ": "; print(__VA_ARGS__); cout << flush; }#else#define debug(...) void(0)#endif#define INT(...) int __VA_ARGS__;scan(__VA_ARGS__)#define LL(...) ll __VA_ARGS__;scan(__VA_ARGS__)#define STR(...) string __VA_ARGS__;scan(__VA_ARGS__)#define CHR(...) char __VA_ARGS__;scan(__VA_ARGS__)#define DBL(...) double __VA_ARGS__;scan(__VA_ARGS__)#define LD(...) ld __VA_ARGS__;scan(__VA_ARGS__)using namespace std;using ll = long long;using ld = long double;using P = pair<int,int>;using LP = pair<ll,ll>;using vi = vector<int>;using vvi = vector<vector<int>>;using vl = vector<ll>;using vvl = vector<vector<ll>>;using vd = vector<double>;using vvd = vector<vector<double>>;using vs = vector<string>;using vc = vector<char>;using vvc = vector<vector<char>>;using vb = vector<bool>;using vvb = vector<vector<bool>>;using vp = vector<P>;using vvp = vector<vector<P>>;template<class S,class T> istream& operator>>(istream &is,pair<S,T> &p) { return is >> p.first >> p.second; }template<class S,class T> ostream& operator<<(ostream &os,const pair<S,T> &p) { return os<<'{'<<p.first<<","<<p.second<<'}'; }template<class T> istream& operator>>(istream &is,vector<T> &v) { for(T &t:v){is>>t;} return is; }template<class T> ostream& operator<<(ostream &os,const vector<T> &v) { os<<'[';rep(i,v.size())os<<v[i]<<(i==int(v.size()-1)?"":","); return os<<']';}template<class T> void vecout(const vector<T> &v,char div='\n') { rep(i,v.size()) cout<<v[i]<<(i==int(v.size()-1)?'\n':div);}template<class T> bool chmin(T& a,T b) {if(a > b){a = b; return true;} return false;}template<class T> bool chmax(T& a,T b) {if(a < b){a = b; return true;} return false;}void scan(){}template <class Head, class... Tail> void scan(Head& head, Tail&... tail){ cin >> head; scan(tail...); }template<class T> void print(const T& t){ cout << t << '\n'; }template <class Head, class... Tail> void print(const Head& head, const Tail&... tail){ cout<<head<<' '; print(tail...); }template<class... T> void fin(const T&... a) { print(a...); exit(0); }const string yes[] = {"no","yes"};const string Yes[] = {"No","Yes"};const string YES[] = {"NO","YES"};const int inf = 1001001001;const ll linf = 1001001001001001001;//@formatter:onint main() {ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);INT(n, k);vl a(n);cin >> a;vl sum(n + 1);rep(i, n) sum[i + 1] = sum[i] + a[i];if (k == 0) {ll ans = linf;chmin(ans, a[0] * n - sum[n]);chmin(ans, -a.back() * n + sum[n]);fin(ans);}ll ans = linf;int l = 0, r = k;rep(i, n) {while (r < n) {ll dis = 0;dis -= abs(a[i] - a[l]);dis += abs(a[i] - a[r]);if (dis >= 0) break;l++, r++;}assert(l <= i and i < r);ll now = 0;{now += sum[l];now -= a[i] * l;now -= sum[i] - sum[l];now += a[i] * (i - l);}{now -= sum[n] - sum[r];now += a[i] * (n - r);now += sum[r] - sum[i + 1];now -= a[i] * (r - (i + 1));}chmin(ans, now);}print(ans);}