贪心策略——项目利润

题目描述:
输入:
cost[]:每个项目的花费
profits[]:每个项目的利润(纯利润)
k:最多能做k个项目
w:表示初始资金
输出:最后获得的最大钱数
说明:一次只能做一个项目,且做完一个之后马上就能获得收益,可以支持做下一个项目

/*
贪心策略:当资金有限时如何选择那个项目使得挣的钱最多
算法思想:
(1)将所有项目按照花费值costs加入到小根堆中
(2)从小根堆中取出可以做的项目(当前的花费值小于初始资金w)按照利润值profits加入到大根堆中
(3)开始做项目,更新w的值;直到做满k次,或者新的w值仍不满足做新的项目(资金不够,不足以支撑新的项目)
*/

/*
测试用例:
4(项目总数)
5 10 6 20 (costs)
3 2 4 9 (profits)
7 (初始资金)
2 (项目执行最多次数)
输出:14
算法思想:按照costs最低入小根堆(全部项目);从小根堆中取出costs低于已有资金(初始资金+做项目完成收益)
的项目,按照profits最高入大根堆,并做项目;直到已有资金不足以支持做下一个项目或者项目次数达到k时,说明
此时收益最大
*/

#include
#include
using namespace std;

struct Node{
    int c;//收益
    int p;//花费
    Node(int cost, int profit){
        this -> c = cost;
        this -> p = profit;
    }
};

//小根堆只比较花费,不用比较利润
struct MinCostComparator{
    bool operator()(Node a, Node b){
        return a.c > b.c;
    }
};

//大根堆只比较利润,不用比较花费
struct MaxProfitComparator{
    bool operator()(Node a, Node b){
        return a.p < b.p;
    }
};

int MaxProfits(int w, int k, int *costs, int *profits, int n){
    priority_queuevector, MinCostComparator > MinCostsQue ;//按照costs小的节点入小根堆
    priority_queuevector, MaxProfitComparator > MaxProfitsQue ;//按照profits大的节点入大根堆

    //将所有的项目加到小根堆中去
    for(int i = 0; i < n; i ++){
        Node *node = new Node(costs[i], profits[i]);
        MinCostsQue.push(*node);    
    }

    //最多做k次项目
    for(int s = 0; s < k; s ++){
        while(!MinCostsQue.empty() && MinCostsQue.top().c <= w){
            Node topQue = MinCostsQue.top();
            MaxProfitsQue.push(topQue); //将小根堆中的节点加入到大根堆中
            MinCostsQue.pop();
        }
        if(MaxProfitsQue.empty()){  //做的项目还未达到k次就必须停止(跟数据状况有关)
            return w;
        }
        w += MaxProfitsQue.top().p;
        MaxProfitsQue.pop();
    }
    return w;
}

//for test
int main()
{
    int n;//n代表项目数
    cin>>n;
    int *costs = new int[n];//costs代表每个项目对应的消费
    int *profits = new int[n];//profits代表每个项目对应的收益
    for(int i = 0; i < n; i ++){
        cin>>costs[i];
    }
    for(int j = 0; j < n; j ++){
        cin>>profits[j];
    }
    int w, k;//w为初始本金,k为项目最多次数
    cin>>w>>k;
    cout<return 0;
}

你可能感兴趣的:(C++,数据结构,算法)