-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
Copy pathmaximum-xor-score-subarray-queries.cpp
57 lines (55 loc) · 1.79 KB
/
maximum-xor-score-subarray-queries.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
// Time: O(n^2 + q)
// Space: O(n^2)
// dp
class Solution {
public:
vector<int> maximumSubarrayXor(vector<int>& nums, vector<vector<int>>& queries) {
vector<vector<int>> dp(size(nums));
for (int i = 0; i < size(nums); ++i) {
dp[i].resize(size(nums) - i);
dp[i][0] = nums[i];
}
for (int i = size(nums) - 1; i >= 0; --i) {
for (int l = 1; l + i < size(nums); ++l) {
dp[i][l] = dp[i][l - 1] ^ dp[i + 1][l - 1];
}
}
for (int i = size(nums) - 1; i >= 0; --i) {
for (int l = 1; l + i < size(nums); ++l) {
dp[i][l] = max({dp[i][l], dp[i][l - 1], dp[i + 1][l - 1]});
}
}
vector<int> result(size(queries));
for (int i = 0; i < size(queries); ++i) {
result[i] = dp[queries[i][0]][queries[i][1] - queries[i][0]];
}
return result;
}
};
// Time: O(n^2 + q)
// Space: O(n^2)
// dp
class Solution2 {
public:
vector<int> maximumSubarrayXor(vector<int>& nums, vector<vector<int>>& queries) {
vector<vector<int>> dp(size(nums), vector<int>(size(nums)));
for (int i = 0; i < size(nums); ++i) {
dp[i][i] = nums[i];
}
for (int i = size(nums) - 1; i >= 0; --i) {
for (int j = i + 1; j < size(nums); ++j) {
dp[i][j] = dp[i][j - 1] ^ dp[i + 1][j];
}
}
for (int i = size(nums) - 1; i >= 0; --i) {
for (int j = i + 1; j < size(nums); ++j) {
dp[i][j] = max({dp[i][j], dp[i][j - 1], dp[i + 1][j]});
}
}
vector<int> result(size(queries));
for (int i = 0; i < size(queries); ++i) {
result[i] = dp[queries[i][0]][queries[i][1]];
}
return result;
}
};