CodeForces 521D nice贪心

//521D

//有三种升级技能的策略：1、=； 2、+=；3、*=。最优的升级顺序显然是先使用策略1,再策略2,再策略3

//赋值变加（对于每一种技能只考虑增益最大的赋值操作），加变乘（对于每一种技能优先考虑增益最大的加法），在不超过总升级方案的前提下，排序后选出 m 种所乘系数最大的升级方案。因为再原方案都是乘法方案的情况下，先乘哪个数并不会影响最终结果，所以只需要再将 m 种方案处理成合法的顺序即可（根据策略种类排序；原方案是加法方案同理）

#include "iostream"
#include "cstdio"
#include "cstring"
#include "vector"
#include "utility"
#include "algorithm"
using namespace std;
vector<int> a;
int t[100005][3];
vector<pair<int, int> > t1;
vector<vector<pair<int, int> > > t2;
vector<pair<double, int> > t3;
vector<pair<int, int> > sol;
int k, n, m;

int main()
{
    int i, j;
    scanf("%d%d%d", &k, &n, &m);
    a.resize(k + 1);
    t1.resize(k + 1);
    t2.resize(k + 1);
    for(i = 1; i <= k; ++i) {
        scanf("%d", &a[i]);
    }
    for(i = 1; i <= n; ++i) {
        scanf("%d%d%d", &t[i][0], &t[i][1], &t[i][2]);
        switch(t[i][0]) {
        case 1:
            if(t1[t[i][1]].first < t[i][2]) {
                t1[t[i][1]].first = t[i][2];
                t1[t[i][1]].second = i;
            }
            break;
        case 2:
            t2[t[i][1]].push_back(make_pair(t[i][2], i));
            break;
        case 3:
            t3.push_back(make_pair(t[i][2], i));
        }
    }
    /*for(i = 1; i <= k; ++i) {
        printf("t1 %d %d
", t1[i].first, t1[i].second);
        for(j = 0; j < t2[i].size(); ++j)
            printf("t2 %d %d
", t2[i][j].first, t2[i][j].second);
    }
    for(i = 0; i < t3.size(); ++i)
        printf("t3 %lf %d
", t3[i].first, t3[i].second);*/

    for(i = 1; i <= k; ++i) {
        if(t1[i].first > a[i]) {
            t2[i].push_back(make_pair(t1[i].first - a[i], t1[i].second));
            //printf("zhuan %d %d
", make_pair(t1[i].first - a[i], t1[i].second).first, make_pair(t1[i].first - a[i], t1[i].second).second);
        }
        sort(t2[i].begin(), t2[i].end());
        double sum = a[i];
        for(j = t2[i].size() - 1; j >= 0; --j) {
            t3.push_back(make_pair((sum + t2[i][j].first) / sum, t2[i][j].second));
            sum += t2[i][j].first;
        }
    }
    sort(t3.begin(), t3.end());
    /*for(i = t3.size() - 1; i >= 0; --i)
        printf("t3 %f %d
", t3[i].first, t3[i].second);*/
    
    m = min(m, (int)t3.size());
    for(i = t3.size() - 1; i >= (int)t3.size() - m; --i) {
        sol.push_back(make_pair(t[t3[i].second][0], t3[i].second));
    }
    sort(sol.begin(), sol.end());
    printf("%d
", m);
    if(m) {
        printf("%d", sol[0].second);
        for(i = 1; i <= m - 1; ++i) {
            printf(" %d", sol[i].second);
        }
        printf("
");
    }
}