TopCoder SRM604 DIV1 250

原题链接

题目大意

       外星人想吃汉堡。每一个汉堡有两种属性type_i与taste_i。他获得的快乐值为Y*A，其中Y为他吃的汉堡中有多少种不同类型汉堡，A为他吃到的汉堡taste_i之和。

题解

首先

       我们可以发现，对于任意的汉堡i，如果有taste_i>=0成立，吃掉它对我们的答案是有利的(至少是无害)的。我们得到现在的ans=Y₁*A₁。

其次

       剩下的taste_i<0的汉堡并不是完全不利的，对于第i个汉堡，如果他吃过type_i类汉堡，明显没有必要再吃这个汉堡；如果他没有吃过type_i类汉堡，我们可以得到(Y₁+1)*(A₁+taste_i)的快乐值；如果我们准备吃一个taste_i<0并且type_i没有出现的汉堡，我选择吃掉taste_i最小的汉堡，答案一定会更优。所以我们总是试着吃去type_i没有出现的汉堡中taste_i最小的汉堡。

最后

       我们对taste_i<0的汉堡taste_i的大小按进行从小到大排序。枚举这些汉堡吃几个是最优的来更新ans的值。

代码

#include <vector>
#include <algorithm>
#include <iostream>
#include <cstdio>
using namespace std;
bool f[300];
vector<pair<int,int> >a;
class AlienAndHamburgers {
public:
    int getNumber(vector<int> type, vector<int> taste) {
        int n = type.size();
        int num = 0, sum = 0;
        for (int i = 0; i < n; ++i) {
            if (taste[i] >= 0) {
                sum += taste[i];
                if (f[type[i]])continue;
                num++;
                f[type[i]] = 1;
            } else {
                a.push_back(make_pair(-taste[i], type[i]));
            }
        }
        sort(a.begin(), a.end());
        int ans = sum * num;
        for (int i = 0; i < a.size(); ++i) {
            if (f[a[i].second])continue;
            f[a[i].second] = 1;
            num++;
            sum -= a[i].first;
            ans = max(ans, num * sum);
        }
        return ans;
    }
};