-
Notifications
You must be signed in to change notification settings - Fork 78
/
weighted_spanning_tree.mzn
45 lines (40 loc) · 2.26 KB
/
weighted_spanning_tree.mzn
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
include "fzn_wst.mzn";
include "fzn_wst_reif.mzn";
include "fzn_dwst.mzn";
include "fzn_dwst_reif.mzn";
/** @group globals.graph
Constrains the set of edges \a es of a given directed graph to be a weighted spanning tree rooted at \a r of weight \a W.
@param N: the number of nodes in the given graph
@param E: the number of edges in the given graph
@param from: the leaving node 1..\a N for each edge
@param to: the entering node 1..\a N for each edge
@param w: the weight of each edge
@param r: the root node (which may be variable)
@param es: a Boolean for each edge whether it is in the subgraph
@param K: the weight of the tree
*/
predicate d_weighted_spanning_tree(int: N, int: E, array[int] of int: from, array[int] of int: to, array[int] of int: w,
var int: r, array[int] of var bool: es, var int: K) =
assert(index_set(from) = 1..E,"dwst: index set of from must be 1..\(E)") /\
assert(index_set(to) = 1..E,"dwst: index set of to must be 1..\(E)") /\
assert(index_set(es) = 1..E,"dwst: index set of es must be 1..\(E)") /\
assert(index_set(w) = 1..E,"dwst: index set of w must be 1..\(E)") /\
fzn_dwst(N,E,from,to,w,r,es,K);
/** @group globals.graph
Constrains the set of edges \a es of a given undirected graph to be a weighted spanning tree of weight \a W.
@param N: the number of nodes in the given graph
@param E: the number of edges in the given graph
@param from: the leaving node 1..\a N for each edge
@param to: the entering node 1..\a N for each edge
@param w: the weight of each edge
@param es: a Boolean for each edge whether it is in the subgraph
@param K: the weight of the tree
**/
predicate weighted_spanning_tree(int: N, int: E, array[int] of int: from, array[int] of int: to, array[int] of int: w,
array[int] of var bool: es, var int: K) =
assert(index_set(from) = 1..E,"wst: index set of from must be 1..\(E)") /\
assert(index_set(to) = 1..E,"wst: index set of to must be 1..\(E)") /\
assert(index_set(es) = 1..E,"wst: index set of es must be 1..\(E)") /\
assert(index_set(w) = 1..E,"dwst: index set of w must be 1..\(E)") /\
fzn_wst(N,E,from,to,w,es,K);
%-----------------------------------------------------------------------------%