返回散列表

时间:2017-10-16 20:31:57

标签: graph rust connected-components

我使用petgraph,我想提取连接组件。

我想要HashMap<u32, Vec<&petgraph::graph::NodeIndex>> 使用u32作为连接组件的标识符,将Vec作为容器,引用连接组件中的所有节点。

如果这是一个糟糕的设计,请不要犹豫,指出一个更好的设计;我是一个铁锈的初学者。

我试过这样的事情:

extern crate fnv;
extern crate petgraph;

use petgraph::visit::Dfs;

use fnv::FnvHashMap; // a faster hash for small key
use fnv::FnvHashSet;


// structure definition
pub struct NodeAttr {
    pub name_real: String,
}

impl Default for NodeAttr {
    fn default() -> Self {
        NodeAttr {
            name_real: "default_name_for_testing".to_string(),
        }
    }
}


pub struct EdgesAttr {
    pub eval: f64,
    pub pid: f32,
    pub cov: f32, // minimum coverage
}

impl Default for EdgesAttr {
    fn default() -> Self {
        EdgesAttr {
            eval: 0.0,
            pid: 100.0,
            cov: 100.0,
        }
    }
}

pub fn cc_dfs<'a>(
    myGraph: &petgraph::Graph<NodeAttr, EdgesAttr, petgraph::Undirected>,
) -> FnvHashMap<u32, Vec<&'a petgraph::graph::NodeIndex>> {
    let mut already_visited = FnvHashSet::<&petgraph::graph::NodeIndex>::default();
    let mut map_real_index: FnvHashMap<u32, Vec<&petgraph::graph::NodeIndex>> =
        FnvHashMap::with_capacity_and_hasher(myGraph.node_count(), Default::default());

    let mut cpt = 0;

    for current_node_indice in myGraph.node_indices() {
        let mut current_vec: Vec<&petgraph::graph::NodeIndex> = Vec::new();
        if already_visited.contains(&current_node_indice) {
            continue;
        }
        let mut dfs = Dfs::new(&myGraph, current_node_indice);
        while let Some(nx) = dfs.next(&myGraph) {
            // the problem is around here
            // I believe the just assigned nx live only for the while
            //But it should live for the upper for loop. What to do?
            current_vec.push(&nx);
            already_visited.insert(&nx);
        }
        map_real_index.insert(cpt, current_vec);
        cpt = cpt + 1
    }

    return map_real_index;
}

fn main() {}

Cargo.toml:

enter[dependencies]
fnv="*"
petgraph="*" 

编译错误:

error[E0597]: `nx` does not live long enough
  --> src/main.rs:59:31
   |
59 |             current_vec.push(&nx);
   |                               ^^ does not live long enough
60 |             already_visited.insert(&nx);
61 |         }
   |         - borrowed value only lives until here
   |
note: borrowed value must be valid for the lifetime 'a as defined on the function body at 40:1...
  --> src/main.rs:40:1
   |
40 | / pub fn cc_dfs<'a>(
41 | |     myGraph: &petgraph::Graph<NodeAttr, EdgesAttr, petgraph::Undirected>,
42 | | ) -> FnvHashMap<u32, Vec<&'a petgraph::graph::NodeIndex>> {
43 | |     let mut already_visited = FnvHashSet::<&petgraph::graph::NodeIndex>::default();
...  |
66 | |     return map_real_index;
67 | | }
   | |_^

error[E0597]: `nx` does not live long enough
  --> src/main.rs:61:9
   |
60 |             already_visited.insert(&nx);
   |                                     -- borrow occurs here
61 |         }
   |         ^ `nx` dropped here while still borrowed
...
67 | }
   | - borrowed value needs to live until here

我克隆了载体中的节点索引并且有效:

current_vec.push(nx.clone()); // instead of (&nx)
already_visited.insert(nx.clone());`

我相信(可能是错误的)使用引用会比复制更有效。

1 个答案:

答案 0 :(得分:5)

这段小得多的代码表现出同样的问题(playground):

let mut v = Vec::new(); // Vec<&'a NodeIndex> ... but what is 'a?
for n in 0..10 {
    let nx: NodeIndex = NodeIndex::new(n);
    v.push(&nx);
}

即,你在一个循环中创建了一个短暂的NodeIndex并试图在一个寿命更长的Vec中存储对它的引用。

在这种情况下,解决方案非常简单:只需移动NodeIndex而不是参考。

    v.push(nx)

在原始代码中,修复方法没有区别。

// nit: "indices" is the plural of "index"; there is no singular word "indice"
for current_node_index in myGraph.node_indices() {
    // actually you don't need to supply a type here, but if you did...
    let mut current_vec: Vec<petgraph::graph::NodeIndex> = Vec::new();
    if already_visited.contains(&current_node_index) {
        continue;
    }
    let mut dfs = Dfs::new(&myGraph, current_node_index);
    while let Some(nx) = dfs.next(&myGraph) {
        current_vec.push(nx);
        //               ^-----v- Look Ma, no &s!
        already_visited.insert(nx);
    }
    map_real_index.insert(cpt, current_vec);
    cpt = cpt + 1
}

“但是,”你说,“我不想复制整个NodeIndex!我只想指向它!NodeIndex是一个很胖的毛发结构,对吗? “

好吧,如果那个(拥有指针)真的是你需要的,Box就是你几乎总想要的。但首先看一下NodeIndex的定义并查看source code,如果你想知道这些指数的重量级是多么重要:

pub struct NodeIndex<Ix=DefaultIx>(Ix);

NodeIndex只是一个Ix,如果您查找DefaultIx,则只是u32的别名。在64位PC上,实际上比你试图存储的指针小,而在Rust中,你不需要支付任何额外的使用成本 - 在运行时,它真的是只是一个u32

方便地NodeIndex is Copy(当IxCopy时),所以你甚至不需要再添加.clone();您可以像我上面那样current_vec.push(nx)后跟already_visited.insert(nx)。 (但即使你写了.clone(),你也不需要支付运行时费用;这只是不必要的。)