2020美团笔试题目：送餐小区数量

2020美团笔试题目：送餐小区数量

题目：现在有n个订单，为了帮助送餐小哥增加送餐量，可以让送餐小哥将同一个小区的订单同时送过去，但是现在只有订单之间的关系，要通过订单之间的关系找找到有多少个小区，将同一个小区的订单id进行输出。
输入：第一行两个整数n，m表示有n个订单，m个订单之间的关系，后面m行表示两个订单是同一个小区的

5 3
2 1
4 3
5 2

输出：

2
1 2 5
3 4

有两个小区，以及每个小区的订单id
解法：参考之前的找老乡的方法，对每一对输入建立父子关系，直到找到祖先结点停止。

#include <iostream>
#include <vector>
#include <algorithm>
#include <set>
#include <stack>
#include <map>

using namespace std;

int main(){
    int n,m;
    cin>>n>>m;
    vector<int> connect(n,0);
    for(int i=0;i<n;i++){		 // 结点关系初始化
        connect[i] = -i;
    }
    for(int i=0;i<m;i++){
        int a,b;
        cin>>a>>b;
        if(a<b){
            if(connect[b-1] < 0)
                connect[b-1] = a - 1;	// 大订单的父节点是小订单
            else
                connect[a-1] = connect[b-1];	// 大订单已经有父节点，让小订单指向有父节点的大订单
        }
        if(a>b){
            if(connect[a-1] < 0)
                connect[a-1] = b - 1;
            else
                connect[b-1] = connect[a-1];
        }

    }
    int count = 0;
    map<int,set<int>> nodeNum;  // 祖先结点，和祖先一个小区的结点
    for(int i=0;i<n;i++){
        int j = i;
        set<int> temp{j};      // 存放从当前结点到最终祖先结点的所有中间结点
        while(j != -connect[j]){  // 还有父结点的话继续查找
            temp.insert(j);
            j = connect[j];      // 更新遍历结点
        }
        if(nodeNum.find(j) != nodeNum.end()){
            nodeNum[j].insert(temp.begin(),temp.end());
        } else
            nodeNum[j] = temp;
    }
    cout<<nodeNum.size()<<endl;
    for(auto i : nodeNum){
        for(auto j:i.second){
            cout<<j+1<<' ';
        }
        cout<<endl;
    }
    return 0;
}

输出测试：