字节题-最大点

编程题1_字节跳动2018校招测试开发方向（第一批）_牛客网

先写个暴力的，思路就是先排序，然后对于每个元素都将它与它后面的全部元素进行比较。可以过60%：

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main(){
    int n;
    cin >> n;
    vector<pair<int, int>> vec(n);
    for(int i = 0; i < n; ++i)
        cin >> vec[i].first >> vec[i].second;
    sort(vec.begin(), vec.end());//默认会按照第一个元素排序
    for(int i = 0; i < n; ++i){
        for(int j = i+1; j < n; ++j ){
            if(vec[j].second > vec[i].second)   break;
            if(j == n-1)    cout << vec[i].first << " " << vec[i].second << endl;
        }
    }
    //处理最后一个
    if(n > 1)   cout << vec.back().first << " " << vec.back().second << endl;
    return 0;
}

超时的原因很简单，数据量比较大的时候，挨个遍历是相当费时的。

那怎么减少查找耗时呢，可以使用哈希表，但是unordered_set是不支持以pair作为key的，所以可以使用set。那我一开始的思路是：

使用两个set，一个排序规则是按左元素大小，另一个是按右元素大小。然后遍历左元素中的数据，如果set1中的某个数据符合要求，那么它在set2中的位置应该是最后一个位置，就可以输出这个元素，然后在set2中删除这个元素。

代码：

#include <iostream>
#include <set>
#include <algorithm>

using namespace std;

struct cmp{
    bool operator()(const pair<int, int> &a, const pair<int, int> &b){
        return a.second < b.second;
    }
};

int main(){
    int n;
    cin >> n;
    set<pair<int, int>> set1;
    set<pair<int, int>, cmp> set2;
    for(int i = 0; i < n; ++i){
        int a, b;
        cin >> a >> b;
        set1.insert({a, b});
        set2.insert({a, b});
    }
    
    for(const auto &v : set1){
        if(set2.find(v) == --set2.end()) cout << v.first << " " << v.second << endl;
        set2.erase(v);
    }
    
    return 0;
}

比较遗憾的是这次不是超时，是超内存了，但是通过了80%，下面解决超内存的问题。

思路还是先进行排序，可以按照y降序，然后遍历数组，记录遍历过程中最大的x的值max_x:

• 如果当前点x <= max_x，因为y是降序的，所以当前点的y肯定比之前的要小，所以当前点必然不满足要求
• 如果当前点x > max_x，因为y是降序的，所以当前点的y肯定比之后所有的y要大，所以当前点满足要求

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

bool cmp(const pair<int, int> &a, const pair<int, int> &b){  
    return a.second > b.second;  //y降序
}
pair<int, int> v[500001];

int main(){    
    int n;
    cin >> n;
    for(int i = 0; i < n; ++i)  cin >> v[i].first >> v[i].second;
    sort(v, v+n, cmp);
    int max_x = -1;//记录当前最大的x
    for(int i = 0; i < n; ++i){
        if(v[i].first > max_x){
            max_x = v[i].first;
            cout << v[i].first << " " << v[i].second << endl;
        }
    }
    return 0;
}

能够ac了，但是效率很一般。

在给定二维平面整数点集输出“最大点集”算法（今日头条面试题） - RonTang - 博客园里面，看到一个过滤的思路，数据规模越大的时候，其实会有很多很多的点是不符合要求的，我们可以寻找一个比较好的点，将那些不符合要求的点先过滤掉，然后再排序(之前的算法复杂度主要就是排序的复杂度，而简单的过滤可以大量减小数据规模)，再进行上一步的操作。

那么这个点的选取可以比较随意一点，比如选取x+y最大的那个点，或者选取x和y的和减去x和y差的绝对值最大的点，或者xy乘积最大。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

bool cmp(const pair<int, int> &a, const pair<int, int> &b){  
    return a.second > b.second;  //y降序
}

bool cmp_max(const pair<int, int> &a, const pair<int, int> &b){  
    return a.first*a.second > b.first*b.second;  //y降序
}


pair<int, int> v[500001];
pair<int, int> filter[500001];

int main(){    
    int n;
    cin >> n;
    for(int i = 0; i < n; ++i)  cin >> v[i].first >> v[i].second;
    auto filter_p = *max_element(v, v+n,cmp_max );
    int cnt = 0;//过滤后的元素个数
    for(int i = 0; i < n; ++i){
        if(v[i].first >= filter_p.first || v[i].second >= filter_p.second)
            filter[cnt++] = v[i];
    }

    sort(filter, filter+cnt, cmp);
    int max_x = -1;//记录当前最大的x
    for(int i = 0; i < cnt; ++i){
        if(filter[i].first > max_x){
            max_x = filter[i].first;
            cout << filter[i].first << " " << filter[i].second << endl;
        }
    }
    return 0;
}