基础算法-数组元素目标和

数组元素目标和

问题：给定两个升序排序的有序数组A和B，以及一个目标值x。数组下标从0开始。请你求出满足A[i] + B[j] = x的数对(i, j)。

Method

经验：做这种类型的题目先使用暴力方法求解，然后分析其规律，再考虑优化解法，比如双指针解法。

1、暴力解法(时间复杂度O(n*n))：

for (int i = 0; i < n; i ++)
    for (int j = 0; j < m; j ++)
        if (A[i] + B[j] == x) pfintf("输出结果！");

2、根据给出的升序排序的有序数组A和B，可以使用单调性来解决这个问题。

//i为A数组首位值，j为B数组末尾值
for (int i = 0; i < n; i ++)
//A数组从最小的数(起始值)开始与B数组最大的数(末尾值)进行相加，如果大于x,则B数组末尾数j往前移动一位，直到A + B <= x.
    while(j >= 0 && A[i] + A[j] > x) i --;

这样求出来的时间复杂度为O(n + m)。

Example

给定两个升序排序的有序数组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≤数组元素≤1091≤数组元素≤109

输入样例：

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

输出样例：

1 1

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;

int n, m, x;
int a[N], b[N];

int main()
{
    scanf("%d%d%d", &n, &m, &x);//第一行包含三个整数n，m，x，分别表示A的长度，B的长度以及目标值x
    for (int i = 0; i < n; i ++) scanf("%d", &a[i]);//第二行包含n个整数，表示数组A。
    for (int j = 0; j < m; j ++) scanf("%d", &b[j]);//第三行包含m个整数，表示数组B。
    
    for (int i = 0, j = m - 1; i < n; i ++)//i从首位0开始，j从末尾m - 1开始
    {
        while (j >= 0 && a[i] + b[j] > x) j --;//a[i]和b[i]进行相加，如果大于x,则B数组末尾数j往前移动一位，直到A + B <= x.
        if (a[i] + b[j] == x)//满足条件输出i, j的值。
        {
            printf("%d %d\n", i, j);
            break;
        }
    }
    return 0;
}