三种语言实现双指针解决数组元素的目标和(C++/Python/Java)

题目

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

数组下标从 0 开始。

请你求出满足 A[i]+B[j]=x的数对 (i,j)。

数据保证有唯一解。

输入格式

第一行包含三个整数 n,m,x,分别表示 A 的长度,B 的长度以及目标值 x。

第二行包含 n 个整数,表示数组 A。

第三行包含 m 个整数,表示数组 B。

输出格式

共一行,包含两个整数 i 和 j。

数据范围

数组长度不超过 1e5。
同一数组内元素各不相同。
1≤数组元素≤1e9

输入样例:

4 5 6
1 2 4 7
3 4 6 8 9

输出样例:

1 1

C++

#include <bits/stdc++.h>
 
using namespace std;
 
const int N = 100010;
int a[N], b[N];
 
int main()
{
    int n, m, x;
    cin >> n >> m >> x;
    for(int i = 1; i <= n; i++)
        cin >> a[i];
    for(int i = 1; i <= m; i++)
        cin >> b[i];
    for(int i = 1, j = m; i <= n && j >= 1; i++)
    {
        while(a[i] + b[j] > x)
            j--;
        if(a[i] + b[j] == x)
        {
            cout << i-1 << " " << j-1 << endl;
            break;
        }
    }
}

Python

n, m, x = map(int, input().split())

a = list(map(int, input().split()))
b = [int(x) for x in input().split()]

i = 0
j = m-1

while i < n and j >= 0:
    while a[i] + b[j] > x:
        j -= 1
    if a[i] + b[j] == x:
        print("%d %d" % (i,j))
        break
    i += 1

Java

import java.util.*;
 
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int x = sc.nextInt();
        int[] a = new int[n+1];
        int[] b = new int[m+1];
        for(int i = 1; i <= n; i++)
            a[i] = sc.nextInt();
        for(int i = 1; i <= m; i++)
            b[i] = sc.nextInt();
        for(int i = 1, j = m; i <= n && j >= 1; i++)
        {
            while(a[i] + b[j] > x)
                j--;
            if(a[i] + b[j] == x)
            {
                System.out.printf("%d %d",i-1,j-1);
                break;
            }
        }
    }
}
posted @ 2024-08-01 11:10  火车驶向云外0218  阅读(30)  评论(0)    收藏  举报