SRM608 Div1 Easy MysticAndCandies

概要

n個の箱にキャンディーがlow[i]個以上high[i]以下入っていることと,合計のキャンディーの個数Xが与えられる.確実にC個キャンディーを得るには最低何個箱を選ぶ必要があるか.

解法

確実にC個取ることができるのは以下の2パターン.

  • 合計がCを超えるまでlowが高い方から取ってくる.
  • 合計がX-Cを超えるまでhighが低い方から選んでいき,それ以外を取る.

どちらでも確定しないなら仕方なくn個を取る.

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(ll i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(ll i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
class MysticAndCandies{
public:
    int minBoxes(int C,int X,vi low,vi high){
        int n=low.size();
        sort(all(low),greater<int>());
        sort(all(high));
        ll ans=n;
        ll sum=0;
        REP(i,n){
            sum+=low[i];
            if(sum>=X){
                chmin(ans,i+1);
                break;
            }
        }
        sum=0;
        REP(i,n){
            sum+=high[i];
            if(C-sum>=X){
                chmin(ans,n-i-1);
            }
        }
        return ans;
    }
};