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 
#include 
#include 
#include 
#include 
#include 

using namespace std;

int main(){
    int n,m;
    cin>>n>>m;
    vector connect(n,0);
    for(int i=0;i>a>>b;
        if(ab){
            if(connect[a-1] < 0)
                connect[a-1] = b - 1;
            else
                connect[b-1] = connect[a-1];
        }

    }
    int count = 0;
    map> nodeNum;  // 祖先结点,和祖先一个小区的结点
    for(int i=0;i 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<

输出测试:
2020美团笔试题目:送餐小区数量_第1张图片
2020美团笔试题目:送餐小区数量_第2张图片

你可能感兴趣的:(算法)