2018今日头条春招的一道笔试题

 

题目如下：

 

 　　这道题我们最先想到的做法，应该就是2重循环枚举数对，然后把数对放在set里去重，最后输出set的大小，即输出set.size( )。
代码如下：

#include<iostream>
#include<set>
using namespace std;

int n, k, a[100000];
set<pair<int, int>> mypairs;

int main()
{
    cin >> n >> k;
    for(int i = 0; i < n; i ++)
        cin >> a[i];
    for(int i = 0; i < n; i ++)
    {
        for(int j = 0; j < n; j ++)
        {
            if(a[i] + k == a[j])
                mypairs.insert(make_pair(a[i], a[j]));
        }
    }
    cout << mypairs.size() << endl;
    return 0;
}

 

通过使用哈希表进行枚举优化

优化后的思路：

　　 如果我们只枚举a[i]，比如a[i]=3，那么如果存在数对的话，这个数对肯定是(a[i], a[i]+k)。比如样例里，假设我枚举数对里较小的值是3，那根据差是2，较大的肯定是5。所以，问题就变成，查找a[i]+k在不在输入的数组里。

 

该代码（支持C++11标准）：

　　针对这样的问题，哈希表是一个非常好用的工具。而且更方便的是，C++的STL已经帮我们把这些工具都实现好了，提供了非常方便的接口，我们直接用行了。下面介绍一下unordered_set，unordered_set可以把它想象成一个集合，它提供了几个函数让我们可以增删查：unordered_set::insert、unordered_set::find、unordered_set::erase.....

#include<iostream>
#include<unordered_set>
using namespace std;

int n, k, x, ans = 0;
unordered_set<int> myset;

int main()
{
    cin >> n >> k;

    for(int i = 0; i < n; i ++)
    {
        cin >> x;
        myset.insert(x);
    }
    for(int x : myset)
    {
        if(myset.find(x+k) != myset.end())
            ans ++;
    }
    cout << ans << endl;
    return 0;
}

或者：(编译器不支持C++11标准的代码)

#include<iostream>
#include<set>
using namespace std;

int n, k, x, ans = 0;
set<int> myset;

int main()
{
    cin >> n >> k;

    for(int i = 0; i < n; i ++)
    {
        cin >> x;
        myset.insert(x);
    }
    for(set<int> ::iterator i = myset.begin(); i != myset.end(); i ++)
    {
        if(myset.find((*i)+k) != myset.end())
            ans ++;
    }
    cout << ans << endl;
    return 0;
}

 

运行结果图如下：