美文网首页
寻找和为定值的两个数

寻找和为定值的两个数

作者: MinoyJet | 来源:发表于2017-07-29 20:34 被阅读0次

    寻找和为定值的两个数

    题目描述:

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

    分析和解法:

    解法一:暴力穷举

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

    #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

    相关文章

      网友评论

          本文标题:寻找和为定值的两个数

          本文链接:https://www.haomeiwen.com/subject/pfmclxtx.html