Partition Array into Disjoint Intervals

时间:2023-03-09 16:34:39
Partition Array into Disjoint Intervals

2020-02-10 22:16:50

问题描述:

Partition Array into Disjoint Intervals

问题求解:

解法一:MultiSet O(nlog)

看了下数据规模,第一个想到的是multiset,肯定可以ac的,就直接敲了出来。

    public int partitionDisjoint(int[] A) {
TreeMap<Integer, Integer> map = new TreeMap<>();
for (int num : A) map.put(num, map.getOrDefault(num, 0) + 1);
int n = A.length;
int curr = -1;
for (int i = 0; i < n - 1; i++) {
curr = Math.max(curr, A[i]);
map.put(A[i], map.get(A[i]) - 1);
if (map.get(A[i]) == 0) map.remove(A[i]);
int key = map.firstKey();
if (key >= curr) return i + 1;
}
return -1;
}

解法二:left & right O(n)

这个方法就是先做一次预处理,使用一个数组去从右遍历到当前数字的最小值,之后再从左遍历得到当前的最大值并和后面的最大值比较即可。

解法三:O(n)

解法三就比较巧妙了。

max :记录到目前为止的最小值。

localMax :当前最左的localMax

如果我们遍历到一个数字比localMax要小的话,那么其必定是在left的,此时更新idx和localMax即可。

    public int partitionDisjoint(int[] a) {
int localMax = a[0], partitionIdx = 0, max = localMax;
for (int i = 1; i < a.length; i++) {
max = Math.max(max, a[i]);
if (localMax > a[i]) {
localMax = max;
partitionIdx = i;
}
}
return partitionIdx + 1;
}