結果

問題 No.3051 Make All Divisible
ユーザー shobonvip
提出日時 2025-03-07 22:25:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,258 bytes
コンパイル時間 4,892 ms
コンパイル使用メモリ 252,456 KB
実行使用メモリ 8,608 KB
最終ジャッジ日時 2025-03-07 22:25:19
合計ジャッジ時間 6,378 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24 WA * 7
権限があれば一括ダウンロードができます

ソースコード

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

/**
    author:  shobonvip
    created: 2025.03.07 22:05:11
**/
#include<bits/stdc++.h>
using namespace std;
//* ATCODER
#include<atcoder/all>
using namespace atcoder;
typedef modint998244353 mint;
//*/
/* BOOST MULTIPRECISION
#include<boost/multiprecision/cpp_int.hpp>
using namespace boost::multiprecision;
//*/
typedef long long ll;
#define rep(i, s, n) for (int i = (int)(s); i < (int)(n); i++)
#define rrep(i, s, n) for (int i = (int)(n)-1; i >= (int)(s); i--)
#define all(v) v.begin(), v.end()
template <typename T> bool chmin(T &a, const T &b) {
    if (a <= b) return false;
    a = b;
    return true;
}
template <typename T> bool chmax(T &a, const T &b) {
    if (a >= b) return false;
    a = b;
    return true;
}
template <typename T> T max(vector<T> &a){
    assert(!a.empty());
    T ret = a[0];
    for (int i=0; i<(int)a.size(); i++) chmax(ret, a[i]);
    return ret;
}
template <typename T> T min(vector<T> &a){
    assert(!a.empty());
    T ret = a[0];
    for (int i=0; i<(int)a.size(); i++) chmin(ret, a[i]);
    return ret;
}
template <typename T> T sum(vector<T> &a){
    T ret = 0;
    for (int i=0; i<(int)a.size(); i++) ret += a[i];
    return ret;
}
void solve() {
    ll k, n;
    cin >> n >> k;
    vector<ll> a(n);
    vector<ll> b(n);
    vector<ll> c(n);
    ll s = 0;
    rep(i,0,n) {
        cin >> a[i];
        b[i] = a[i] - a[i] / k * k;
        c[i] = a[i] / k;
        s += b[i];
    }
    if (s % k != 0) {
        cout << -1 << '\n';
        return;
    }
    if (s == 0) {
        cout << 0 << '\n';
        return;
    }
    auto check = [&](ll i) {
        ll p = i + s/k;
        ll freedom = 0;
        ll cost = 0;
        vector<ll> amari(k);
        rep(i,0,n) {
            if (b[i] > p) {
                cost += p;
            }else {
                freedom += min(c[i], (p - b[i]) / k);
                if (a[i] > p) {
                    amari[(p - b[i]) % k] += 1;
                }
                cost += b[i];
            }
        }
        ll left = i;
        {
            ll targ = min(freedom, left);
            freedom -= targ;
            left -= targ;
            cost += targ * k;
        }
        rrep(i,0,k) { 
            ll targ = min(amari[i], left);
            left -= targ;
            amari[i] -= targ;
            cost += targ * i;
        }
        if (p*k<=cost) return true;
        return false;
    };
    for(ll i=0;i<=min(sum(c),2*(n+k)+4);i++) {
        if (check(i)) {
            cout << i + s/k << '\n';
            return;
        }
    }
    cout << -1 << '\n';
    return;
    
}
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int t; cin >> t;
    while(t--) solve();
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0