-
-
Notifications
You must be signed in to change notification settings - Fork 298
/
Copy path312.cpp
83 lines (81 loc) · 2.6 KB
/
312.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
__________________________________________________________________________________________________
sample 4 ms submission
class Solution {
public:
int maxCoins(vector<int>& nums) {
nums.insert(nums.begin(), 1);
nums.push_back(1);
int base = applyHeuristic(nums);
return base + applyDP(nums);
}
int applyHeuristic(vector<int>& nums) {
const int n = nums.size() - 2;
const int maximum = *std::max_element(nums.begin(), nums.end());
int result = 0;
int i = 1;
int j = 2;
while (j <= n + 1) {
int a = nums[i - 1];
int b = nums[i];
int c = nums[j];
if (b <= a && b <= c && (a * c * (b + maximum)) >= (b * maximum * (a + c))) {
result += a * b * c;
if (i > 1) {
--i;
} else {
nums[i] = nums[j];
++j;
}
} else {
++i;
nums[i] = nums[j];
++j;
}
}
nums.resize(i + 1);
return result;
}
int applyDP(const vector<int>& nums) {
const int n = nums.size();
vector<int> dr(n, 0);
vector<vector<int>> dp(n, dr);
for (int k = 2; k < n; ++k){
for (int left = 0; left < n - k; ++left) {
int right = left + k;
for (int i = left + 1; i < right; ++i) {
dp[left][right] = max(
dp[left][right],
nums[left] * nums[i] * nums[right] + dp[left][i] + dp[i][right]);
}
}
}
return dp[0][n - 1];
}
};
__________________________________________________________________________________________________
sample 8536 kb submission
class Solution
{
public:
int maxCoins(vector<int>& nums)
{
int arr[nums.size()+2];
for(int i=1;i<nums.size()+1;++i)arr[i] = nums[i-1];
arr[0] = arr[nums.size()+1] = 1;
int dp[nums.size()+2][nums.size()+2]={};
int n = nums.size()+2;
for(int k=2;k<n;++k)
{
for(int left = 0;left<n-k;++left)
{
int right = left + k;
for(int i=left+1;i< right; ++i)
{
dp[left][right] = max(dp[left][right],arr[left]*arr[i]*arr[right] + dp[left][i] + dp[i][right]);
}
}
}
return dp[0][n-1];
}
};
__________________________________________________________________________________________________