算法训练营 day01

sxx377

已于 2024-08-02 20:25:37 修改

阅读量849

点赞数 26

文章标签：算法数据结构 c++ leetcode

于 2024-07-31 21:59:27 首次发布

本文链接：https://blog.csdn.net/2203_75539867/article/details/140813275

版权

704. 二分查找

给定一个 n 个元素有序的（升序）整型数组 nums 和一个目标值 target ，写一个函数搜索 nums 中的 target，如果目标值存在返回下标，否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

提示：

你可以假设 nums 中的所有元素是不重复的。
n 将在 [1, 10000]之间。
nums 的每个元素都将在 [-9999, 9999]之间。

注意事项:

1，数组是有序的，否则需要自己进行排序。（二分查找）
2，题目已经假设，数组中的数都是不重复的，因为如果重复，结果就不唯一了。

代码如下:

循环结构代码:

#include "iostream"
#include <vector>
using namespace std;
class Solution {
public:

    int search(vector<int>& nums, int target) {
        int left = 0, right = nums.size() - 1;     //二分查找，left为左指针，right为右指针
        while (left <= right) {                    //使用循环
            int mid = left + (right - left) / 2;   //mid为中间值
            if (target == nums[mid]) return mid;     //如果数组当前中间值等于目标值，直接返回mid
            if (target < nums[mid]) right = mid - 1;  //如果数组当前中间值小于目标值，更新数组右指针，指向mid-1的位置
            if (target > nums[mid]) left = mid + 1;     //如果数组当前中间值大于目标值，更新数组左指针，指向mid+1的位置
        }
        return -1;                                    //如果找不到目标值，返回-1
    }
};
int main() {
    int n,target; 
    cin >> n>>target; //输入数组个数n，目标值target
    vector<int> nums(n);  //定义数组
    for (int i = 0; i < n; i++) cin >> nums[i]; //输入数组
    Solution p;
    cout << p.search(nums, target);  
}

也可以使用递归，其核心代码如下(省略主函数):

class Solution {
public:

    int BinSearch2(vector<int>& nums, int target,int left,int right) {   //二分查找
        int mid = left + (right - left) / 2;                              //定义mid为中间值
        if (left > right) return -1;                                      
        if (target > nums[mid]) return BinSearch2(nums, target, mid + 1, right);  //目标值大于数组中间值，更新left为mid+1
        if (target < nums[mid]) return BinSearch2(nums, target, left, mid - 1);   //目标值小于数组中间值，更新right为mid-1
        if (target == nums[mid]) return mid;                 //目标值等于数组中间值，返回mid
        return -1;                       //找不到就返回-1
    }
    int search(vector<int>& nums, int target) {
        return BinSearch2(nums, target, 0, nums.size() - 1);
    }
};

35. 搜索插入位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

提示:

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为 无重复元素 的升序排列数组
-104 <= target <= 104

#include "iostream"
#include <vector>
using namespace std;
class Solution {
public:
    
    int searchInsert(vector<int>& nums, int target) {
        int left = 0, right = nums.size() - 1;
        int mid = left + (right - left) / 2;
        while (left < right) {
            
            if (target == nums[mid]) return mid;
            if (target < nums[mid]) right = mid;
            if (target > nums[mid]) left = mid+1 ;
            mid = left + (right - left) / 2;
        };
        if (mid == nums.size() - 1&&target>nums[mid]) return mid + 1;
        return mid;
    }
};
int main() {
    int n,target;
    cin >> n>>target;
    vector<int> nums(n);
    for (int i = 0; i < n; i++) cin >> nums[i];
    Solution p;
    cout << p.searchInsert(nums, target);
}

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

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

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

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

示例 1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例 2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

思路:定义左边界函数去寻找左边界，右边界函数去寻找右边界

寻找target在数组里的左右边界，有如下三种情况：

情况一：target 在数组范围的右边或者左边，例如数组{3, 4, 5}，target为2或者数组{3, 4, 5},target为6，此时应该返回{-1, -1}
情况二：target 在数组范围中，且数组中不存在target，例如数组{3,6,7},target为5，此时应该返回{-1, -1}
情况三：target 在数组范围中，且数组中存在target，例如数组{3,6,7},target为6，此时应该返回{1, 1}

#include "iostream"
#include <vector>
using namespace std;
class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        int leftBorder = getLeftBorder(nums, target);
        int rightBorder = getRightBorder(nums, target);
        // 情况一
        if (leftBorder == -2 || rightBorder == -2) return { -1, -1 };
        // 情况三
        if (rightBorder - leftBorder > 1) return { leftBorder + 1, rightBorder - 1 };
        // 情况二
        return { -1, -1 };
    }
private:
    int getRightBorder(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int rightBorder = -2; // 记录一下rightBorder没有被赋值的情况
        while (left <= right) {
            int middle = left + ((right - left) / 2);
            if (nums[middle] > target) {
                right = middle - 1;
            }
            else { // 寻找右边界，nums[middle] == target的时候更新left
                left = middle + 1;
                rightBorder = left;
            }
        }
        return rightBorder;
    }
    int getLeftBorder(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int leftBorder = -2; // 记录一下leftBorder没有被赋值的情况
        while (left <= right) {
            int middle = left + ((right - left) / 2);
            if (nums[middle] >= target) { // 寻找左边界，nums[middle] == target的时候更新right
                right = middle - 1;
                leftBorder = right;
            }
            else {
                left = middle + 1;
            }
        }
        return leftBorder;
    }
};
int main() {
    int n,target;
    cin >> n>>target;
    vector<int> nums(n);
    for (int i = 0; i < n; i++) cin >> nums[i];
    Solution p;
    vector<int> v = p.searchRange(nums, target);
    cout << v[0] <<"  "<< v[1];
}

27. 移除元素

给你一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。

假设 nums 中不等于 val 的元素数量为 k，要通过此题，您需要执行以下操作：

更改 nums 数组，使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。

用户评测：

评测机将使用以下代码测试您的解决方案：

int[] nums = [...]; // 输入数组
int val = ...; // 要移除的值
int[] expectedNums = [...]; // 长度正确的预期答案。
                            // 它以不等于 val 的值排序。

int k = removeElement(nums, val); // 调用你的实现

assert k == expectedNums.length;
sort(nums, 0, k); // 排序 nums 的前 k 个元素
for (int i = 0; i < actualLength; i++) {
    assert nums[i] == expectedNums[i];
}

如果所有的断言都通过，你的解决方案将会通过。

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2,_,_]
解释：你的函数函数应该返回 k = 2, 并且 nums 中的前两个元素均为 2。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3,_,_,_]
解释：你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。
注意这五个元素可以任意顺序返回。
你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。

提示：

0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100

思路一:定义双指针，一左一右，左指针i指向数组第一个数，右指针j指向最后一个数组

#include "iostream"
#include <vector>
#include <algorithm>

using namespace std;
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int i = 0, j = nums.size() - 1;          //定义左指针i指向下标0，右指针j指向最后一个数字下标
        while (i <= j) {                        //循环
            while (nums[j] == val) {             //当最后一个数字为val的时候，更新j--
                j--;
                if (j < 0) return 0;              //如果j小于0的时候，说明此时数组所有值均为val,返回0
            }
            if (i > j) return j+1;                 
            if (nums[i] == val) {               //若i指针为val，就交换i,j所指向的数字，并更新i,j
                swap(nums[i], nums[j]);
                j--;
                i++;
            }
            else {
                i++;                         //若i不为val，i++即可
            }
            
        }
        return i;                      //返回i，i左边均不为val，右边均为val
    }
};
int main() {
    int n,val;
    cin >> n>>val;
    vector<int> nums(n);
    for (int i = 0; i < n; i++) cin >> nums[i];
    Solution p;
    cout << p.removeElement(nums, val);
}

思路二:定义快慢指针

#include "iostream"
#include <vector>
#include <algorithm>

using namespace std;
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
if (val != nums[fastIndex]) {
nums[slowIndex++] = nums[fastIndex];
}
}
return slowIndex;
}
};
int main() {
int n,val;
cin >> n>>val;
vector<int> nums(n);
for (int i = 0; i < n; i++) cin >> nums[i];
Solution p;
cout << p.removeElement(nums, val);
}

977. 有序数组的平方

给你一个按 非递减顺序 排序的整数数组 nums，返回 每个数字的平方 组成的新数组，要求也按 非递减顺序 排序。

示例 1：

输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]

示例 2：

输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

提示：

1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按 非递减顺序 排序

进阶：

请你设计时间复杂度为 O(n) 的算法解决本问题

#include "iostream"
#include <vector>
#include <algorithm>

using namespace std;
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> res(nums.size(),-1);
        int k = res.size() - 1;
        int i = 0, j = nums.size() - 1;
        while (i <= j) {
            if (abs(nums[i]) > abs(nums[j])) {
                res[k] = nums[i] * nums[i];
                i++;
                k--;
            }
            else {
                res[k] = nums[j] * nums[j];
                j--;
                k--;
            }
        }
    }
};
int main() {
    int n,val;
    cin >> n>>val;
    vector<int> nums(n);
    for (int i = 0; i < n; i++) cin >> nums[i];
    Solution p;
    
}

sxx377

关注

26
点赞
踩
26

收藏

觉得还不错? 一键收藏
0
评论
算法训练营 day01

你的函数应该返回 k = 5，并且 nums 中的前五个元素为 0,0,1,3,4。思路一:定义双指针，一左一右，左指针i指向数组第一个数，右指针j指向最后一个数组。2，题目已经假设，数组中的数都是不重复的，因为如果重复，结果就不唯一了。你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。你在返回的 k 个元素之外留下了什么并不重要（因此它们并不计入评测）。1，数组是有序的，否则需要自己进行排序。平方后，数组变为 [16,1,0,9,100]排序后，数组变为 [0,1,9,16,100]
复制链接

扫一扫