Leetcode 133. Clone Graph

复制一张图,图的顶点的数据结构如下:

struct UndirectedGraphNode {
    int label;
    vector<UndirectedGraphNode *> neighbors;
    UndirectedGraphNode(int x) : label(x) {};
};

每个顶点存储了顶点的数值和与此顶点邻接的顶点集合。其中,每个顶点的label各不相同。
这道题可以采用图的广度优先遍历来解决。对图遍历两次,第一次复制顶点的label值,第二次复制顶点之间的邻接关系。由于每个顶点的label各不相同,可以采用map作为哈希存储结构,键值即顶点的label。代码如下:

UndirectedGraphNode *cloneGraph(UndirectedGraphNode *node) {
    if( node == NULL )
        return NULL;
    queue<UndirectedGraphNode*> q;
    q.push(node);
    map<int,UndirectedGraphNode*> nodeMap;
    nodeMap[node->label] = new UndirectedGraphNode(node->label);
    while( !q.empty() ){
        UndirectedGraphNode* cur = q.front();
        q.pop();
        for(int i=0;i < cur->neighbors.size(); i++){
            if( nodeMap.find(cur->neighbors[i]->label) == nodeMap.end() ){
                nodeMap[cur->neighbors[i]->label] = new UndirectedGraphNode(cur->neighbors[i]->label);
                q.push(cur->neighbors[i]);
            }
        }
    }
    q.push(node);
    for(int i=0;i < node->neighbors.size(); i++){
        nodeMap[node->label]->neighbors.push_back(nodeMap[node->neighbors[i]->label]);
    }
    while( !q.empty() ){
        UndirectedGraphNode* cur = q.front();
        q.pop();
        for(int i=0;i < cur->neighbors.size(); i++){
            if( nodeMap[cur->neighbors[i]->label]->neighbors.size()==0 && cur->neighbors[i]->neighbors.size() > 0 ){
                for(int j=0;j < cur->neighbors[i]->neighbors.size(); j++){
                    nodeMap[cur->neighbors[i]->label]->neighbors.push_back(nodeMap[cur->neighbors[i]->neighbors[j]->label]);
                }
                q.push(cur->neighbors[i]);
            }
        }
    }
    return nodeMap[node->label];
}

第一次遍历的时候,通过在map中查找节点的label来判断节点是否已访问:如果节点未访问,则在map中复制节点,并将节点入队列。第二次遍历的时候,通过判断map中与当前访问节点label相同的节点是否已被赋予邻接关系来判断节点是否已访问:如果没有,则复制邻接关系,并将节点入队列。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值