抽签问题的解法分析

题目描述

你的朋友提议玩一个游戏：将写有数字的n个纸片放入口袋中，你可以从口袋中抽取4次纸片，每次记下纸片上的数字后都将其放回口袋中。如果这4个数字的和是m，就是你赢，否则就是你的朋友赢。你挑战了好几回，结果一次也没赢过，于是怒而撕破口袋，取出所有纸片，检查自己是否真的有赢的可能性。请你编写一个程序，判断当纸片上所写的数字是 ki，k和，…,kn时，是否存在抽取4次和为m的方案。如果存在，输出Yes；否则，输出No。

限制条件
1≤n≤50
m≤108
1k≤108

示例1
输入：
3 9
1 3 5
输出：
No
示例2
输入：
3 10
1 3 5
输出：
Yes

题解

题解一

假设取到的四张卡片分别是a,b,c,d，如果使用一般的遍历，则复杂度为O(n^4),代码如下所示：

bool solve() {
    for (int a = 0; a < n; a++)
        for (int b = 0; b < n; b++)
            for (int c = 0; c < n; c++)
                for (int d = 0; d < n; d++)
                    if (k[a] + k[b] + k[c] + k[d] == m)
                        return true;
    return false;
}

题解二

除了暴力搜索，我们也可以通过公式：

d=m-a-b-c

让最后一个需要的数字在数组中进行二分查找（需要先对数组排序，复杂度nlogn),这样总体的复杂度可以优化为O(n^3*logn)。

//二分查找
bool binary_search(int x,int arr[],int n) {
    int l = 0, r = n;
    while (r - l >= 1) {
        int i = (l + r) / 2;
        if (arr[i] == x)return true;
        else if (arr[i] < x)l = i + 1;
        else r = i;
    }
    return false;
}

bool solve() {
    //对数组k[]进行排序
    sort(k, k + n);
    //进行二分查找
    for(int a=0;a<n;a++)
        for (int b = 0; b < n; b++)
            for (int c = 0; c < n; c++) {
                if (binary_search(m - k[a] - k[b]-k[c],k,n))
                    return true;
            }
    return false;
}

题解三

既然对四个数字中的一个进行二分查找可以使复杂度从n⁴降到n³logn，那么我将四个数中的两个数进行二分查找岂不是更快？所以我么可以利用公式：

c+d=m-a-b

通过构建一个两数之和（c+d)的数组，再通过二分查找判断是否存在有满足条件的数，此时的复杂度为n²logn(构造数组n²，排序n²logn，遍历+二分n²logn)。

//二分查找
bool binary_search(int x,int arr[],int n) {
    int l = 0, r = n;
    while (r - l >= 1) {
        int i = (l + r) / 2;
        if (arr[i] == x)return true;
        else if (arr[i] < x)l = i + 1;
        else r = i;
    }
    return false;
}
bool solve() {
    //给两数之和数组赋值
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            kk[i * n + j] = k[i] + k[j];
    //进行排序
    sort(kk, kk + n * n);
    //对所有的两数和进行二分查找
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++) {
            if (binary_search(m - k[i] - k[j],kk,n*n))
                return true;
        }
    return false;
}

整体代码

#include<iostream>
#include<algorithm>

using namespace std;
const int MAX_N = 50;
int n, m, k[MAX_N];
int kk[MAX_N * MAX_N];

//二分查找
bool binary_search(int x) {
    int l = 0, r = n * n;
    while (r - l >= 1) {
        int i = (l + r) / 2;
        if (kk[i] == x)return true;
        else if (kk[i] < x)l = i + 1;
        else r = i;
    }
    return false;
}
bool solve() {
    //给两数之和数组赋值
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
            kk[i * n + j] = k[i] + k[j];
    //进行排序
    sort(kk, kk + n * n);
    //对所有的两数和进行二分查找
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++) {
            if (binary_search(m - k[i] - k[j]))
                return true;
        }
    return false;
}

int main() {
    cin >> n >> m;
    for (int i = 0; i < n; i++)
        cin >> k[i];

    bool f = solve();
    if (f)cout << "Yes" << endl;
    else cout << "No" << endl;
    //    cout<<"Yes"<<endl;
    return 0;
}