Java - 计蒜客 - 寻找插入位置

给定一个已经升序排好序的数组，以及一个数 targettarget，如果 targettarget 在数组中，返回它在数组中的位置。
否则，返回 targettarget 插入数组后它应该在的位置。

假设数组中没有重复的数。以下是简单的示例：

[1,3,5,6], 5 → 2

[1,3,5,6], 2 → 1

[1,3,5,6], 7 → 4

[1,3,5,6], 0 → 0

输入格式

第一行输入一个整数 nn。

第二行输入 nn 个整数，表示数组

A

。

第三行输入 targettarget。

输出格式

输出一行，为要求返回的结果。

样例输入

3
1 3 5
2

样例输出

1

import java.util.Scanner;
public class Main{
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[]a = new int
;
for(int i=0 ;i<n ;i++){
a[i] = sc.nextInt();
}
int b = sc.nextInt();
int index1 = 0;
for(int i=0 ;i<n ;i++){
if(a[n-1]<b){
System.out.print(n);
break;
}
if(a[i]>=b){
System.out.println(index1);
break;
}else{
index1++;
}
}
}
}