Acwing -800. 数组元素的目标和

800. 数组元素的目标和

给定两个升序排序的有序数组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 TLE 代码 :

#include<iostream>
#include<cstdio>
using namespace std ; 
const int MAX = 100005 ;
int a[MAX] ;
int b[MAX] ;
int main() {
    
    int n , m , x;  
    cin >> n>>m >>x;  
    for(int i = 0 ; i<n; i++ )cin >>a[i] ; 
    for(int j = 0 ; j<m; j++ )cin >>b[j] ; 
    
    for(int i = 0 ; i<n ; i++ ) {
        
        for(int j = 0 ;j<m ;j++) {
            if(a[i]+b[j] == x) {
                printf("%d %d\n",i,j) ;
            }
        }
        
    }
     
    
    return 0 ; 
}

思路 : 因为两个数组都是有序的,我们可以用二分做,我们对枚举a数组 , 二分查找可以使a+b==x的位置. AC代码 :

#include<iostream>
#include<cstdio>
using namespace std ; 
const int MAX = 100005 ;
int a[MAX] ;
int b[MAX] ;
int main() {
    
    int n , m , x;  
    cin >> n>>m >>x;  
    for(int i = 0 ; i<n; i++ )cin >>a[i] ; 
    for(int j = 0 ; j<m; j++ )cin >>b[j] ; 
    int p = 0 ; 
    while(p < n) {
        int l = 0, r = m - 1;
        while(l < r) {
           int mid = l + r >> 1;
           if(a[p] + b[mid] >= x) r = mid;
           else l = mid + 1;
        }

        if(a[p] + b[l] == x) {
           cout << p <<  ' ' << l << endl; 
           return 0;
        }
        p++;
    }
    return 0 ; 
}

评论

Your browser is out-of-date!

Update your browser to view this website correctly. Update my browser now

×