题目来源:AcWing 800. 数组元素的目标和
一、题目描述
给定两个升序排序的有序数组 A A A 和 B B B,以及一个目标值 x x x。
数组下标从 0 0 0 开始。
请你求出满足 A [ i ] + B [ j ] = x A[i]+B[j]=x A[i]+B[j]=x 的数对 ( i , j ) (i,j) (i,j)。
数据保证有唯一解。
输入格式
第一行包含三个整数
n
,
m
,
x
n,m,x
n,m,x,分别表示 A 的长度,B 的长度以及目标值 x。
第二行包含 n n n 个整数,表示数组 A。
第三行包含 m m m 个整数,表示数组 B。
输出格式
共一行,包含两个整数
i
i
i 和
j
j
j。
数据范围
数组长度不超过
1
0
5
10^5
105。
同一数组内元素各不相同。
1
≤
1≤
1≤ 数组元素
≤
1
0
9
≤10^9
≤109
输入样例:
4 5 6
1 2 4 7
3 4 6 8 9
输出样例:
1 1
二、算法思路
首先,双指针算法需要确定一个暴力的方法,然后才能用双指针去优化它。
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
if (a[i] + b[j] == x)
{
printf("%d %d\n", i, j);
break;
}
那么我们看看能否找打单调性的规律,来用双指针优化。
对于
a
a
a 数组每一个
i
i
i ,都在
b
b
b 数组中找到最小的
j
j
j,使得
a
[
i
]
+
b
[
j
]
>
x
a [i] + b[j] > x
a[i]+b[j]>x,当
i
i
i 往后移动的时候,
j
j
j 一定是往前移动的。
三、代码
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int a[N], b[N], n, m, x;
int main()
{
scanf("%d%d%d", &n, &m, &x);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
for (int i = 0; i < m; i++) scanf("%d", &b[i]);
for (int i = 0, j = m - 1; i < n; i++)
{
// 因为保证了一定有解
while (a[i] + b[j] > x) j--;
if (a[i] + b[j] == x)
{
printf("%d %d\n", i, j);
break;
}
}
return 0;
}