贪心
贪心算法的核心就是每次选择最优解。
最简单的例子:
561. 数组拆分
给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), …, (an, bn) ,使得从 1 到 n 的 min(ai, bi) 总和最大。
要找到总和最大,那么就要使每次找最小值时,数字之差最小,这样加起来的损失就最小。
于是可以先对数组进行排序,数组递增,每次选两个数字对的最小值,也就是选在前面的那个数字,一次for循环即可找到总和的最大值。
class Solution {
public:
int arrayPairSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
int sum=0,tmin = nums[0];
for(int i=0;i<nums.size()-1;i+=2){
sum += nums[i];
}
return sum;
}
};
605. 种花问题
判断连续三个元素是否都为0,即fl[i-1]、fl[i]、fl[i+1]。
注意还有i=0和i=len-1时的情况,在if条件判断里,要先将这两个写在或语句的前一个,不然会导致运行时间错误,因为下标越界了。
class Solution {
public:
bool canPlaceFlowers(vector<int>& fl, int n) {
if(fl.size()==1 && fl[0]==0) return true;
if(n==0) return true;
int count=0;
int len = fl.size();
for(int i=0;i<len;i++){
if((i==0 || fl[i-1]==0) && fl[i]==0 && (i==len-1||fl[i+1]==0)){
count++;
fl[i]=1;
}
cout<<count<<endl;
}
if(n<=count) return true;
return false;
}
};
455. 分发饼干
同样是贪心,对两个数组进行排序即可。
注意:
- 可以用while循环找到能分发的第一块饼干
- 在循环中i是否已经到达数组边界,需要进行判断,会导致count多加一次。
class Solution {
public:
int findContentChildren(vector<int>& g, vector<int>& s) {
sort(g.begin(),g.end());
sort(s.begin(),s.end());
int count=0,k=0,i=0,j=0;
for(i=0,j=0;i<s.size() && j<g.size();i++,j++){
while(i<s.size()&&s[i]<g[j]) i++;
if(i<s.size())
count++;
}
return count;
}
};