leetCode_Minimum Height Trees

题意：给定一个具有树性质的图（其实就是无向无环图），若把某一个节点作为树的根节点，则每个树都有一个高度。求最小高度树对应的根节点。

解法：每一次遍历，把叶子节点都剪掉，留到最后的1——2个节点就是答案。

代码：

vector<int> findMinHeightTrees(int n, vector<pair<int, int>>& edges)
{
    vector<int> du(n,0);
    map<int,set<int>> e;
    vector<bool> isUsed(n,false);
    int i,j,k=0,ok,x;
    for(i=0; i<edges.size(); i++)
    {
        du[edges[i].first]++;
        du[edges[i].second]++;
        e[edges[i].first].insert(edges[i].second);
        e[edges[i].second].insert(edges[i].first);
    }

    while(true)
    {
        vector<int> one;
        if(n-k<=2) break;
        for(i=0; i<n; i++)
        {
            if(du[i]==1)
            {
                isUsed[i]=true;
                k++;
                one.push_back(i);
            }
        }
        for(i=0;i<one.size();i++)
        {
            du[one[i]]--;
            du[*(e[one[i]].begin())]--;
            e[*(e[one[i]].begin())].erase(one[i]);
        }
    }
    vector<int> ans;
    for(i=0;i<n;i++)
    {
        if(isUsed[i]==false)
        {
            ans.push_back(i);
        }
    }
    return ans;
}