acwing800.数组元素的目标和(双指针)

题目传送门:800.数组元素的目标和

给定两个升序排序的有序数组 A 和 B,以及一个目标值 x。

数组下标从 0 开始。

请你求出满足 A[i]+B[j]=x 的数对 (i,j)。

数据保证有唯一解。

输入格式

第一行包含三个整数 n,m,x,分别表示 A 的长度,B 的长度以及目标值 x。

第二行包含 n 个整数,表示数组 A。

第三行包含 m 个整数,表示数组 B。

输出格式

共一行,包含两个整数 i 和 j。

数据范围

数组长度不超过 1e5。
同一数组内元素各不相同。
1≤数组元素≤1e9

输入样例:
4 5 6
1 2 4 7
3 4 6 8 9
输出样例:
1 1

试题解析:

题意解析:

本题给了A和B两个有序数组,长度分别为n,m。
有唯一情况,使A中某元素与B中某元素的和为x,求出两元素的下标。

方法解析:
  • 暴力做法:两个for循环依次匹配,直至找出结果,时间复杂度为O(m*n)。
  • 双指针做法:已知AB为有序数组,可以通过一次for循环遍历元素,i从0开始向后遍历,j从m-1开始向前遍历。
    已知数组单调递增:
    若a[i]+b[j]>x,那么可以继续判断a[i]+b[j-1]是否>x。
    若a[i]+b[j]x。
    最终的目的是找到和为x的两个元素,由于答案唯一且数组单调递增,我们无需考虑其他情况。
代码如下:
#include
using namespace std;
constexpr int N = 100010;
int a[N], b[N];
int n, m, x;
int main() {

	cin >> n >> m >> x;
	for (int i = 0; i < n; i++) cin >> a[i];
	for (int i = 0; i < m; i++) cin >> b[i];
	for (int i = 0, j = m - 1; i < n; i++) {
		while (j && a[i] + b[j] > x) j--;//在a元素固定的情况下,使b元素减小
		if (a[i] + b[j] == x){
			cout << i << " " << j << endl;
			break;
		}
	}

	return 0;
}

 

你可能感兴趣的:(算法)