另外两个经典题的答案
打家劫舍 [2, 7, 9, 3, 1],最大子段和 [-2, 1, -3, 4, -1, 2, 1, -5, 4]。
运行下面这段程序:
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
using namespace std;
vector<int> rob(const vector<int>& a) {
vector<int> dp(a.size() + 1, 0);
if (!a.empty()) dp[1] = a[0];
for (size_t i = 2; i <= a.size(); i++) {
dp[i] = max(dp[i - 1], dp[i - 2] + a[i - 1]);
}
return dp;
}
int max_sub(const vector<int>& a) {
int best = a[0];
int cur = a[0];
for (size_t i = 1; i < a.size(); i++) {
cur = max(a[i], cur + a[i]);
best = max(best, cur);
}
return best;
}
int main() {
cout << rob({2, 7, 9, 3, 1}).back() << "/" << max_sub({-2, 1, -3, 4, -1, 2, 1, -5, 4}) << endl;
}
(本题用 g++ -std=c++17 -O0 编译。)
全部评论