6096. 咒语和药水的成功对数

最新推荐文章于 2024-07-17 18:07:42 发布

紫菜(Nori)

最新推荐文章于 2024-07-17 18:07:42 发布

阅读量190

点赞数

分类专栏：数据结构与算法 C/C++ 文章标签： c++ leetcode 算法

本文链接：https://blog.csdn.net/qq_36790074/article/details/125250700

版权

数据结构与算法同时被 2 个专栏收录

111 篇文章 0 订阅

订阅专栏

C/C++

27 篇文章 0 订阅

订阅专栏

利用hash记录spells单个数字出现的位置下标，对spells进行递增排序，对potions进行递减排序，遍历spells时，可以确定已经遍历过potions的位置不需要再遍历，接着上一次跳出的位置，继续遍历即可，这样可以保证potions只被遍历一遍。

class Solution {
public:
    vector<int> successfulPairs(vector<int>& spells, vector<int>& potions, long long success) {
      int len = spells.size();
      vector<int> ans(len, 0);
      
      // key: val; 
      // value: val在spells中的所有下标
      map<int, vector<int>> indMap;
      // key: val;
      // value: 该val与potions的元素相乘大于等于 success的次数
      map<int, int> resMap;
      
      // 存储spells的元素，但每个元素保证只出现一次
      vector<int> spe;
      
      // 填充上面的集合
      for(int i = 0; i < len; i++){
        int num = spells[i];
        
        auto it = resMap.find(num);
        if(it == end(resMap)){
          // 首次出现

          resMap.emplace(num, 0);
          
          spe.push_back(num);  
          
          // 大部分val只出现一次这里直接初始化容量为1即可
          vector<int> vec(1, i);
          indMap.emplace(num, vec);

        }else{
          // 其它位置又出现相同的下标，直接将这个下标追加到列表中
          indMap[num].push_back(i);
        }
      }  
      
      // 进行递增排序
      sort(begin(spe), end(spe));

      // 进行递减排序
      sort(begin(potions), end(potions), [](int ele0, int ele1){ return ele0 > ele1;});
      
      long long numI, numJ;

      // 存储上一个结果
      int res = 0;
      for(int i = 0; i < spe.size(); i++){
        // 当前val
        numI = spe[i];
        
        // 继续上一次终止遍历的下标
        // 由于spe的有序性，这里可以保证res之后的一定满足
        int j = res;
        
        for(; j < potions.size();){
          numJ = potions[j];
          
          // 如果不满足直接跳出
          if(numI * numJ < success){  
            break;
          }
        
          // 记录当前结果
          res = (++j);
        }
        
        // 记录当前val对应的结果
        resMap[numI] = res;
      }
      
      // 填充所有val的结果到ans
      for(auto &pair : resMap){
        // key 为对应的val
        int val = pair.first;
        
        // value 为成功的次数
        int times = pair.second;
        
        // 相同的val但是存在多个，填充对应位置
        auto &vec = indMap[val];
        for(auto &ind : vec){
          ans[ind] = times;
        }
      }
      
      return ans;
    }
};