贪心算法
贪心算法(又称贪婪算法,Greedy):在对问题求解时,总是做出在当前看来是最好的选择。也就是说,不从整体最优上加以考虑,算法得到的是在某种意义上的局部最优解。
附加:c++ numeric库
其中包含:求和accumulate、差分、向量内积、生成递增序列等功能
贪心策略适用的前提:局部最优策略能导致产生全局最优解。
基本思路:
- 建立数学模型来描述问题
- 把求解的问题分成若干个子问题
- 对每个子问题求解,得到子问题的局部最优解
- 把子问题的解局部最优解合成原来问题的一个解
如何将目标函数最大化?如何证明?有了idea,写代码!
算法伪代码:
从问题的某一初始解出发:
while (朝给定总目标前进一步)
{
利用可行的决策,求出可行解的一个解元素。
}
由所有解元素组合成问题的一个可行解;
例题1:玩筹码
例题2:两地调度
解题思路:考虑将所有人都派往B地,再将其从B地派往A地,则
白话为:对于这2N个人,按照去A地和去B地的费用之差从小到大排序选出前N个去A地,后N个去B地
附加:algorithm中的sort函数
bool myfunction (int i,int j) { return (i<j); }
struct myclass {
bool operator() (int i,int j) { return (i<j);}
} myobject;
// using default comparison (operator <):
std::sort (myvector.begin(), myvector.begin()+4); //(12 32 45 71)26 80 53 33
// using function as comp
std::sort (myvector.begin()+4, myvector.end(), myfunction); // 12 32 45 71(26 33 53 80)
// using object as comp
std::sort (myvector.begin(), myvector.end(), myobject); //(12 26 32 33 45 53 71 80
据此,该题的写法多样,如下:
int twoCitySchedCost(vector<vector<int>>& costs) {
sort(costs.begin(), costs.end(), [](vector<int> a, vector<int>b){return a[0] - a[1] < b[0] - b[1];});
int n = costs.size() / 2, cost = 0;
for(int i = 0; i < n; i++){
cost += (costs[i][0] + costs[i+n][1]);//排序后前N个人去A,后N个人去B
}
return cost;
}
// 或者定义结构体函数如下:
struct myclass {
bool operator() (vector<int> i,vector<int> j) { return (i[0]-i[1]<j[0]-j[1]);}
} myobject;
sort(costs.begin(),costs.end(),myobject);
扩展:algorithm学习
其中有partition函数,可以用于按条件进行划分
将一个数组按奇数和偶数划分如下:
bool IsOdd (int i) { return (i%2)==1; }
std::vector<int>::iterator bound;
//IsOdd为真将被排在前面,同时返回划分的界限迭代器pivot== bound
bound = std::partition (myvector.begin(), myvector.end(), IsOdd);
例题3:最后一块石头
这题没啥可说的,多复习复习algorithm中的用法。尤其是要了解max_element这些用法
int lastStoneWeight2(vector<int>& stones) {
// 多于两块时,继续
while(stones.size()>=2){
sort(stones.begin(),stones.end());
int newStone = stones.back()-stones[stones.size()-2];
stones.pop_back();stones.pop_back();
if(newStone!=0) stones.push_back(newStone);
}
if(stones.size()==1) return stones[0];
return 0;
}
int lastStoneWeight(vector<int>& stones) {
// 多于两块时,继续
while(stones.size()>=2){
// 找到最大的两块石头的位置
auto maxIter = max_element(stones.begin(),stones.end());
int newStone = *maxIter;
// 去除最大的
stones.erase(maxIter);
// 找次大的
auto secIter = max_element(stones.begin(),stones.end());
// 记录最大和次大之差
newStone -=*secIter;
stones.erase(secIter);
if(newStone!=0) stones.push_back(newStone);
}
if(stones.size()==1) return stones[0];
return 0;
}
分割线,以下为中级题
例题4: