题目链接:https://www.luogu.com.cn/problem/P1102
将A-B=C转发为A=B+C,然后不断的使用二分寻找数据中等于A的数据即可
- 二分查找A(在Java内部有内置的api可以在数组中进行二分查找Arrays.binarySearch(arr, target))找到target所在的索引index后,看看index的左右是否存在等于target的值
public static void main(String[] args) throws Exception{
BufferedReader in=new BufferedReader(new InputStreamReader(System.in));
String[] tmps=in.readLine().split(" ");
int n=Integer.valueOf(tmps[0]),c=Integer.valueOf(tmps[1]);
tmps=in.readLine().split(" ");
int res=0;
int[] data=new int[n];
for (int i = 0; i < data.length; i++) {
data[i]=Integer.valueOf(tmps[i]);
}
Arrays.sort(data);
for (int i = 0; i < data.length; i++) {
int target=data[i]+c;
int index=Arrays.binarySearch(data, target);
if(index<0) continue;
res--;
//往左找
for (int j = index; j >= 0; j--) {
if(data[j]!=target) break;
res++;
}
//往右找
for (int j = index; j < data.length; j++) {
if(data[j]!=target) break;
res++;
}
}
System.out.println(res);
in.close();
}
03.解题思路2(Hash+二分)
将A-B=C转发为A=B+C,然后不断的使用二分寻找数据中等于A的数据即可
- 使用map存贮数组中元素的个数Map
public static void main(String[] args) throws Exception{
BufferedReader in=new BufferedReader(new InputStreamReader(System.in));
String[] tmps=in.readLine().split(" ");
int n=Integer.valueOf(tmps[0]),c=Integer.valueOf(tmps[1]);
tmps=in.readLine().split(" ");
long res=0;
//key为输入数据 value为数据个数
Map map=new HashMap<>();
for (int i = 0; i < tmps.length; i++) {
int key=Integer.valueOf(tmps[i]);
if(map.containsKey(key)) {
map.put(key, map.get(key)+1);
}else {
map.put(key, 1L);
}
}
int[] data=new int[map.size()];
int index=0;
for(Entry entry:map.entrySet()) {
data[index]=entry.getKey();
index++;
}
Arrays.sort(data);
for (int i = 0; i < data.length; i++) {
int target=data[i]+c;
index=Arrays.binarySearch(data, target);
if(index>0) res+=map.get(data[index])*map.get(data[i]);
}
System.out.println(res);
in.close();
}
04.思路3两次二分
将A-B=C转发为A=B+C,然后不断的使用二分寻找数据中等于A的数据即可
- 二分查找查找>=A的第一个元素的下标,记为l如果data[l]==A,则继续 二分查找查找<=A的最后一个元素的下标,记为r本次枚举中,data[i]=A的个数为(l-r+1)
public class Main {
public static void main(String[] args) throws Exception{
BufferedReader in=new BufferedReader(new InputStreamReader(System.in));
String[] tmps=in.readLine().split(" ");
int n=Integer.valueOf(tmps[0]),c=Integer.valueOf(tmps[1]);
tmps=in.readLine().split(" ");
long res=0;
int[] data=new int[n];
for (int i = 0; i =target的第一个元素
private static int searchRight(int[] arr, int target) {
int l=0,r=arr.length-1;
while (l>1;
if(arr[mid]>=target) r=mid;
else l=mid+1;
}
return l;
}
//查找<=target的最后一个元素
private static int searchLeft(int[] arr, int target) {
int l=0,r=arr.length-1;
while (l>1;
if(arr[mid]<=target) l=mid;
else r=mid-1;
}
return l;
}
}
05.继续深入练习
https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/
06.补充:二分常用套路总结https://blog.csdn.net/qq_46237746/article/details/123813308



