F. Bomb

原题链接

题解

贪心的每次挑选当前最大的，但是要挑选k次，因此我们没法去遍历挑选的过程，因此我们考虑最终形态，由于每次挑选最大的元素，因此最后所有数一定不超过某个数，二分由此而来

code

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,k;

ll a[200005],b[200005];

ll check(ll x)
{
    ll res=0;
    for(ll i=1;i<=n;i++)
    {
        if(a[i]>x)
        {
            res+=(a[i]-x)/b[i]+((a[i]-x)%b[i]!=0);
        }
    }
    return res;
}

ll cal(ll x)
{
    //printf("x:%lld\n",x);
    ll ans=0;
    ll tem=k;
    for(int i=1;i<=n;i++)
    {
        if(a[i]>x)
        {
            ll num=min((a[i]-x)/b[i]+((a[i]-x)%b[i]!=0),tem);
            //printf("ai:%lld  bi:%lld  num:%lld\n",a[i],b[i],num);
            ans+=a[i]*num-num*(num-1)/2*b[i];
            a[i]-=num*b[i];
            tem-=num;
        }
    }

    for(ll i=1;i<=n;i++)
    {
        if(!tem) break;
        if(a[i]==x)
        {
            ans+=x;
            tem--;
        }
    }
    return ans;
}

void solve()
{
    cin>>n>>k;

    for(ll i=1;i<=n;i++) cin>>a[i];
    for(ll i=1;i<=n;i++) cin>>b[i];

    ll l=-1,r=1e9;
    while(l+1<r)
    {
        ll mid=(l+r)/2;
        if(check(mid)<=k) r=mid;
        else l=mid;
    }

    cout<<cal(r)<<'\n';
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t=1;
    cin>>t;
    while(t--) solve();
    return 0;
}