結果

問題 No.738 平らな農地
ユーザー bal4u
提出日時 2019-08-05 07:46:58
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 19 ms / 2,000 ms
コード長 4,436 bytes
コンパイル時間 1,389 ms
コンパイル使用メモリ 32,628 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-18 10:40:13
合計ジャッジ時間 3,504 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 87
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:11:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
   11 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:17:24: note: in expansion of macro 'gc'
   17 |         int n = 0, c = gc();
      |                        ^~

ソースコード

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

// yukicoder: No.738 平らな農地
// 2019.8.4 bal4u
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
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 += (ll)(pm-m)*(szqmin-ngmin), smax += (ll)(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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0