题目描述
给定两个升序排序的有序数组A和B,以及一个目标值x。数组下标从0开始。
请你求出满足A[i] + B[j] = x的数对(i, j)。
数据保证有唯一解。
输入格式
第一行包含三个整数n,m,x,分别表示A的长度,B的长度以及目标值x。
第二行包含n个整数,表示数组A。
第三行包含m个整数,表示数组B。
输出格式
共一行,包含两个整数 i 和 j。
数据范围
数组长度不超过100000。
同一数组内元素各不相同。
1≤数组元素≤109
输入样例:
4 5 6
1 2 4 7
3 4 6 8 9
输出样例:
1 1
算法:双指针
题目保证有唯一解,所以在找到答案的时候就可以直接跳出循环。
这题给的两个序列都是升序的,所以我们定义两个指针变量,一个指向第一个序列的第一个,
一个指向第二个序列的最后一个,判断这两个值的和跟目标值的大小情况。如果,大于,那么j
就往前遍历,如果等于正好就直接输出。不存在小于,因为题目保证有唯一解。
时间复杂度
O(n+m)
参考文献
y总讲解视频
C++ 代码
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100010;
//a数组:存放第一个序列 b数组存放第二个序列
int a[N],b[N];
int main(){
//n第一个数组的个数 m第二个数组的个数 x目标值
int n , m , x;
scanf("%d%d%d",&n,&m,&x);
for(int i = 0 ; i < n ; i++)scanf("%d",&a[i]);
for(int j = 0 ; j < m ; j++ )scanf("%d",&b[j]);
//双指针。i从第一个序列的第一个开始遍历,j从第二个序列的最后一个
//进行遍历,当a[i] + b[j]等于目标值x的时候,就直接输出,本题保证
//有唯一解,所以找到的时候直接跳出循环。如果不是的话,另外在进行考虑
for(int i = 0 , j = m - 1 ; i < n ;i++){
//当两个值大于目标值的时候j指针往前移,因为序列是升序的,所以往前
//才能找到。
while(a[i] + b[j] > x) j--;
if(a[i] + b[j] == x){
printf("%d %d",i,j);
break;
}
}
return 0;
}