add: 220101, I am vegetable
This commit is contained in:
parent
a24a1632b2
commit
92baff81df
|
|
@ -0,0 +1,95 @@
|
|||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <cstring>
|
||||
|
||||
/**
|
||||
* Usage of dynamic programming
|
||||
*
|
||||
* Initial attempt:
|
||||
* try to merge 2 arrays which is calculated before.
|
||||
* [3, 1, 7] => 49; [5, 8] => 48
|
||||
* [3, 1, 7, 5, 8] => 501
|
||||
* dp[L][R] <- max{merge(dp[L][L + i], dp[L + i + 1][R])}
|
||||
*
|
||||
* Why it fails:
|
||||
* Since the solution might be in a completely shuffled
|
||||
* order, making the previous results useless.
|
||||
*
|
||||
* How the solution solved it:
|
||||
* The solution iterates from smaller intervals to
|
||||
* larger intervals.
|
||||
* dp[L][R] <- max{dp[L][L + i - 1] + dp[L + i + 1][R] + n[L + i] * n[L - 1] * n[R + 1]}
|
||||
*
|
||||
* Why solution works?
|
||||
* In my previous attempt, There is no information such as
|
||||
* *Which one is eliminated the first [Elimination Order]?*
|
||||
* thus the order to eliminate remains unknown.
|
||||
* The solution iterates *the last number to eliminate*,
|
||||
* avoided to consider the order. (by brute force)
|
||||
*
|
||||
*/
|
||||
|
||||
class Solution {
|
||||
public:
|
||||
static int maxCoins(const std::vector<int>& nums) {
|
||||
const auto n = nums.size();
|
||||
int num[n + 2];
|
||||
std::memcpy(num + 1, &nums[0], n * sizeof(int));
|
||||
num[0] = num[n + 1] = 1;
|
||||
|
||||
int dp[n + 2][n + 2];
|
||||
std::memset(dp, 0, sizeof dp);
|
||||
for (int i = 1; i <= n; ++i)
|
||||
dp[i][i] = num[i - 1] * num[i] * num[i + 1];
|
||||
dp[0][0] = num[1];
|
||||
dp[n + 1][n + 1] = num[n];
|
||||
|
||||
for (int interval = 2; interval <= n; ++interval)
|
||||
for (int L = 1, R = interval; R <= n; ++L, ++R)
|
||||
for (int M = L; M <= R; ++M)
|
||||
dp[L][R] = std::max(dp[L][R], dp[L][M - 1] + dp[M + 1][R] + num[M] * num[L - 1] * num[R + 1]);
|
||||
|
||||
return dp[1][n];
|
||||
}
|
||||
};
|
||||
|
||||
std::map<std::vector<int>, int> m;
|
||||
int cnt = 0;
|
||||
|
||||
int bruteForce(const std::vector<int>& nums) {
|
||||
if (m[nums]) { return m[nums]; }
|
||||
++cnt;
|
||||
if (nums.size() == 1)
|
||||
return nums[0];
|
||||
else if (nums.empty())
|
||||
return 0;
|
||||
int ret = -1;
|
||||
for (auto it = nums.begin(); it != nums.end(); ++it) {
|
||||
std::vector<int> d(nums.begin(), it);
|
||||
d.insert(d.end(), it + 1, nums.end());
|
||||
int t = (*it * (it == nums.begin() ? 1 : *(it - 1)) * ((it + 1 == nums.end()) ? 1 : *(it + 1))) + bruteForce(d);
|
||||
ret = (t > ret) ? t : ret;
|
||||
}
|
||||
return m[nums] = ret;
|
||||
}
|
||||
|
||||
int betterBruteForce(const std::vector<int>& nums, int append) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
int main() {
|
||||
// 3, 1, 5, 87, 9 --> 4101
|
||||
// 3, 1, 5, 87, 9, 51 --> 63102
|
||||
srand(time(nullptr));
|
||||
const int LIM = 15;
|
||||
std::vector<int> s;
|
||||
s.reserve(LIM);
|
||||
for (int i = 0; i < LIM; ++i) {
|
||||
s.push_back(rand() % 100 + 1);
|
||||
}
|
||||
std::printf("%d\n", Solution::maxCoins(s));
|
||||
std::printf("%d\n", bruteForce(s));
|
||||
std::printf("Count = %d\n", cnt);
|
||||
return 0;
|
||||
}
|
||||
|
|
@ -3,4 +3,4 @@ PROJECT(2201)
|
|||
|
||||
SET(CMAKE_CXX_STANDARD 23)
|
||||
|
||||
ADD_EXECUTABLE(2201 220101-CN.cpp)
|
||||
ADD_EXECUTABLE(2201 220101.cpp)
|
||||
Loading…
Reference in New Issue