Add vulnerability gradient

ranjanan · Mar 15, 2016 · cef1ed0 · cef1ed0
1 parent 98ee7e3
commit cef1ed0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/graph/grad.jl b/graph/grad.jl
@@ -0,0 +1,14 @@
+using Gunrock
+function vulnerability(a::SparseMatrixCSC, n::Integer)
+ l = size(a, 1)
+ d = Dict{Int,Vector{Int}}()
+ dists = sssp(a, n)
+ max = maximum(dists)
+ for i = 0:max
+ setindex!(d, Int[], i)
+ end
+ for i = 1:l
+ push!(d[dists[i]], i)
+ end
+ d
+end