Leetcode 310. Minimun Height Trees使用set快速删除元素

本文链接：https://blog.csdn.net/plcokefly100/article/details/50765750

set是C++STL中的一种集合类型，可以实现对元素的查找、添加、删除、修改。Set是一个集合，其中不能包含重复元素。删除set中的指定元素是一种很常见的操作，使用迭代器即可。例如:

void delete_set(set<int> &s,int m){
    set<int>::iterator it;
    for(it=s.begin();it!=s.end();){
        if( *it == m ){
            s.erase(it);
            break;
        }
    }
}

但是，这样的方式需要遍历整个集合，时间复杂度为O(n)。事实上，set的erase函数可以直接传入想要删除的元素作为参数，在O(1)时间内完成删除操作。例如：

set<int> s;
s.insert(5);
cout << s.size() << endl;//打印出1
s.erase(5);
cout << s.size() << endl;//打印出0

有关set的这个用法是本人在刷leetcode上的题310. Minimum Height Trees时搜到的。这里使用set保存图中每一个顶点的邻接点，在执行类拓扑排序的过程中可以很方便的实现删除顶点操作。之前使用vector一直报时间超时，换成set后直接通过。本题代码如下：

vector<int> findMinHeightTrees(int n, vector<pair<int, int>>& edges) {
        map<int,set<int>> graph;
        queue<int> q;
        vector<int> result;
        int *degree = new int[n];
        for(int i=0;i < n; i++)
            degree[i]=0;
        if( n == 1){
            result.push_back(0);
            return result;
        }
        int rest=n;
        for(int i=0;i < edges.size(); i++){
            degree[edges[i].first]++;
            degree[edges[i].second]++;
            if( graph.find(edges[i].first) == graph.end()){
                graph[edges[i].first] = set<int>();
            }
            if( graph.find(edges[i].second) == graph.end()){
                graph[edges[i].second] = set<int>(); 
            }
            graph[edges[i].first].insert(edges[i].second);
            graph[edges[i].second].insert(edges[i].first);
        }
        for(int i=0;i < n; i++){
            if( degree[i] == 1){
                q.push(i);
                rest--;
            }
        }
        while( rest > 0 ){
            queue<int> c;
            while( !q.empty() ){
                int p=q.front();
                q.pop();
                auto it = graph[p].begin();
                degree[*it]--;
                graph[*it].erase(p);
                if( degree[*it] == 1){
                    c.push(*it);
                    rest--;
                }
            }
            q=c;
        }
        while( !q.empty()){
            result.push_back(q.front());
            q.pop();
        }
        return result;
    }