2021.7.10

 寻找第K大的数

给定整数数组 nums 和整数 k，请返回数组中第 k 个最大的元素。

请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int beg = 0;
        int end = nums.size() - 1;
        while (true) {
            auto mid = partion(nums, beg, end);
            if (mid == k - 1) return nums[mid];
            if (mid > k - 1) end = mid - 1;
            else beg = mid + 1;
        }
    }
    
    // [beg, end]闭区间，pivot取末尾元素，比pivot大的放左边，比pivot小的放右边
    int partion(vector<int>& v, int beg, int end) {
        auto k = beg;
        for (auto i = beg; i < end; ++i)
            if (v[i] > v[end])
                swap(v[k++], v[i]);
        swap(v[k], v[end]);
        return k;
    }
};

 不过最方便的还是用C++自带的优先队列，代码如下：

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int, vector<int>, greater<int>> q;
        for (auto e : nums) {
            if (q.size() < k) q.push(e);
            else if (e > q.top()) {
                q.pop();
                q.push(e);
            }
        }
        return q.top();
    }
};

两数之和

题目：给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍

• 暴力解法

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> ans;
        for(int i=0;i<nums.size();i++){
              for(int j=i+1;j<nums.size();j++){
                  if(nums[i]+nums[j]==target){
                      ans.push_back(i);
                      ans.push_back(j);
                      return ans;
                  }
              }
        }
        return ans;
    }
};

• hashmap

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> ans;
        unordered_map<int,int>hashmap;
        for(int i=0;i<nums.size();i++){
           int another = target - nums[i];
           if(hashmap.count(another)){
               ans = vector<int> ({hashmap[another], i});
               break;
           }
           hashmap[nums[i]] = i;
        }
        return ans;
    }
};