AcWing 800. 数组元素的目标和

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

题目

给定两个升序排序的有序数组 A 和 B，以及一个目标值 x。

数组下标从 0 开始。

请你求出满足 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

思路

双指针算法

i指针从 0开始 从前往后遍历

j指针从 m - 1开始 从后向前遍历

如果a[i]+b[j]比目标数字大，则j--

如果a[i]+b[j]比目标数字小，则i++

如果a[i]+b[j] 等于目标数组的大小，就返回 i 和 j

注意:数据保证有唯一解,且数组下标从 0 开始

时间复杂度O(n)

ac代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100001;
int n,m,x;
int a[N],b[N];
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(j >= 0 && a[i] + b[j] == x) cout << i << ' ' << j << endl;
    }
    return 0;
}

Java

import java.io.*;
class Main{
    static int N=100010;
    static int[] a=new int[N];
    static int[] b=new int[N];

    public static void main(String[] args) throws IOException{
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        String[] cur=in.readLine().split(" ");
        int n=Integer.parseInt(cur[0]);
        int m=Integer.parseInt(cur[1]);
        int x=Integer.parseInt(cur[2]);

        String[] A=in.readLine().split(" ");
        for(int i=0;i<n;i++) a[i]=Integer.parseInt(A[i]);
        String[] B=in.readLine().split(" ");
        for(int i=0;i<m;i++) b[i]=Integer.parseInt(B[i]);

        int i=0;
        int j=m-1;
        while(i<n && j>0){
            if(a[i]+b[j]>x){
                j--;
            }else if(a[i]+b[j]<x){
                i++;
            }else{
                System.out.println(i+" "+j);
                break;
            }
        }
    }
}