SRM605 Div1 Easy AlienAndHamburgers

概要

ハンバーガーがn個あり,i個目は種類a_i,美味しさb_iである.
これらのうち幾つかを食べ,食べた種類Yと美味しさの和AとしてY\times Aを最大化せよ.

解法

ある種類を食べたいとして,その種類に美味しさが正のものがあればそれらをすべて食べ,なければ最大のものを1つ食べるのが最善.
全種類に対してそれを食べたときの美味しさの総和を計算し,ソートした後に値が大きい種類から食べていき答えを更新していく.

ソースコード

#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<int,int>;
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 AlienAndHamburgers{
public:
    int getNumber(vi a,vi b){
        int n=a.size();
        REP(i,n){
            a[i]--;
        }
        vp vec(100);
        REP(i,n){
        	if(!vec[a[i]].se){
        		vec[a[i]].fi=b[i];
        	}else{
        		if(vec[a[i]].fi<0){
        			chmax(vec[a[i]].fi,b[i]);
        		}else if(b[i]<0){
        			continue;
        		}else{
        			vec[a[i]].fi+=b[i];
        		}
        	}
        	vec[a[i]].se++;
        }
        sort(all(vec),greater<P>());
        int cnt=0;
        int ans=0;
        int sum=0;
        REP(i,100){
            if(!vec[i].se){
                continue;
            }
            cnt++;
            sum+=vec[i].fi;
            cerr<<cnt*sum<<endl;
            chmax(ans,cnt*sum);
        }
        return ans;
    }
};