arc126c

arc126c

思路

首先，有不等式 $\gcd (a_1,a_2,...,a_n)\le \min (a_1,a_2,...,a_n)$

如果 k 足够大，以至于所有的数都加到 $\max (a_1,a_2,...,a_n)$ 后还有剩余，那么此时，每当 $n\le k$ ，就可以贪心的将所有数加一，同时 gcd 加一

如果 k 不够大，我们可以断定 $\gcd (a_1,...,a_n)<\max (a_1,a_2,...,a_n)\le 3\ast 10^5$ 在无论何种分配下都成立

那么，我们可以尝试枚举 gcd 的值，设为 $x$

不难发现 $kx+1$ 到 $(k+1)x$ 之内的数的最优解是变成 $(k+1)x$

设 $cn{t}_i$ 为值不比 $i$ 大的数的个数， $pr{e}_i$ 为所有小于等于 $i$ 的数距离值 $i$ 的距离的总和

不难发现 $cn{t}_{i+1}=cn{t}_i+nu{m}_{i+1}$ 且 $pr{e}_{i+1}=pr{e}_i+cn{t}_i$ ，其中 $nu{m}_i$ 代表值等于 $i$ 的数的个数

那么从 $kx+1$ 到 $(k+1)x$ 的代价就是 $pr{e}_{(k+1)x}-pr{e}_{kx}-cn{t}_{kx}\ast x$

那么对于数 $x$ ，统计一次的复杂度为 $\Theta (3e10/x)$

总复杂度为 $\Theta (N+3e10\log 3e10)$

代码

#include <bits/stdc++.h>
using namespace std;

#ifdef _DEBUG
#define pr1(x) cout << x << ' ';
#define pr2(x) cout << x << endl;
#else
#define pr1(x)
#define pr2(x)
#endif
typedef long long ll;
typedef unsigned long long ull;
typedef const int& cint;
typedef pair<int, int> pii;

const int mod = 1e9+7;
const int inf_int = 0x7fffffff;
const ll inf_ll = 0x7fffffffffffffff;
const double ept = 1e-9;

ll n, k;
ll a[600300];
ll cnt[600300];
ll pre[600300];

int main() {
    ll mx = 0;
    ll sum = 0;
    cin >> n >> k;
    for(int i=1; i<=n; i++) {
        cin >> a[i];
        ++cnt[a[i]];
        mx = max(mx, a[i]);
        sum += a[i];
    }
    for(int i=1; i<=2*mx; i++) {
        cnt[i] += cnt[i-1];
        pre[i] += pre[i-1] + cnt[i-1];
    }
    // for(int i=1; i<=mx; i++)
    //     cout << pre[i] << ' ';
    //     cout << endl;
    if(1ll*mx*n-k <= sum) {
        k -= 1ll*mx*n-sum;
        cout << mx + k/n << endl;
    }
    else {
        int ans = 1;
        for(int i=2; i<mx; i++) {
            ll rsum = 0;
            for(int j=i; j<2*mx; j+=i) {
                int l = j-i;
                rsum += pre[j] - pre[l] - 1ll*cnt[l]*i;
            }
            // cout << i << ' ' << rsum << endl;
            if(rsum <= k) ans = i;
        }
        cout << ans << endl;
    }
    return 0;
}