800. 数组元素的目标和
给定两个升序排序的有序数组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
TLE 代码 :
#include<iostream>
#include<cstdio>
using namespace std ;
const int MAX = 100005 ;
int a[MAX] ;
int b[MAX] ;
int main() {
int n , m , x;
cin >> n>>m >>x;
for(int i = 0 ; i<n; i++ )cin >>a[i] ;
for(int j = 0 ; j<m; j++ )cin >>b[j] ;
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) ;
}
}
}
return 0 ;
}
思路 : 因为两个数组都是有序的,我们可以用二分做,我们对枚举a数组 , 二分查找可以使a+b==x的位置.
AC代码 :
#include<iostream>
#include<cstdio>
using namespace std ;
const int MAX = 100005 ;
int a[MAX] ;
int b[MAX] ;
int main() {
int n , m , x;
cin >> n>>m >>x;
for(int i = 0 ; i<n; i++ )cin >>a[i] ;
for(int j = 0 ; j<m; j++ )cin >>b[j] ;
int p = 0 ;
while(p < n) {
int l = 0, r = m - 1;
while(l < r) {
int mid = l + r >> 1;
if(a[p] + b[mid] >= x) r = mid;
else l = mid + 1;
}
if(a[p] + b[l] == x) {
cout << p << ' ' << l << endl;
return 0;
}
p++;
}
return 0 ;
}