Codeforces Round #835 (Div. 4) F. Quests

Codeforces Round #835 (Div. 4) F. Quests

Let’s fix $k$ and find the maximum number of coins we can get. Here we can do a greedy solution: at every step, we should always take the most rewarding quest. (Intuitively, it makes sense, since doing more rewarding quests earlier allows us to do them again later.) If no quests are available, we do nothing.

To implement this, sort the quests in decreasing order, and $0$-index them. On day $i$ we should do quest $i$ $mod$ $k$, provided that this value is less than $n$. This is because after every $k$ days, we cycle back to the first quest. Thus we solved the problem for a fixed $k$ in $O(d)$ with $O(nlogn)$ precomputation to sort the array.

Now to solve the problem, we can binary search on the answer, since if some $k$ works, then all smaller $k$ work. The minimum value of $k$ is $0$, and the maximum value is n (for larger $k$, we won’t be able to do the same quest multiple times anyways, so it’s useless to consider them).

If we find that $k$ always goes towards the smaller end of our binary search and $k=0$ still fails, we output $Impossible$. If we find that $k$ always goes towards the larger end of our binary search and $k=n$ still fails, we output $Infinity$. Otherwise, just output $k$.

The overall time complexity is $O(nlogn+dlogn)$.

Remark. It is not hard to improve the solution to $O(nlogn)$. Originally, I proposed the problem this way, but we ended up removing this part of the problem because the implementation of this solution was tricky enough.

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

typedef long long LL;
const int N = 200010;
LL a[N];

int main()
{
    int T=1;cin>>T;
    while(T--)
    {
        LL n,c,d;cin>>n>>c>>d;
        for(int i=1;i<=n;i++) cin>>a[i];

        sort(a+1,a+n+1);
        reverse(a+1,a+n+1);
        for(int i=2;i<=n;i++) a[i]+=a[i-1];

        if(a[1]*d<c)
        {
            cout<<"Impossible"<<endl;
            continue ;
        }

        if(a[min(n,d)]>=c)
        {
            cout<<"Infinity"<<endl;
            continue ;
        }

        auto check=[&](int mid){
            LL sum=0;
            LL dd=d/mid;
            LL yy=d%mid;
            if(mid>=n) sum+=dd*a[n];
            else sum+=dd*a[mid];

            if(yy>=n) sum+=a[n];
            else sum+=a[yy];

            if(sum<c) return false;
            else return true;
        };

        int l=1,r=200000;
        while(l<r)
        {
            int mid=(l+r+1)>>1;
            if(check(mid)) l=mid;
            else r=mid-1;
        }
        cout<<l-1<<endl;
    }
    return 0;
}