結果

問題 No.738 平らな農地
ユーザー face4
提出日時 2020-01-30 14:37:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 5,994 bytes
コンパイル時間 1,080 ms
コンパイル使用メモリ 87,992 KB
実行使用メモリ 8,704 KB
最終ジャッジ日時 2024-09-16 03:34:01
合計ジャッジ時間 9,296 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 87
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include<iostream>
#include<random>
#include<vector>
#include<climits>
using namespace std;
// rand
// 0-indexed
template<class T, class F>
struct Treap{
struct Node{
T val, sum;
int pri;
int cnt;
Node* ch[2];
Node(T v, int p) : val(v), sum(v), pri(p){
cnt = 1;
ch[0] = ch[1] = NULL;
}
};
Node* root;
vector<T> v;
F f;
T def;
Treap(T def, F f) : f(f), def(def) {
root = NULL;
}
int count(Node* t) {return t==NULL ? 0 : t->cnt;}
T sum(Node* t) {return t==NULL ? def : t->sum;}
Node* update(Node* t){
t->cnt = count(t->ch[0]) + count(t->ch[1]) + 1;
t->sum = f(sum(t->ch[0]), f(t->val, sum(t->ch[1])));
return t;
}
Node* rotate(Node* t, int b){
Node* s = t->ch[b];
t->ch[b] = s->ch[1-b];
s->ch[1-b] = t;
update(t);
update(s);
return s;
}
Node* insert(Node* t, int k, T val, int pri){
if(t == NULL) return new Node(val, pri);
int leftCnt = count(t->ch[0]);
int b = k > leftCnt;
t->ch[b] = insert(t->ch[b], k-(b ? leftCnt+1 : 0), val, pri);
update(t);
if(t->ch[b]->pri > t->pri) t = rotate(t, b);
return t;
}
Node* insert_value(Node* t, T val, int pri){
if(t == NULL) return new Node(val, pri);
int b = val > t->val;
t->ch[b] = insert_value(t->ch[b], val, pri);
update(t);
if(t->ch[b]->pri > t->pri) t = rotate(t, b);
return t;
}
Node* erase(Node* t, int k, bool me=false){
if(!me && count(t) <= k) return t; // out-of-range
me |= count(t->ch[0])==k;
if(me){
if(t->ch[0]==NULL && t->ch[1]==NULL){
delete t;
return NULL;
}
int b = (t->ch[0]==NULL ? -1 : t->ch[0]->pri) < (t->ch[1]==NULL ? -1 : t->ch[1]->pri);
t = rotate(t, b);
t->ch[1-b] = erase(t->ch[1-b], k, me);
}else{
if(count(t->ch[0]) > k){
t->ch[0] = erase(t->ch[0], k, me);
}else{ // definitely count(t->ch[0]) < k
t->ch[1] = erase(t->ch[1], k-count(t->ch[0])-1, me);
}
}
return update(t);
}
Node* erase_value(Node* t, T val, bool me=false){
if(t == NULL) return NULL; // there's no node whose value is val
me |= t->val==val;
if(me){
if(t->ch[0]==NULL && t->ch[1]==NULL){
delete t;
return NULL;
}
int b = (t->ch[0]==NULL ? -1 : t->ch[0]->pri) < (t->ch[1]==NULL ? -1 : t->ch[1]->pri);
t = rotate(t, b);
t->ch[1-b] = erase_value(t->ch[1-b], val, me);
}else{
if(val < t->val){
t->ch[0] = erase_value(t->ch[0], val, me);
}else{ // definitely count(t->ch[0]) < k
t->ch[1] = erase_value(t->ch[1], val, me);
}
}
return update(t);
}
Node* merge(Node* l, Node* r){
Node* tmp = new Node(def, INT_MAX); // rand()
tmp->ch[0] = l;
tmp->ch[1] = r;
update(tmp);
bool me = true;
return root = erase(tmp, -1, me);
}
// [0, k), [k, n)
pair<Node*, Node*> split(int k){
root = insert(root, k, def, INT_MAX);
return make_pair(root->ch[0], root->ch[1]);
}
T get(Node* t, int k){
int leftCnt = count(t->ch[0]);
if(leftCnt > k) return get(t->ch[0], k);
if(leftCnt == k) return t->val;
return get(t->ch[1], k-1-leftCnt);
}
void set(Node* t, int k, T newVal){
if(k < count(t->ch[0])){
set(t->ch[0], k, newVal);
}else if(k == count(t->ch[0])){
t->val = newVal; // or f(t->val, newVal) //
}else if(k > count(t->ch[0])){
set(t->ch[1], k-count(t->ch[0])-1, newVal);
}
update(t);
}
// a, bt-oriented
T query(Node* t, int a, int b){
if(t == NULL) return def;
if(a == 0 && count(t) <= b+1) return t->sum;
T ret = def;
int leftCnt = count(t->ch[0]);
if(a < leftCnt) ret = f(ret, query(t->ch[0], a, b));
if(a <= leftCnt && leftCnt <= b) ret = f(ret, t->val);
if(b > leftCnt) ret = f(ret, query(t->ch[1], max(0, a-leftCnt-1), b-leftCnt-1));
return ret;
}
Node* insert(int k, T val){
return root = insert(root, k, val, rand());
}
Node* erase(int k){
return root = erase(root, k);
}
T get(int k){
return get(root, k);
}
void set(int k, T val){
set(root, k, val);
}
T query(int a, int b){
return query(root, a, b);
}
Node* insert_value(T val){
return root = insert_value(root, val, rand());
}
Node* erase_value(T val){
return root = erase_value(root, val);
}
void flatten(Node* t){
if(t==NULL) return;
if(t->ch[0] != NULL) flatten(t->ch[0]);
v.push_back(t->val);
if(t->ch[1] != NULL) flatten(t->ch[1]);
}
vector<T> flatten(){
v.clear();
flatten(root);
return v;
}
};
typedef long long ll;
int main(){
auto f = [](ll x, ll y)->ll{
return x+y;
};
Treap<ll, decltype(f)> t(0, f);
int n, k;
scanf("%d %d", &n, &k);
ll ans = 1ll<<60;
vector<ll> v(n);
for(int i = 0; i < n; i++) scanf("%lld", v.begin()+i);
for(int i = 0; i < n; i++){
if(i >= k) t.erase_value(v[i-k]);
t.insert_value(v[i]);
if(i < k-1) continue;
ll med = t.get(k/2);
ll tmp = 0;
if(k/2 > 0) tmp += k/2*med-t.query(0,k/2-1);
if(k/2+1 <= k-1) tmp += t.query(k/2+1,k-1)-(k-1-k/2)*med;
ans = min(ans, tmp);
}
printf("%lld\n", ans);
return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0