Problem:
从数列A[0], A[1], A[2], ..., A[N-1]中选若干个数,要求对于每个i(0<=i<N-1),A[i]和A[i+1]至少选一个数,求能选出的最小和.
1 <= N <= 100000, 1 <= A[i] <= 1000
请为下面的Solution类实现解决上述问题的函数minSum,函数参数A是给出的数列,返回值为所求的最小和.
class Solution {
public:
int minSum(vector<int>& A) {
}
};
例1:A = {2, 5, 2},答案为4.
例2:A = {2, 5, 4},答案为5.
注意:你只需要提交Solution类的代码,你在本地可以编写main函数测试程序,但不需要提交main函数的代码. 注意不要修改类和函数的名称.
Solution:
#include<algorithm> #include<stdio.h> #include<iostream> #include<cstring> #include<stdlib.h> #include<vector> using namespace std; class Solution { public: int minSum(vector<int>& A) { vector<int> f[2];//f[0]为未选邻居节点,f[1]为选择邻居节点。 int asize = A.size(); f[0].resize(asize); f[0][0] = 0; f[1].resize(asize); f[1][0] = A[0]; for (int i = 1; i < asize; i++) { f[0][i] = f[1][i - 1]; f[1][i] = A[i] + min(f[1][i - 1], f[0][i - 1]); } return min(f[0][asize - 1], f[1][asize - 1]); } };