你的国家有无数个湖泊,所有湖泊一开始都是空的。当第 n 个湖泊下雨的时候,如果第 n 个湖泊是空的,那么它就会装满水,否则这个湖泊会发生洪水。你的目标是避免任意一个湖泊发生洪水。

    给你一个整数数组 rains ,其中:

    rains[i] > 0 表示第 i 天时,第 rains[i] 个湖泊会下雨。

    rains[i] == 0 表示第 i 天没有湖泊会下雨,你可以选择 一个 湖泊并 抽干 这个湖泊的水。

    请返回一个数组 ans ,满足:

    ans.length == rains.length

    如果 rains[i] > 0 ,那么ans[i] == -1 。

    如果 rains[i] == 0 ,ans[i] 是你第 i 天选择抽干的湖泊。

    如果有多种可行解,请返回它们中的 任意一个 。如果没办法阻止洪水,请返回一个 空的数组 。

    请注意,如果你选择抽干一个装满水的湖泊,它会变成一个空的湖泊。但如果你选择抽干一个空的湖泊,那么将无事发生(详情请看示例 4)。

    示例 1:

    输入:rains = [1,2,3,4]
    输出:[-1,-1,-1,-1]
    解释:第一天后,装满水的湖泊包括 [1]
    第二天后,装满水的湖泊包括 [1,2]
    第三天后,装满水的湖泊包括 [1,2,3]
    第四天后,装满水的湖泊包括 [1,2,3,4]
    没有哪一天你可以抽干任何湖泊的水,也没有湖泊会发生洪水。
    示例 2:

    输入:rains = [1,2,0,0,2,1]
    输出:[-1,-1,2,1,-1,-1]
    解释:第一天后,装满水的湖泊包括 [1]
    第二天后,装满水的湖泊包括 [1,2]
    第三天后,我们抽干湖泊 2 。所以剩下装满水的湖泊包括 [1]
    第四天后,我们抽干湖泊 1 。所以暂时没有装满水的湖泊了。
    第五天后,装满水的湖泊包括 [2]。
    第六天后,装满水的湖泊包括 [1,2]。
    可以看出,这个方案下不会有洪水发生。同时, [-1,-1,1,2,-1,-1] 也是另一个可行的没有洪水的方案。
    示例 3:

    输入:rains = [1,2,0,1,2]
    输出:[]
    解释:第二天后,装满水的湖泊包括 [1,2]。我们可以在第三天抽干一个湖泊的水。
    但第三天后,湖泊 1 和 2 都会再次下雨,所以不管我们第三天抽干哪个湖泊的水,另一个湖泊都会发生洪水。
    示例 4:

    输入:rains = [69,0,0,0,69]
    输出:[-1,69,1,1,-1]
    解释:任何形如 [-1,69,x,y,-1], [-1,x,69,y,-1] 或者 [-1,x,y,69,-1] 都是可行的解,其中 1 <= x,y <= 10^9
    示例 5:

    输入:rains = [10,20,20]
    输出:[]
    解释:由于湖泊 20 会连续下 2 天的雨,所以没有没有办法阻止洪水。

    提示:

    1 <= rains.length <= 10^5
    0 <= rains[i] <= 10^9

    1. class Solution {
    2. public:
    3. vector<int> avoidFlood(vector<int>& rains) {
    4. unordered_map<int, int> hash_map; // 存储已经装满水的湖泊id与哪一天<river_id, i>
    5. list<int> zero_idxes; // 记录可以进行抽水的天id
    6. int len = rains.size();
    7. vector<int> ans(len, 1);
    8. // bool Is_Flood = false;
    9. int start = 0; // 从start天开始下雨
    10. long rains_start = rains[start]; // 如果前几天不下雨==0,则没有可以排水的湖泊
    11. while(rains_start == 0 && start < len){
    12. ans[start] = 1;
    13. rains_start = rains[++start];
    14. }
    15. //
    16. for(int i=start; i<len; ++i){
    17. long river_id = rains[i];
    18. if (river_id == 0){
    19. zero_idxes.push_back(i);
    20. // 暂时不确定排哪一个湖泊的水: ans不定
    21. }
    22. else{
    23. if(hash_map.find(river_id) == hash_map.end()){
    24. hash_map[river_id] = i;
    25. ans[i] = -1;
    26. }
    27. else{
    28. ans[i] = -1;
    29. int idx = hash_map[river_id]; // idx天时候river_id已经满了
    30. // 在idx天之前是否存在可以抽水的天将river_id抽空
    31. int cnt = 0;
    32. int sizes = zero_idxes.size();
    33. if (cnt == sizes){
    34. ans = vector<int>{};
    35. return ans;
    36. }
    37. while(cnt < sizes){
    38. int zero_idx = zero_idxes.front();
    39. if (zero_idx > idx)
    40. {
    41. ans[zero_idx] = river_id;
    42. hash_map[river_id] = i; // river_id第一次出现的id重新更新为i了;
    43. zero_idxes.pop_front();
    44. break;
    45. }
    46. else{
    47. ans[zero_idx] = 1;
    48. zero_idxes.push_back(zero_idx); // 将zero_idx移动到后面,可能其他的湖泊用的着
    49. zero_idxes.pop_front();
    50. }
    51. ++cnt;
    52. // 没法将river_id抽干,失败
    53. if (cnt == sizes)
    54. {
    55. ans = vector<int>{};
    56. return ans;
    57. }
    58. }
    59. }
    60. }
    61. }
    62. // 还有没有用完的抽水天数,赋值为1
    63. /*
    64. while(!zero_idxes.empty())
    65. {
    66. int idx = zero_idxes.front();
    67. ans[idx] = 1;
    68. zero_idxes.pop_front();
    69. }
    70. */
    71. return ans;
    72. }
    73. };