結果
| 問題 | 
                            No.198 キャンディー・ボックス2
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2015-10-05 02:22:14 | 
| 言語 | C++11(廃止可能性あり)  (gcc 13.3.0)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 2 ms / 1,000 ms | 
| コード長 | 1,664 bytes | 
| コンパイル時間 | 1,038 ms | 
| コンパイル使用メモリ | 85,372 KB | 
| 実行使用メモリ | 6,824 KB | 
| 最終ジャッジ日時 | 2024-11-16 09:39:33 | 
| 合計ジャッジ時間 | 2,019 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge3 / judge4 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 26 | 
ソースコード
#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
#include <numeric>
#include <functional>
#include <cmath>
#include <queue>
#include <stack>
#include <set>
#include <map>
#define repd(i,a,b) for (int i=(a);i<(b);i++)
#define rep(i,n) repd(i,0,n)
typedef long long ll;
using namespace std;
int inputValue(){
    int a;
    cin >> a;
    return a;
};
template <typename T>
void output(T a, int precision) {
    if(precision > 0){
        cout << setprecision(precision)  << a << "\n";
    }
    else{
        cout << a << "\n";
    }
}
int main() {
    
    // source code
    int B = inputValue(); // 手持ち
    ll N; // 箱の数
    ll num = B; // 合計の数
    cin >> N;
    vector<int> C(N);
    rep(i, N){
        C[i] = inputValue(); // 各箱の中の数
        num += C[i];
    }
    
    ll low = 0;
    ll high = num / N;
    
    ll lcnt = 0;
    ll hcnt = 0;
    
    while (low + 2 < high) {
        ll lmid = (2 * low + high) / 3;
        ll hmid = (low + 2 * high) / 3;
        
        lcnt = 0;
        hcnt = 0;
        
        rep(i, N){
            lcnt += abs(C[i] - lmid);
            hcnt += abs(C[i] - hmid);
        }
        
        if (lcnt > hcnt) {
            low = lmid;
        }
        else{
            high = hmid;
        }
    }
    ll lmid = low + 1;
    ll hmid = low + 2;
    
    lcnt = 0;
    ll lmcnt = 0;
    ll hmcnt = 0;
    hcnt = 0;
    rep(i, N){
        lcnt += abs(C[i] - low);
        lmcnt += abs(C[i] - lmid);
        hmcnt += abs(C[i] - hmid);
        hcnt += abs(C[i] - high);
    }
    
    output(min(min(lcnt, lmcnt), min(hmcnt, hcnt)), 0);
    
    return 0;
}