原题链接:AcWing 800. 数组元素的目标和
关键词: 双指针
给定两个升序排序的有序数组 A 和 B,以及一个目标值 x。
数组下标从 0 开始。
请你求出满足 A[i]+B[j]=x 的数对 (i,j)。
数据保证有唯一解。
输入格式
第一行包含三个整数 n,m,x,分别表示 A 的长度,B 的长度以及目标值 x。
第二行包含 n 个整数,表示数组 A。
第三行包含 m 个整数,表示数组 B。
输出格式
共一行,包含两个整数 i 和 j。
数据范围
数组长度不超过 105。
同一数组内元素各不相同。
1≤数组元素≤109
输入样例:
4 5 6
1 2 4 7
3 4 6 8 9
输出样例:
1 1
题目大意:
给出两个按增序排列的序列,并给出目标值x,找到两个序列中元素和为x的数的下标
思路:
如果按照朴素做法(两重循环)肯定会超时,双指针做法可以使时间复杂度从O(n2)变为O(n),在此处适用。
i指针从a数组左边开始,j指针从b数组右边开始,只要a[i]+b[j] > x 就把j指针左移,然后判断和是否等于x,等于就输出下标
代码实现:
#include <iostream>
using namespace std;
int n, m, x;
const int maxn = 100010;
int a[maxn], b[maxn];
int main(){
cin >> n >> m >> x;
for(int i = 0; i < n; i ++ ) cin >> a[i];
for(int j = 0; j < m; j ++ ) cin >> b[j];
for(int i = 0, j = m-1; i < n; i ++ ){
while(j >= 0 && a[i] + b[j] > x) j--;
if(a[i] + b[j] == x){
cout << i << " " << j << endl;
break;
}
}
return 0;
}