Acwing 800

209 阅读2分钟

给定两个升序排序的有序数组 AA 和 BB,以及一个目标值 xx。

数组下标从 00 开始。

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

数据保证有唯一解。

输入格式

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

第二行包含 nn 个整数,表示数组 AA。

第三行包含 mm 个整数,表示数组 BB。

输出格式

共一行,包含两个整数 ii 和 jj。

数据范围

数组长度不超过 105105。
同一数组内元素各不相同。
1≤数组元素≤1091≤数组元素≤109

输入样例:

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

输出样例:

1 1

思路

双指针

首先寻找暴力方法,就是两个循环遍历,判断是否满足条件,当然是超时的。

// foreverking
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <vector>
using namespace std;

const int N = 100010;
int n, m, x;
int a[N], b[N];
int ans_a, ans_b;

 void check(int x[], int y[]) {
     for (int i = 0; i < n; i++) {
         for (int j = 0; j < m; j++) {
             if (x[i] == y[j]) {
                 ans_a = i;
                 ans_b = j;
             }
         }
     }
 }



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];

    check();

    return 0;
}

因为a,b是升序数组,可知a,b构成递增单调性,由于ab数组单调递增且答案唯一那么就可采用双指针算法,这也是典型的双指针算法。寻找满足a[i] + b[j] >= x条件的i,j,其中,当i增大时,j就只能减小或者不变,即b的下标只能向右移动。所以,i从0开始,j从m-1开始,寻找满足a[i] + b[j] >= x的j的值,找到满足条件的j的最小值,如果等于,就找到了,否则继续循环。

// foreverking
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <vector>
using namespace std;

const int N = 100010;
int n, m, x;
int a[N], b[N];
int ans_a, ans_b;

// void check(int x[], int y[]) {
//     for (int i = 0; i < n; i++) {
//         for (int j = 0; j < m; j++) {
//             if (x[i] == y[j]) {
//                 ans_a = i;
//                 ans_b = j;
//             }
//         }
//     }
// }

void check() {
    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;
            break;
        }
    }
}

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];

    check();

    return 0;
}

二分

没错可以二分来做,循环a数组,在每次循环中,对b数组进行二分,找x - a[i]

// foreverking
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <vector>
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++) cin >> a[i];
    for (int i = 0; i < m; i++) cin >> b[i];

    for (int i = 0; i < n; i++) {
        int temp = x - a[i];//条件,需要在b数组中找到等于temp的值
        int l = 0, r = m - 1;//二分边界
        while(l < r) {
            int mid = (l + r + 1) >> 1;
            if (b[mid] <= temp) l = mid;//更新左端点
            else
                r = mid - 1;//更新右端点
        }
        if(b[l] == temp){//找到了
            cout << i << " " << l;
            break;
        }
    }
    return 0;
}