题目描述
给定两个升序排序的有序数组A和B,以及一个目标值x。数组下标从0开始。
请你求出满足A[i] + B[j] = x的数对(i, j)。
数据保证有唯一解。
输入格式
第一行包含三个整数n,m,x,分别表示A的长度,B的长度以及目标值x。
第二行包含n个整数,表示数组A。
第三行包含m个整数,表示数组B。
输出格式
共一行,包含两个整数 i 和 j。
样例
4 5 6
1 2 4 7
3 4 6 8 9
1 1
算法1
二分
C++ 代码
#include <bits/stdc++.h>
using namespace std;
const int N = 100005;
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; i < n; ++i){
int idx = lower_bound(b, b + m, x - a[i]) - b;
for(int j = idx; j < m; ++j){
if(a[i]+b[j]==x) {
printf("%d %d\n", i, j);
return 0;
}
else break;
}
}
return 0;
}
算法2
双指针
C++ 代码
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 100005;
int n, m, x;
LL a[N], b[N];
int main(){
scanf("%d%d%d", &n, &m, &x);
for(int i = 0 ;i < n; ++i) scanf("%lld", &a[i]);
for(int i = 0; i < m; ++i) scanf("%lld", &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) {
printf("%d %d \n", i, j);
break;
}
}
return 0;
}