二分查找
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
class Solution{
public int search(int[] nums, int target){
int left = 0, right = nums.length - 1;
while(left <= right){
int mid = (right - left) / 2 + left;
int num = nums[mid];
if(num == target){
return mid;
}else if(num > target){
right = mid - 1;
}else{
left = mid + 1;
}
}
return -1;
}
}
这段代码实现了在有序数组中进行二分查找的功能。主要步骤如下:
定义两个指针 left 和 right,分别指向数组的起始和末尾位置。
在循环中,计算中间位置 mid,并获取该位置的元素值 num。
如果 num 等于目标值 target,则返回 mid,表示找到了目标值所在的位置。
如果 num 大于目标值 target,则将 right 指针向左移动到 mid - 1,缩小搜索范围。
如果 num 小于目标值 target,则将 left 指针向右移动到 mid + 1,缩小搜索范围。
如果循环结束时仍未找到目标值,则返回 -1,表示目标值不在数组中。
这段代码的时间复杂度为 O(logn),其中 n 是数组的长度,因为每次循环都将搜索范围缩小一半。
为了将二分查找算法改成符合ACM竞赛模式,我们需要调整代码以使用标准输入和输出。下面是修改后的代码:
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
// 读取数组的大小
int size = scanner.nextInt();
// 创建数组并读取数组元素
int[] nums = new int[size];
for (int i = 0; i < size; i++) {
nums[i] = scanner.nextInt();
}
// 读取目标值
int target = scanner.nextInt();
// 创建Solution对象并调用search方法查找目标值
Solution solution = new Solution();
int result = solution.search(nums, target);
// 输出结果
System.out.println(result);
}
}
class Solution {
public int search(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = (right - left) / 2 + left;
int num = nums[mid];
if (num == target) {
return mid;
} else if (num > target) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return -1;
}
}
说明
-
输入格式:
- 第一行输入一个整数
size
,表示数组的大小。 - 第二行输入
size
个整数,表示数组的元素。 - 第三行输入一个整数
target
,表示要查找的目标值。
- 第一行输入一个整数
-
输出格式:
- 输出一个整数,表示目标值在数组中的索引。如果目标值不在数组中,则输出
-1
。
- 输出一个整数,表示目标值在数组中的索引。如果目标值不在数组中,则输出
使用示例
假设输入如下:
5
1 2 3 4 5
3
程序会输出:
2
因为数组[1, 2, 3, 4, 5]
中3
的索引是2
。
你可以将这段代码保存到一个文件中,比如Main.java
,然后在命令行中使用javac Main.java
进行编译,使用java Main
运行,并在运行时通过标准输入提供数据。
移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
class Solution {
public int removeElement(int[] nums, int val) {
int n = nums.length;
int left = 0;
for (int right = 0; right < n; right++) {
if (nums[right] != val) {
nums[left] = nums[right];
left++;
}
}
return left;
}
}
这段代码用于移除数组中指定值 val 的元素,并返回新数组的长度。主要步骤如下:
定义两个指针 left 和 right,初始时都指向数组的起始位置。
使用 right 指针遍历数组,从头到尾依次检查数组中的元素。
如果 right 指针指向的元素不等于指定值 val,则将该元素复制到 left 指针指向的位置,并将 left 指针向右移动一位。
继续遍历数组直到 right 指针遍历完整个数组。
最后返回 left 指针的值,即为新数组的长度。
这段代码的时间复杂度为 O(n),其中 n 是数组的长度,因为只需遍历一次数组。
为了将 removeElement
方法改成符合ACM竞赛模式,我们需要调整代码以使用标准输入和输出。以下是修改后的代码:
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
// 读取数组的大小
int size = scanner.nextInt();
// 创建数组并读取数组元素
int[] nums = new int[size];
for (int i = 0; i < size; i++) {
nums[i] = scanner.nextInt();
}
// 读取要移除的元素的值
int val = scanner.nextInt();
// 创建Solution对象并调用removeElement方法
Solution solution = new Solution();
int newLength = solution.removeElement(nums, val);
// 输出新的数组长度
System.out.println(newLength);
// 输出移除元素后的数组内容
for (int i = 0; i < newLength; i++) {
System.out.print(nums[i] + " ");
}
System.out.println();
}
}
class Solution {
public int removeElement(int[] nums, int val) {
int n = nums.length;
int left = 0;
for (int right = 0; right < n; right++) {
if (nums[right] != val) {
nums[left] = nums[right];
left++;
}
}
return left;
}
}
说明
-
输入格式:
- 第一行输入一个整数
size
,表示数组的大小。 - 第二行输入
size
个整数,表示数组的元素。 - 第三行输入一个整数
val
,表示要移除的元素的值。
- 第一行输入一个整数
-
输出格式:
- 输出一个整数,表示移除
val
后的数组的新长度。 - 输出移除
val
后的数组内容,新数组的长度由之前的输出长度决定。
- 输出一个整数,表示移除
使用示例
假设输入如下:
5
3 2 2 3 4
3
程序会输出:
3
2 2 4
因为数组[3, 2, 2, 3, 4]
中移除3
后的新数组为[2, 2, 4]
,长度是3
。
你可以将这段代码保存到一个文件中,比如Main.java
,然后在命令行中使用javac Main.java
进行编译,使用java Main
运行,并在运行时通过标准输入提供数据。