g1401_1500.s1488_avoid_flood_in_the_city.Solution Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-java11 Show documentation
Show all versions of leetcode-in-java11 Show documentation
Java Solution for LeetCode algorithm problems, continually updating
The newest version!
package g1401_1500.s1488_avoid_flood_in_the_city;
// #Medium #Array #Hash_Table #Greedy #Binary_Search #Heap_Priority_Queue #Binary_Search_II_Day_18
// #2022_04_05_Time_82_ms_(75.08%)_Space_64.3_MB_(81.23%)
import java.util.HashMap;
import java.util.TreeSet;
public class Solution {
public int[] avoidFlood(int[] rains) {
HashMap hm = new HashMap<>();
TreeSet tree = new TreeSet<>();
int[] ans = new int[rains.length];
for (int i = 0; i < rains.length; i = i + 1) {
int val = rains[i];
if (val != 0) {
if (hm.containsKey(val)) {
int mapVal = hm.get(val);
if (tree.ceiling(mapVal) != null) {
ans[tree.ceiling(mapVal)] = val;
hm.put(val, i);
tree.remove(tree.ceiling(mapVal));
} else {
return new int[0];
}
} else {
hm.put(val, i);
}
ans[i] = -1;
} else {
tree.add(i);
}
}
for (int tr : tree) {
ans[tr] = 1;
}
return ans;
}
}