给定两个升序排序的有序数组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
观众老爷们,如果本思路对您有帮助,求关注一波~~~
思路
本题采用的是双指针法,很容易想到
模板
for(int i = 0, j = 0; i < n; i ++){
while(check(i, j)) j ++;
//其他操作
}
该算法是求两个有序数组a, b之中各取一个数字,使得这两个数字相加的和 == x;
如果选择了a数组中的一个最小值,为了能够使得其与b中的某个数结果取得和x,需要在b中选择最大的位置向前遍历
for(int i = 0, j = m - 1; i < n; i ++){
while(j > 0 && a[i] + b[j] > x) j --;
//a[i] + b[j] <= x
if(a[i] + b[j] == x){
printf("%d %d\n", i, j);
break;
}
}
java
import java.util.Scanner;
public class Main{
static int n, m, x;
static int N = 100010;
static int[] a = new int[N], b = new int[N];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
m = sc.nextInt();
x = sc.nextInt();
for(int i = 0; i < n; i ++){
a[i] = sc.nextInt();
}
for(int j = 0; j < m; j ++){
b[j] = sc.nextInt();
}
//O(n + m)
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) {
System.out.println(i + " " + j);
break;
}
}
}
}
python
a = [0] * 10010
b = [0] * 10010
def main():
n, m, x = list(map(int, input().split(" ")))
print(n)
global a
global b
a = list(map(int, input().split(" ")))
b = list(map(int, input().split(" ")))
i, j = 0, m - 1
while(i < n):
while(j > 0 and a[i] + b[j] > x):
j -= 1
if(a[i] + b[j] == x):
print(i + ' ' + j)
break
i += 1
main()
c++
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int n, m, x;
int a[N], b[N];
int main(){
cin >> 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, j = m - 1; i < n; i ++){
while(j > 0 && a[i] + b[j] > x) j --;
//a[i] + b[j] <= x
if(a[i] + b[j] == x){
printf("%d %d\n", i, j);
break;
}
}
return 0;
}