// yukicoder: No.738 平らな農地 // 2019.8.4 bal4u #include #include #include typedef long long ll; #if 1 #define gc() getchar_unlocked() #else #define gc() getchar() #endif int in() { // 非負整数の入力 int n = 0, c = gc(); do n = 10 * n + (c & 0xf); while ((c = gc()) >= '0'); return n; } #define MAX 100005 typedef struct { int v, id; } QUE; QUE qmin[MAX]; int szqmin; QUE qmax[MAX]; int szqmax; #define PARENT(i) ((i)>>1) #define LEFT(i) ((i)<<1) #define RIGHT(i) (((i)<<1)+1) void min_heapify(int i) { int l, r, min; l = LEFT(i), r = RIGHT(i); if (l < szqmin && (qmin[l].v < qmin[i].v || (qmin[l].v == qmin[i].v && qmin[l].id < qmin[i].id))) min = l; else min = i; if (r < szqmin && (qmin[r].v < qmin[min].v || (qmin[r].v == qmin[min].v && qmin[r].id < qmin[min].id))) min = r; if (min != i) { QUE qt = qmin[i]; qmin[i] = qmin[min]; qmin[min] = qt; min_heapify(min); } } int deqmin() { qmin[0] = qmin[--szqmin]; min_heapify(0); return 1; } void enqmin(int id, int v) { int i, min; i = szqmin++; qmin[i].id = id, qmin[i].v = v; while (i > 0 && (qmin[min = PARENT(i)].v > qmin[i].v || (qmin[min].v == qmin[i].v && qmin[min].id > qmin[i].id))) { QUE qt = qmin[i]; qmin[i] = qmin[min]; qmin[min] = qt; i = min; } } void max_heapify(int i) { int l, r, max; l = LEFT(i), r = RIGHT(i); if (l < szqmax && (qmax[l].v > qmax[i].v || (qmax[l].v == qmax[i].v && qmax[l].id < qmax[i].id))) max = l; else max = i; if (r < szqmax && (qmax[r].v > qmax[max].v || (qmax[r].v == qmax[max].v && qmax[r].id < qmax[max].id))) max = r; if (max != i) { QUE qt = qmax[i]; qmax[i] = qmax[max], qmax[max] = qt; max_heapify(max); } } void deqmax() { qmax[0] = qmax[--szqmax]; max_heapify(0); } void enqmax(int id, int v) { int i, max; i = szqmax++; qmax[i].id = id, qmax[i].v = v; while (i > 0 && (qmax[max = PARENT(i)].v < qmax[i].v || (qmax[max].v == qmax[i].v && qmax[max].id > qmax[i].id))) { QUE qt = qmax[i]; qmax[i] = qmax[max], qmax[max] = qt; i = max; } } #define ABS(x) ((x)>=0?(x):-(x)) int N; int a[MAX]; QUE b[MAX]; int cmp(const void *a, const void *b) { int t = ((QUE *)a)->v - ((QUE *)b)->v; if (t) return t; return ((QUE *)a)->id - ((QUE *)b)->id; } int main() { int i, K, k, m, pm, ngmin, ngmax, diff; ll smin, smax, ans, t; N = in(), K = in(), k = K >> 1; if (K == 1) { puts("0"); return 0; } #if 1 if (K == 2) { m = in(), ans = 0x7fffffff; while (--N) { pm = m, m = in(); if ((t = ABS(pm-m)) < ans) ans = t; } goto done; } #endif for (i = 0; i < N; i++) a[i] = in(); for (i = 0; i < K; i++) b[i].v = a[i], b[i].id = i; qsort(b, K, sizeof(QUE), cmp); m = b[k].v, smin = smax = 0, ngmin = ngmax = 0; for (i = 0; i < k; i++) enqmax(b[i].id, b[i].v), smax += m-b[i].v; for ( ; i < K; i++) enqmin(b[i].id, b[i].v), smin += b[i].v-m; ans = smin + smax; if (ans == 0) goto done; // printf("K=%d, k=%d, m=%d, szqmin=%d, szqmax=%d, smin=%lld, smax=%lld, ans=%lld\n", // K, k, m, szqmin, szqmax, smin, smax, ans); for (i = K; i < N; i++) { int x = a[i-K]; if (x > qmin[0].v) ngmin++, smin -= x-m; else if (x < qmax[0].v) ngmax++, smax -= m-x; else if (i-K == qmin[0].id) deqmin(), smin -= x-m; else if (i-K == qmax[0].id) deqmax(), smax -= m-x; else { printf("Panic median=%d, a[%]=%d\n", m, i-K, a[i-K]); return 0; } //printf("szqmin=%d, ngmin=%d, smin=%lld, sqzmax=%d, ngmax=%d, smax=%lld\n", // szqmin, ngmin, smin, szqmax, ngmax, smax); if (a[i] >= m) enqmin(i, a[i]), smin += a[i]-m; else enqmax(i, a[i]), smax += m-a[i]; while (1) { QUE qt; diff = (szqmin-ngmin) - (szqmax-ngmax); //printf("a[%d]=%d, diff=%d\n", i, a[i], diff); if (K & 1) { if (diff == 1) break; } else if (diff == 0) break; if (diff > 0) { qt = qmin[0], deqmin(); if (qt.id <= i-K) ngmin--; else enqmax(qt.id, qt.v), smin -= qt.v-m, smax += m-qt.v; } else { qt = qmax[0], deqmax(); if (qt.id <= i-K) ngmax--; else enqmin(qt.id, qt.v), smin += qt.v-m, smax -= m-qt.v; } } pm = m; if (diff >= 0) m = qmin[0].v; else m = qmax[0].v; if (m != pm) smin += (pm-m) * (szqmin-ngmin), smax += (m-pm)*(szqmax-ngmax); t = smin + smax; if (t < ans) { ans = t; if (ans == 0) goto done; } //printf("pm=%d, m=%d, smin=%lld, smax=%lld, t=%lld, ans=%lld\n\n", pm,m,smin,smax,t,ans); } done: printf("%lld\n", ans); return 0; }