给定两个升序排序的有序数组 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
双指针的题,就先考虑暴力做法,然后找单调性进行优化。
#include <iostream>
using namespace std;
const int N = 100010;
int n, m, x;
int a[N], b[N];
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(j >= 0 && a[i] + b[j] > x) j --;
if(a[i] + b[j] == x) // 此时a[i] + b[j] 要么相等,要么小于x,相等就输出了,小于一定是a[i]太小
{
printf("%d %d", i, j);
break;
}
}
return 0;
}
二分思路:(核心代码)
for(int i = 0, j = m - 1; i < n; i ++ )
{
int l = 0, r = m - 1; // 二分出b数组的答案索引
while(l < r)
{
int mid = l + r >> 1;
if(a[i] + b[mid] >= x)
r = mid;
else
l = mid + 1;
}
if(a[i] + b[l] == x)
{
printf("%d %d", i, l);
break;
}
}