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

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

class Solution {
public:
    vector successfulPairs(vector& spells, vector& potions, long long success) {
      int len = spells.size();
      vector ans(len, 0);
      
      // key: val; 
      // value: val在spells中的所有下标
      map> indMap;
      // key: val;
      // value: 该val与potions的元素相乘大于等于 success的次数
      map resMap;
      
      // 存储spells的元素,但每个元素保证只出现一次
      vector 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 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;
    }
};

你可能感兴趣的:(数据结构与算法,C/C++,c++,leetcode,算法)