寻找和为定值的两个数

寻找和为定值的两个数

题目描述:

输入一个数组和一个数字,在数组中查找两个数,使得它们的和正好是输入的那个数字。如果有多对数字的和等于输入的数字,输出任意一对即可。

分析和解法:

解法一:暴力穷举

直接穷举,从数组中任意选取两个数,判定它们的和是否为输入的那个数字。
源代码如下:

#include <iostream>

using namespace std;

int main()
{
    int a[100];
    int i = 0;
    char ch;
    while((ch = cin.get()) != '\n')  //此处也可以用 cin.peek() ,只预读,不取出 
    {
        cin.unget();
        cin >> a[i++];
    }
    int sum;
    cin >> sum;
    int j;
    for (j = 0; j < i - 1; j++)
        for (int k = j + 1; k < i; k++)
        {
            if (a[j] + a[k] == sum)
            {
                cout << a[j] << " " << a[k] << endl;    
                return 0;
            }
        }
    if (j = i - 1)  cout << "None" << endl;
    return 0;
}

分析:时间复杂度为 O(n ^ 2)。

解法二:排序 + 二分查找

题目相当于,对每个 a[i] ,查找 sum - a[i] 是否也在原始序列中。我们可以先排序,再用二分查找去查找 sum - a[i] 。
源代码如下:

#include <iostream>
#include <cstdlib>
#include <algorithm>

using namespace std;

int Compare(const void *elem1, const void *elem2)
{
    return *((int *)(elem1)) - *((int *)(elem2));
}

int main()
{
    int a[100];
    int i = 0;
    char ch;
    while((ch = cin.get()) != '\n')  //此处也可以用 cin.peek() ,只预读,不取出 
    {
        cin.unget();
        cin >> a[i++];
    }
    int sum;
    cin >> sum;
    qsort(a, i, sizeof(int), Compare);
    int j;
    for (j = 0; j < i; j++)
    {
        if (binary_search(a, a + i, sum - a[j]))
        {
            cout << a[j] << " " << sum - a[j] << endl;
            return 0;
        }
    }
    if (j = i)  cout << "None" << endl;
    return 0;
}

分析:时间复杂度为 O(n logn)。

解法三:前后指针扫描

此种解法针对有序的序列可以做到 O(n)。无序的序列要求先进行排序。
根据前面的分析,a[i] 在序列中,如果 a[i] + a[k] = sum 的话,那么 sum - a[i](a[k])也必然在序列中。 举个例子,如下:
原始序列:
1、 2、 4、 7、11、15
用输入数字 15 减一下各个数,得到对应的序列为:
14、13、11、8、4、 0
第一个数组以一指针 i 从数组最左端开始向右扫描,第二个数组以一指针 j 从数组最右端开始向左扫描,如果第一个数组出现了和第二个数组一样的数,即 a[i] = a[j],就找出这俩个数来了。 如上,i,j 最终在第一个和第二个序列中找到了相同的数 4 和 11 ,所以符合条件的两个数,即为 4 + 11 = 15。
源代码如下:

#include <iostream>
#include <cstdlib>
#include <algorithm>

using namespace std;

int Compare(const void *elem1, const void *elem2)
{
    return *((int *)(elem1)) - *((int *)(elem2));
}

int main()
{
    int a[100];
    int i = 0;
    while(cin.get() != '\n')  //此处也可以用 cin.peek() ,只预读,不取出 
    {
        cin.unget();
        cin >> a[i++];
    }
    int sum;
    cin >> sum;
    qsort(a, i, sizeof(int), Compare);
    int b[100];
    for (int k = 0; k < i; k++)
        b[k] = sum - a[k];
    int j;
    for (j = 0; j <= i/2; )
    {
        if (a[j] == b[i - 1])  
        {
            cout << a[j] << " " << sum - a[j] << endl;
            break;
        }
        if (a[j] < b[i - 1])  
        {
            j++;
            continue;
        }
        if (a[j] > b[i - 1])
        {
            i--;
            continue;
        }
    }
    if (j > i/2)  cout << "None" << endl;
    return 0;
}

分析:此种解法针对有序序列的时间复杂度为 O(n),空间复杂度为 O(n)。如果是无序序列,则需要先排序。

解法四:哈希表

两数与定值的一半的差的绝对值相等。
利用这一性质,我们建立起哈希映射关系。其中考虑到定值为奇数时,a[k] - sum/2 会取整可能影响结果,所以我们可以用 2 * a[k] - sum 代替。
源代码如下:

#include <iostream>
#include <cstdlib>

using namespace std;

int main()
{
    int a[100];
    int i = 0;
    cout << "输入数的范围为 -99 ~ 99: " << endl; 
    while(cin.get() != '\n')  //此处也可以用 cin.peek() ,只预读,不取出 
    {
        cin.unget();
        cin >> a[i++];
    }
    int sum;
    cin >> sum;
    int b[200] = {};  //初始化全部成员为 0 
    for (int k = 0; k < i; k++)
        b[abs(2 * a[k] - sum)]++;  //两数与定值的一半的差的绝对值相等 
    int j;
    for (j = 0; j < 200; j++)
    {
        if (b[j] > 1)  
        {
            cout << (sum + j) / 2 << " " << (sum - j) / 2 << endl;
            break;
        }
    }
    if (j > 200)  cout << "None" << endl;
    return 0;
}

分析:这种解法不需要序列有序,时间复杂度还为 O(n),但是需要以空间换取时间。

解法总结

不论原序列是有序还是无序,解决这类题有以下三种办法:

  • 1、二分(若无序,先排序后二分),时间复杂度总为 O(N log N),空间复杂度为 O(1);
  • 2、扫描一遍 X - S[i] 映射到一个数组或构造 hash 表,时间复杂度为 O(N),空间复杂度为 O(N);
  • 3、两个指针两端扫描(若无序,先排序后扫描),时间复杂度最后为:有序 O(N),无序 O(N log N + N) = O(N log N),空间复杂度都为 O(1)。
    所以,要想达到时间 O(N),空间 O(1) 的目标,除非原数组是有序的(指针扫描法),不然,当数组无序的话,就只能先排序,后指针扫描法或二分(时间 O(Nlog N),空间O(1)),或映射或 hash(时间 O(N),空间 O(N))。时间或空间,必须牺牲一个,达到平衡。

综上,若是数组有序的情况下,优先考虑两个指针两端扫描法,以达到最佳的时间复杂度O(N),空间复杂度O(1)。否则,如果要排序的话,时间复杂度最快当然是只能达到O(N log N),空间O(1)则不在话下。

特别注意:

  • 数组初始化:
    • 声明的全局数组会被编译器全部初始化为 0
    • 局部数组:
    1. 成员全部初始化:
    int myArray[10] = { 5, 5, 5, 5, 5, 5, 5, 5, 5, 5 };
    2. 将缺省的成员置为 0:
    int myArray[10] = { 1, 2 };  //initialize to 1,2,0,0,0...
    3. 初始化全部成员为 0:
    int myArray[10] = { 0 };  //all elements 0
    4. 在 C++中, 空的初始化也会将全部成员置为 0:
    int myArray[10] = {};  //all elements 0 in C++
    5. 如果声明为 static,则全部成员默认为 0,如果你不去手动初始化的话。
    static int myArray[10]; //all elements 0
    6. for 循环,一个一个赋值
    int myArray[10];
    for (int i = 0; i < 10; i++)
        myArray[i] = 0;
    7. memset(数组名,准备初始化的值,数组的大小),memset 是以字节为单位对数组所在内存进行赋值的
    memset(myArray, 0, sizeof(myArray));
    

参考资料:《编程之法》The Art of Programming By July

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,774评论 0 33
  • 概述 排序有内部排序和外部排序,内部排序是数据记录在内存中进行排序,而外部排序是因排序的数据很大,一次不能容纳全部...
    蚁前阅读 5,222评论 0 52
  • 回来和同住在一起的妹子上周约好周六去逛街,因为下周要降温。早上起来妹子有事要办,所以我就起来的很晚,磨磨蹭蹭的起来...
    拉西西菲阅读 199评论 0 0
  • If one day I bend will you come to me and help me? 如果有一天我...
    只影独行阅读 360评论 0 0