给定两个升序排序的有序数组 A和 B,以及一个目标值 x。
数组下标从 00 开始。
请你求出满足 A[i]+B[j]=x 的数对 (i,j)。
数据保证有唯一解。
输入格式
第一行包含三个整数 n,m,x,分别表示 A 的长度,B 的长度以及目标值 x。
第二行包含 n 个整数,表示数组 A。
第三行包含 m 个整数,表示数组 B。
输出格式
共一行,包含两个整数 i 和 j。
数据范围
数组长度不超过 10^5。
同一数组内元素各不相同。
1≤数组元素≤10^9
输入样例:
4 5 6
1 2 4 7
3 4 6 8 9
输出样例:
1 1
思路:
因为此题是升序给出的两组数组,上下两组数找到一个相加等于X的
锁定一组长度小的数组,然后遍历,去找a[i] + b[j] <= x 的第一个位置
比如:1 2 4 7
3 4 6 8 9
当i=0的时候,j=1;因为 1+4 < 6,证明j前面的数相加都是小于6,所以j先不用动
当i=1的时候,j=1;正好等于6,推出循环
总结:i往后走,j从后往前走,总值是不变的,一个增大,另一个就需要减小
AC代码(两种写法):
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int N = 1e5;
int a[N],b[N];
int n,m,x;
int main()
{
scanf("%d %d %d",&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>=0 && a[i] + b[j] > x) j--;
if(a[i] + b[j] == x)
{
cout << i << ' ' << j;
break;
}
}
return 0;
}
二分:
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
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;i<n;i++)
{
int y = a[i];
int l = 0,r = m-1;
while(l < r)
{
int mid = l + r >> 1;
if(y + b[mid] >= x) r = mid;
else l = mid + 1;
}
if(y+b[l] == x)
{
cout << i << ' ' << l << endl;
break;
}
}
return 0;
}
上述代码,不管是双指针还是二分,关键都在于判断条件 : a[i] + b[i] > x