题目描述
给定两个升序排序的有序数组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. 写出朴素做法
2. 试图找单调性
3. 双指针算法优化
朴素做法 ------- $O(n * m)$
for(int i = 0; i < n; i ++){
for(int j = 0; j < m; j ++){
}
}
双指针算法 --------- $O(n + m)$
for(int i = 0, j = m - 1; i < n; i ++){
while(j >= 0 && a[i] + b[j] > x){
}
}
题目描述
#include <iostream>
using namespace std;
const int N = 100010;
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, j = m - 1; i < n; i ++){
while(j >= 0 && a[i] + b[j] > x){
j --;
}
if(a[i] + b[j] == x){
cout << i << ' ' << j << endl;
break;
}
}
return 0;
}