有 n 道可选的题可以做,每一题都有特定的忧郁值,你会从中选择 k 题。如果这 k 题的忧郁值总和大于等于 m,那么你就会感到忧郁,反之,你就感受不到忧郁。
那么,请问,有没有可能你做完 k 题之后感受不到忧郁?
如果可能,返回 yes。
如果不可能,即你一定会感到忧郁,那么返回 no。
样例
样例 1:
输入:m = 7, k = 3, arr = [5,3,4,2,1]
输出:"yes"
样例 2:
输入:m = 3, k = 3, arr = [5,1,1,2,3,1]
输出:"no"
注意事项
1 ≤n≤ 10000
arr[i] ≤ 10000
k ≤ n
输入测试数据 (每行一个参数)
如何理解测试数据?
思路:直接排序,排完后前面k个值相加,若不超过m即为"yes",否则为"no"
class Solution {
public:
/**
* @param m: the limit
* @param k: the sum of choose
* @param arr: the array
* @return: yes or no
*/
string depress(int m, int k, vector<int> &arr) {
// Write your code here.
sort(arr.begin(),arr.end());
int sum=0;
for (int i = 0; i < k; i++) {
/* code */
sum+=arr[i];
if(sum>=m){return "no" ;break;}
}
return "yes";
}
};