算法博客1 - 找出数组中第K大的元素
题目描述:
从数列A[0], A[1], A[2], …, A[N-1]中选若干个数,要求对于每个i(0<=i <=N-2),A[i]和A[i+1]至少选一个数,求能选出的最小和.
其中 1 <= N <= 100000, 1 <= A[i] <= 1000
例1:A = {2, 5, 2},答案为4.
例2:A = {2, 5, 4},答案为5.
解题思路:
假设数组的最后一个元素时必须选择的。则当数组只有一个元素时,选择的是第一个元素,数组只有两个元素时,选择的是第二个元素。数组元素数量大于两个时,每次选择的是最后一个元素加上前面元素的最小值,或者加上前面的前面元素的最小值,取这两个值的最小值即可。
代码:
class Solution {
public:
//求最小和
int minSum(vector<int>& A) {
int size = A.size();
vector<int> S(size, 0);
S[0] = A[0];
if (size > 1) {
S[1] = A[1];
}
for (int i = 2; i < size; i++) {
S[i] = min(S[i - 2] + A[i], S[i - 1] + A[i]);
}
if (size == 1) {
return S[0];
}
else {
return min(S[size - 1], S[size - 2]);
}
}
};