leetcode-算法基础-1

leetcode-算法基础-1

我想我还是别偷懒的好…

---

---

1.在排序数组中查找元素的第一个和最后一个位置

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

链接：https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0, right = n-1;
        vector<int> ans;

        while ( left <= right ){
            int mid = left + (right-left)/2;
            if ( nums[mid] == target ){
                int tr1 = mid;
                while(tr1 >= 0 && nums[tr1] == target){
                    tr1--;
                }
                ans.push_back(tr1+1);
                int tr2 = mid;
                while (tr2 < n && nums[tr2] == target){
                    tr2++;
                }
                ans.push_back(tr2-1);
                return ans;
            }else if ( nums[mid] < target ){
                left = mid + 1;
            }else{
                right = mid - 1;
            }
        }
        return {-1,-1};
    }

2.搜索旋转排序数组

整数数组 nums 按升序排列，数组中的值 互不相同 。

在传递给函数之前，nums 在预先未知的某个下标 k（0 <= k < nums.length）上进行了 旋转，使数组变为 [nums[k], nums[k+1], …, nums[n-1], nums[0], nums[1], …, nums[k-1]]（下标 从 0 开始 计数）。例如， [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。

给你 旋转后 的数组 nums 和一个整数 target ，如果 nums 中存在这个目标值 target ，则返回它的下标，否则返回 -1 。

你必须设计一个时间复杂度为 O(log n) 的算法解决此问题。

链接：https://leetcode.cn/problems/search-in-rotated-sorted-array

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0, right = n-1;

        while ( left <= right ){
            int mid = (right+left)/2;
            if ( nums[mid] == target ){
                return mid;
            }else if ( nums[mid] < nums[right] ){
                if ( nums[mid] < target && nums[right] >= target ){
                    left = mid + 1;
                }else{
                    right = mid - 1;
                }
            }else{
                if ( nums[mid] > target && nums[left] <= target ){
                    right = mid - 1;
                }else{
                    left = mid + 1;
                }
            }
        }
        return -1;
    }
};

3.搜索二维矩阵

编写一个高效的算法来判断 m x n 矩阵中，是否存在一个目标值。该矩阵具有如下特性：

每行中的整数从左到右按升序排列。
每行的第一个整数大于前一行的最后一个整数。

链接：https://leetcode.cn/problems/search-a-2d-matrix

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int n = matrix.size(), m = matrix[0].size();
        for ( int i = 0 ; i < n ; ++i ){
            if ( matrix[i][0] <= target && matrix[i][m-1] >= target ){
                for ( int j = 0 ; j < m ; ++j ){
                    if ( matrix[i][j] == target ){
                        return true;
                    }
                }
                break;
            }
        }
        return false;
    }
};