题目描述
给定两个升序排序的有序数组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
算法1
(二分+双指针)
C++ 代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 100010;
int a[N], b[N];
int n, m, x;
int find(int l, int r, int x)
{
while (l < r)
{
int mid = l + r >> 1;
if (b[mid] >= x)
r = mid;
else
l = mid + 1;
}
return l;
}
int main()
{
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]);
int j = m - 1;
//指针i从0到n-1遍历a数组
for (int i = 0; i < n; i++)
{
int t = x - a[i];//通过x和a[i]计算数组b里的值,并在b中查找是否有对应的值
j = find(0, j, t);//因为两个数组都是升序的,i增大必定伴随j减小(或不变),所以可以将上一个j作为搜索的右边界
if (b[j] == t)
{
printf("%d %d", i, j);
break;
}
}
return 0;
}