-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
111 lines (100 loc) · 3.22 KB
/
main.cpp
File metadata and controls
111 lines (100 loc) · 3.22 KB
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// Source: https://leetcode.com/problems/max-dot-product-of-two-subsequences
// Title: Max Dot Product of Two Subsequences
// Difficulty: Hard
// Author: Mu Yang <http://muyang.pro>
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Given two arrays `nums1`and `nums2`.
//
// Return the maximum dot product between **non-empty** subsequences of nums1 and nums2 with the same length.
//
// A subsequence of a array is a new array which is formed from the original array by deleting some (can be none) of the characters without disturbing the relative positions of the remaining characters. (ie,`[2,3,5]`is a subsequence of`[1,2,3,4,5]`while `[1,5,3]`is not).
//
// **Example 1:**
//
// ```
// Input: nums1 = [2,1,-2,5], nums2 = [3,0,-6]
// Output: 18
// Explanation: Take subsequence [2,-2] from nums1 and subsequence [3,-6] from nums2.
// Their dot product is (2*3 + (-2)*(-6)) = 18.```
//
// **Example 2:**
//
// ```
// Input: nums1 = [3,-2], nums2 = [2,-6,7]
// Output: 21
// Explanation: Take subsequence [3] from nums1 and subsequence [7] from nums2.
// Their dot product is (3*7) = 21.```
//
// **Example 3:**
//
// ```
// Input: nums1 = [-1,-1], nums2 = [1,1]
// Output: -1
// Explanation: Take subsequence [-1] from nums1 and subsequence [1] from nums2.
// Their dot product is -1.```
//
// **Constraints:**
//
// - `1 <= nums1.length, nums2.length <= 500`
// - `-1000 <= nums1[i], nums2[i] <= 1000`
//
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#include <climits>
#include <vector>
using namespace std;
// 2D-DP
//
// It is similar to sequence alignment algorithm.
// We use the product of any number as the weight.
//
// Let P[i, j] = Num1[i] * Num2[j].
// Let DP[i, j] be the maximum inner product of the any non-empty subsequences of nums1[:i] and nums2[:j].
//
// DP[i, j] = max(
// P[i, j], // start from here, for the non-empty constraint
// DP[i-1, j-1] + P[i, j] // pick current numbers to extend
// DP[i-1, j], DP[i, j-1] // ignore either of current number
// )
class Solution {
constexpr static int kNInf = INT_MIN / 2;
public:
int maxDotProduct(const vector<int>& nums1, const vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
auto dp = vector(m + 1, vector(n + 1, kNInf));
for (auto i = m - 1; i >= 0; --i) {
for (auto j = n - 1; j >= 0; --j) {
int p = nums1[i] * nums2[j];
dp[i][j] = max({
p,
dp[i + 1][j + 1] + p,
dp[i + 1][j],
dp[i][j + 1],
});
}
}
return dp[0][0];
}
};
// 1D-DP
class Solution2 {
constexpr static int kNInf = INT_MIN / 2;
public:
int maxDotProduct(const vector<int>& nums1, const vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
auto curr = vector(n + 1, kNInf);
auto prev = vector(n + 1, kNInf);
for (auto i = m - 1; i >= 0; --i) {
swap(curr, prev);
for (auto j = n - 1; j >= 0; --j) {
int p = nums1[i] * nums2[j];
curr[j] = max({
p,
prev[j + 1] + p,
prev[j],
curr[j + 1],
});
}
}
return curr[0];
}
};