Suma de dos números (dicotomía)

• Titulo

   Dada una matriz y un valor específico m, encuentre dos números de la matriz y haga su suma m.

 

•  Idea de algoritmo

    La primera solución es usar un bucle doble para enumerar todos los números, la complejidad es O (n ^ 2)

for (int i=0; i<n-1; i++)
   for (int j=i+1; j<n; j++)
        if (a[i]+a[j] == m)
            break;

    La solución dos es ordenar primero la matriz, y luego para cada elemento a [i] de la matriz, la dicotomía determina si la matriz es igual a ma [i], y la complejidad es O (nlogn).

do {
    int mid = l + (r-l)/2;
    if (a[mid] == x) return mid;        // 遍历x为m-a[i]
    else if (a[mid] > x) return r = mid-1;
    else l = mid+1;
} while (l<=r);

 

• Solución     

#include <iostream>
using namespace std;
#define maxn 100010
int binarysearch(int a[], int size, int x) {
    int l=0, r=size-1;
    do {
        int mid = l + (r-l)/2;            // 防止溢出
        if (a[mid] == x) return mid;    
        else if (a[mid] > x) r = mid-1;
        else l = mid+1;
    } while (l<=r);

    return -1;
}


int main() {
    int a[maxn],n, m;
    cin >> n >> m;
    for (int i=0; i<n; i++)                // 输入数组
        cin >> a[i];
    
    for (int i=0; i<n; i++) {              // 遍历搜索m-a[i]
        int tmp = binarysearch(a,n,m-a[i]);
        if (a[tmp] + a[i] == m && tmp != i) {
            cout << i << "," << tmp << endl;
            break;
        }
    }
    

    return 0;
}