问题描述
小R最近遇到了一个数组问题。他有一个包含 NN 个元素的数组,记作 a1,a2,...,aNa1,a2,...,aN。为了分析这个数组的特性,小R定义了两个函数 L(i)L(i) 和 R(i)R(i),并希望通过这两个函数来找到一些有趣的结论。
- L(i)L(i) 是满足以下条件的最大的 jj 值:
- j<ij<i
- a[j]>a[i]a[j]>a[i]
- 如果找不到这样的 jj,那么 L(i)=0L(i)=0;如果有多个满足条件的 jj,选择离 ii 最近的那个。
- R(i)R(i) 是满足以下条件的最小的 kk 值:
- k>ik>i
- a[k]>a[i]a[k]>a[i]
- 如果找不到这样的 kk,那么 R(i)=0R(i)=0;如果有多个满足条件的 kk,选择离 ii 最近的那个。
最终,小R定义 MAX(i)=L(i)∗R(i)MAX(i)=L(i)∗R(i),他想知道在 1≤i≤N1≤i≤N 的范围内,MAX(i)MAX(i) 的最大值是多少。 程序 public class Main { public static int solution(int n, int[] array) { // 初始化 left 和 right 数组 int[] left = new int[n]; int[] right = new int[n];
// 计算 L(i)
// 遍历数组,找到每个元素的 L(i)
for (int i = 0; i < n; i++) {
left[i] = 0; // 初始化为 0
for (int j = i - 1; j >= 0; j--) {
if (array[j] > array[i]) {
left[i] = j + 1; // 因为题目中 L(i) 是 1-based index
break;
}
}
}
// 计算 R(i)
// 反向遍历数组,找到每个元素的 R(i)
for (int i = n - 1; i >= 0; i--) {
right[i] = 0; // 初始化为 0
for (int k = i + 1; k < n; k++) {
if (array[k] > array[i]) {
right[i] = k + 1; // 因为题目中 R(i) 是 1-based index
break;
}
}
}
// 计算 MAX(i) 并找到最大值
int maxProduct = 0;
for (int i = 0; i < n; i++) {
int product = left[i] * right[i];
if (product > maxProduct) {
maxProduct = product;
}
}
return maxProduct;
}
public static void main(String[] args) {
// Add your test cases here
System.out.println(solution(5, new int[]{5, 4, 3, 4, 5}) == 8);
}
}