diff options
author | crupest <crupest@outlook.com> | 2020-09-30 15:16:06 +0800 |
---|---|---|
committer | crupest <crupest@outlook.com> | 2020-09-30 15:16:06 +0800 |
commit | 82ba233d73f0b5feb9bd046e1fc20a2f10dc34a9 (patch) | |
tree | 69124dd0dab7db876a6a544b68468faead9e1d6c /works | |
parent | 3731058c7c7cb70506cffc6c97cf166e2fc33a30 (diff) | |
download | crupest-82ba233d73f0b5feb9bd046e1fc20a2f10dc34a9.tar.gz crupest-82ba233d73f0b5feb9bd046e1fc20a2f10dc34a9.tar.bz2 crupest-82ba233d73f0b5feb9bd046e1fc20a2f10dc34a9.zip |
import(solutions): Add problem 213 .
Diffstat (limited to 'works')
-rw-r--r-- | works/solutions/cpp/213.cpp | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/works/solutions/cpp/213.cpp b/works/solutions/cpp/213.cpp new file mode 100644 index 0000000..1414222 --- /dev/null +++ b/works/solutions/cpp/213.cpp @@ -0,0 +1,41 @@ +#include <vector> + +using std::vector; + +class Solution +{ +public: + int rob(vector<int> &nums) + { + int count = nums.size(); + if (count == 0) + return 0; + + if (count == 1) + return nums.front(); + + int not_rob_prev = 0; + int rob_prev = nums.front(); + int not_rob_prev_and_not_rob_first = 0; + int rob_prev_and_not_rob_first = 0; + + for (int i = 1; i < count - 1; i++) + { + int not_rob_prev_cache = not_rob_prev; + int not_rob_prev_and_not_rob_first_cache = not_rob_prev_and_not_rob_first; + not_rob_prev = std::max(not_rob_prev_cache, rob_prev); + rob_prev = std::max(not_rob_prev_cache + nums[i], not_rob_prev); + not_rob_prev_and_not_rob_first = std::max(not_rob_prev_and_not_rob_first_cache, rob_prev_and_not_rob_first); + rob_prev_and_not_rob_first = std::max(not_rob_prev_and_not_rob_first_cache + nums[i], not_rob_prev_and_not_rob_first); + } + + // last houst + { + int not_rob_prev_and_not_rob_first_cache = not_rob_prev_and_not_rob_first; + not_rob_prev_and_not_rob_first = std::max(not_rob_prev_and_not_rob_first_cache, rob_prev_and_not_rob_first); + rob_prev_and_not_rob_first = std::max(not_rob_prev_and_not_rob_first_cache + nums[count - 1], not_rob_prev_and_not_rob_first); + } + + return std::max(rob_prev, rob_prev_and_not_rob_first); + } +}; |