| ... | ... |
@@ -16,8 +16,8 @@ |
| 16 | 16 |
* |
| 17 | 17 |
*/ |
| 18 | 18 |
|
| 19 |
#ifndef LEMON_MAX_MATCHING_H |
|
| 20 |
#define LEMON_MAX_MATCHING_H |
|
| 19 |
#ifndef LEMON_MATCHING_H |
|
| 20 |
#define LEMON_MATCHING_H |
|
| 21 | 21 |
|
| 22 | 22 |
#include <vector> |
| 23 | 23 |
#include <queue> |
| ... | ... |
@@ -745,7 +745,7 @@ |
| 745 | 745 |
typedef RangeMap<int> IntIntMap; |
| 746 | 746 |
|
| 747 | 747 |
enum Status {
|
| 748 |
EVEN = -1, MATCHED = 0, ODD = 1 |
|
| 748 |
EVEN = -1, MATCHED = 0, ODD = 1 |
|
| 749 | 749 |
}; |
| 750 | 750 |
|
| 751 | 751 |
typedef HeapUnionFind<Value, IntNodeMap> BlossomSet; |
| ... | ... |
@@ -844,9 +844,6 @@ |
| 844 | 844 |
} |
| 845 | 845 |
|
| 846 | 846 |
void destroyStructures() {
|
| 847 |
_node_num = countNodes(_graph); |
|
| 848 |
_blossom_num = _node_num * 3 / 2; |
|
| 849 |
|
|
| 850 | 847 |
if (_matching) {
|
| 851 | 848 |
delete _matching; |
| 852 | 849 |
} |
| ... | ... |
@@ -922,10 +919,6 @@ |
| 922 | 919 |
if (_delta3->state(e) != _delta3->IN_HEAP && blossom != vb) {
|
| 923 | 920 |
_delta3->push(e, rw / 2); |
| 924 | 921 |
} |
| 925 |
} else if ((*_blossom_data)[vb].status == UNMATCHED) {
|
|
| 926 |
if (_delta3->state(e) != _delta3->IN_HEAP) {
|
|
| 927 |
_delta3->push(e, rw); |
|
| 928 |
} |
|
| 929 | 922 |
} else {
|
| 930 | 923 |
typename std::map<int, Arc>::iterator it = |
| 931 | 924 |
(*_node_data)[vi].heap_index.find(tree); |
| ... | ... |
@@ -1036,11 +1029,6 @@ |
| 1036 | 1029 |
} |
| 1037 | 1030 |
} |
| 1038 | 1031 |
} |
| 1039 |
|
|
| 1040 |
} else if ((*_blossom_data)[vb].status == UNMATCHED) {
|
|
| 1041 |
if (_delta3->state(e) == _delta3->IN_HEAP) {
|
|
| 1042 |
_delta3->erase(e); |
|
| 1043 |
} |
|
| 1044 | 1032 |
} else {
|
| 1045 | 1033 |
|
| 1046 | 1034 |
typename std::map<int, Arc>::iterator it = |
| ... | ... |
@@ -1117,10 +1105,6 @@ |
| 1117 | 1105 |
if (_delta3->state(e) != _delta3->IN_HEAP && blossom != vb) {
|
| 1118 | 1106 |
_delta3->push(e, rw / 2); |
| 1119 | 1107 |
} |
| 1120 |
} else if ((*_blossom_data)[vb].status == UNMATCHED) {
|
|
| 1121 |
if (_delta3->state(e) != _delta3->IN_HEAP) {
|
|
| 1122 |
_delta3->push(e, rw); |
|
| 1123 |
} |
|
| 1124 | 1108 |
} else {
|
| 1125 | 1109 |
|
| 1126 | 1110 |
typename std::map<int, Arc>::iterator it = |
| ... | ... |
@@ -1157,101 +1141,6 @@ |
| 1157 | 1141 |
(*_blossom_data)[blossom].offset = 0; |
| 1158 | 1142 |
} |
| 1159 | 1143 |
|
| 1160 |
|
|
| 1161 |
void matchedToUnmatched(int blossom) {
|
|
| 1162 |
if (_delta2->state(blossom) == _delta2->IN_HEAP) {
|
|
| 1163 |
_delta2->erase(blossom); |
|
| 1164 |
} |
|
| 1165 |
|
|
| 1166 |
for (typename BlossomSet::ItemIt n(*_blossom_set, blossom); |
|
| 1167 |
n != INVALID; ++n) {
|
|
| 1168 |
int ni = (*_node_index)[n]; |
|
| 1169 |
|
|
| 1170 |
_blossom_set->increase(n, std::numeric_limits<Value>::max()); |
|
| 1171 |
|
|
| 1172 |
(*_node_data)[ni].heap.clear(); |
|
| 1173 |
(*_node_data)[ni].heap_index.clear(); |
|
| 1174 |
|
|
| 1175 |
for (OutArcIt e(_graph, n); e != INVALID; ++e) {
|
|
| 1176 |
Node v = _graph.target(e); |
|
| 1177 |
int vb = _blossom_set->find(v); |
|
| 1178 |
int vi = (*_node_index)[v]; |
|
| 1179 |
|
|
| 1180 |
Value rw = (*_node_data)[ni].pot + (*_node_data)[vi].pot - |
|
| 1181 |
dualScale * _weight[e]; |
|
| 1182 |
|
|
| 1183 |
if ((*_blossom_data)[vb].status == EVEN) {
|
|
| 1184 |
if (_delta3->state(e) != _delta3->IN_HEAP) {
|
|
| 1185 |
_delta3->push(e, rw); |
|
| 1186 |
} |
|
| 1187 |
} |
|
| 1188 |
} |
|
| 1189 |
} |
|
| 1190 |
} |
|
| 1191 |
|
|
| 1192 |
void unmatchedToMatched(int blossom) {
|
|
| 1193 |
for (typename BlossomSet::ItemIt n(*_blossom_set, blossom); |
|
| 1194 |
n != INVALID; ++n) {
|
|
| 1195 |
int ni = (*_node_index)[n]; |
|
| 1196 |
|
|
| 1197 |
for (InArcIt e(_graph, n); e != INVALID; ++e) {
|
|
| 1198 |
Node v = _graph.source(e); |
|
| 1199 |
int vb = _blossom_set->find(v); |
|
| 1200 |
int vi = (*_node_index)[v]; |
|
| 1201 |
|
|
| 1202 |
Value rw = (*_node_data)[ni].pot + (*_node_data)[vi].pot - |
|
| 1203 |
dualScale * _weight[e]; |
|
| 1204 |
|
|
| 1205 |
if (vb == blossom) {
|
|
| 1206 |
if (_delta3->state(e) == _delta3->IN_HEAP) {
|
|
| 1207 |
_delta3->erase(e); |
|
| 1208 |
} |
|
| 1209 |
} else if ((*_blossom_data)[vb].status == EVEN) {
|
|
| 1210 |
|
|
| 1211 |
if (_delta3->state(e) == _delta3->IN_HEAP) {
|
|
| 1212 |
_delta3->erase(e); |
|
| 1213 |
} |
|
| 1214 |
|
|
| 1215 |
int vt = _tree_set->find(vb); |
|
| 1216 |
|
|
| 1217 |
Arc r = _graph.oppositeArc(e); |
|
| 1218 |
|
|
| 1219 |
typename std::map<int, Arc>::iterator it = |
|
| 1220 |
(*_node_data)[ni].heap_index.find(vt); |
|
| 1221 |
|
|
| 1222 |
if (it != (*_node_data)[ni].heap_index.end()) {
|
|
| 1223 |
if ((*_node_data)[ni].heap[it->second] > rw) {
|
|
| 1224 |
(*_node_data)[ni].heap.replace(it->second, r); |
|
| 1225 |
(*_node_data)[ni].heap.decrease(r, rw); |
|
| 1226 |
it->second = r; |
|
| 1227 |
} |
|
| 1228 |
} else {
|
|
| 1229 |
(*_node_data)[ni].heap.push(r, rw); |
|
| 1230 |
(*_node_data)[ni].heap_index.insert(std::make_pair(vt, r)); |
|
| 1231 |
} |
|
| 1232 |
|
|
| 1233 |
if ((*_blossom_set)[n] > (*_node_data)[ni].heap.prio()) {
|
|
| 1234 |
_blossom_set->decrease(n, (*_node_data)[ni].heap.prio()); |
|
| 1235 |
|
|
| 1236 |
if (_delta2->state(blossom) != _delta2->IN_HEAP) {
|
|
| 1237 |
_delta2->push(blossom, _blossom_set->classPrio(blossom) - |
|
| 1238 |
(*_blossom_data)[blossom].offset); |
|
| 1239 |
} else if ((*_delta2)[blossom] > _blossom_set->classPrio(blossom)- |
|
| 1240 |
(*_blossom_data)[blossom].offset){
|
|
| 1241 |
_delta2->decrease(blossom, _blossom_set->classPrio(blossom) - |
|
| 1242 |
(*_blossom_data)[blossom].offset); |
|
| 1243 |
} |
|
| 1244 |
} |
|
| 1245 |
|
|
| 1246 |
} else if ((*_blossom_data)[vb].status == UNMATCHED) {
|
|
| 1247 |
if (_delta3->state(e) == _delta3->IN_HEAP) {
|
|
| 1248 |
_delta3->erase(e); |
|
| 1249 |
} |
|
| 1250 |
} |
|
| 1251 |
} |
|
| 1252 |
} |
|
| 1253 |
} |
|
| 1254 |
|
|
| 1255 | 1144 |
void alternatePath(int even, int tree) {
|
| 1256 | 1145 |
int odd; |
| 1257 | 1146 |
|
| ... | ... |
@@ -1294,39 +1183,42 @@ |
| 1294 | 1183 |
alternatePath(blossom, tree); |
| 1295 | 1184 |
destroyTree(tree); |
| 1296 | 1185 |
|
| 1297 |
(*_blossom_data)[blossom].status = UNMATCHED; |
|
| 1298 | 1186 |
(*_blossom_data)[blossom].base = node; |
| 1299 |
matchedToUnmatched(blossom); |
|
| 1300 |
} |
|
| 1301 |
|
|
| 1187 |
(*_blossom_data)[blossom].next = INVALID; |
|
| 1188 |
} |
|
| 1302 | 1189 |
|
| 1303 | 1190 |
void augmentOnEdge(const Edge& edge) {
|
| 1304 | 1191 |
|
| 1305 | 1192 |
int left = _blossom_set->find(_graph.u(edge)); |
| 1306 | 1193 |
int right = _blossom_set->find(_graph.v(edge)); |
| 1307 | 1194 |
|
| 1308 |
if ((*_blossom_data)[left].status == EVEN) {
|
|
| 1309 | 1195 |
int left_tree = _tree_set->find(left); |
| 1310 | 1196 |
alternatePath(left, left_tree); |
| 1311 | 1197 |
destroyTree(left_tree); |
| 1312 |
} else {
|
|
| 1313 |
(*_blossom_data)[left].status = MATCHED; |
|
| 1314 |
unmatchedToMatched(left); |
|
| 1315 |
} |
|
| 1316 |
|
|
| 1317 |
if ((*_blossom_data)[right].status == EVEN) {
|
|
| 1198 |
|
|
| 1318 | 1199 |
int right_tree = _tree_set->find(right); |
| 1319 | 1200 |
alternatePath(right, right_tree); |
| 1320 | 1201 |
destroyTree(right_tree); |
| 1321 |
} else {
|
|
| 1322 |
(*_blossom_data)[right].status = MATCHED; |
|
| 1323 |
unmatchedToMatched(right); |
|
| 1324 |
} |
|
| 1325 | 1202 |
|
| 1326 | 1203 |
(*_blossom_data)[left].next = _graph.direct(edge, true); |
| 1327 | 1204 |
(*_blossom_data)[right].next = _graph.direct(edge, false); |
| 1328 | 1205 |
} |
| 1329 | 1206 |
|
| 1207 |
void augmentOnArc(const Arc& arc) {
|
|
| 1208 |
|
|
| 1209 |
int left = _blossom_set->find(_graph.source(arc)); |
|
| 1210 |
int right = _blossom_set->find(_graph.target(arc)); |
|
| 1211 |
|
|
| 1212 |
(*_blossom_data)[left].status = MATCHED; |
|
| 1213 |
|
|
| 1214 |
int right_tree = _tree_set->find(right); |
|
| 1215 |
alternatePath(right, right_tree); |
|
| 1216 |
destroyTree(right_tree); |
|
| 1217 |
|
|
| 1218 |
(*_blossom_data)[left].next = arc; |
|
| 1219 |
(*_blossom_data)[right].next = _graph.oppositeArc(arc); |
|
| 1220 |
} |
|
| 1221 |
|
|
| 1330 | 1222 |
void extendOnArc(const Arc& arc) {
|
| 1331 | 1223 |
int base = _blossom_set->find(_graph.target(arc)); |
| 1332 | 1224 |
int tree = _tree_set->find(base); |
| ... | ... |
@@ -1629,7 +1521,7 @@ |
| 1629 | 1521 |
} |
| 1630 | 1522 |
|
| 1631 | 1523 |
for (int i = 0; i < int(blossoms.size()); ++i) {
|
| 1632 |
if ((*_blossom_data)[blossoms[i]]. |
|
| 1524 |
if ((*_blossom_data)[blossoms[i]].next != INVALID) {
|
|
| 1633 | 1525 |
|
| 1634 | 1526 |
Value offset = (*_blossom_data)[blossoms[i]].offset; |
| 1635 | 1527 |
(*_blossom_data)[blossoms[i]].pot += 2 * offset; |
| ... | ... |
@@ -1757,12 +1649,11 @@ |
| 1757 | 1649 |
Value d4 = !_delta4->empty() ? |
| 1758 | 1650 |
_delta4->prio() : std::numeric_limits<Value>::max(); |
| 1759 | 1651 |
|
| 1760 |
_delta_sum = |
|
| 1652 |
_delta_sum = d3; OpType ot = D3; |
|
| 1653 |
if (d1 < _delta_sum) { _delta_sum = d1; ot = D1; }
|
|
| 1761 | 1654 |
if (d2 < _delta_sum) { _delta_sum = d2; ot = D2; }
|
| 1762 |
if (d3 < _delta_sum) { _delta_sum = d3; ot = D3; }
|
|
| 1763 | 1655 |
if (d4 < _delta_sum) { _delta_sum = d4; ot = D4; }
|
| 1764 | 1656 |
|
| 1765 |
|
|
| 1766 | 1657 |
switch (ot) {
|
| 1767 | 1658 |
case D1: |
| 1768 | 1659 |
{
|
| ... | ... |
@@ -1775,8 +1666,13 @@ |
| 1775 | 1666 |
{
|
| 1776 | 1667 |
int blossom = _delta2->top(); |
| 1777 | 1668 |
Node n = _blossom_set->classTop(blossom); |
| 1778 |
Arc e = (*_node_data)[(*_node_index)[n]].heap.top(); |
|
| 1779 |
extendOnArc(e); |
|
| 1669 |
Arc a = (*_node_data)[(*_node_index)[n]].heap.top(); |
|
| 1670 |
if ((*_blossom_data)[blossom].next == INVALID) {
|
|
| 1671 |
augmentOnArc(a); |
|
| 1672 |
--unmatched; |
|
| 1673 |
} else {
|
|
| 1674 |
extendOnArc(a); |
|
| 1675 |
} |
|
| 1780 | 1676 |
} |
| 1781 | 1677 |
break; |
| 1782 | 1678 |
case D3: |
| ... | ... |
@@ -1789,20 +1685,8 @@ |
| 1789 | 1685 |
if (left_blossom == right_blossom) {
|
| 1790 | 1686 |
_delta3->pop(); |
| 1791 | 1687 |
} else {
|
| 1792 |
int left_tree; |
|
| 1793 |
if ((*_blossom_data)[left_blossom].status == EVEN) {
|
|
| 1794 |
left_tree = _tree_set->find(left_blossom); |
|
| 1795 |
} else {
|
|
| 1796 |
left_tree = -1; |
|
| 1797 |
++unmatched; |
|
| 1798 |
} |
|
| 1799 |
int right_tree; |
|
| 1800 |
if ((*_blossom_data)[right_blossom].status == EVEN) {
|
|
| 1801 |
right_tree = _tree_set->find(right_blossom); |
|
| 1802 |
} else {
|
|
| 1803 |
right_tree = -1; |
|
| 1804 |
++unmatched; |
|
| 1805 |
} |
|
| 1688 |
int left_tree = _tree_set->find(left_blossom); |
|
| 1689 |
int right_tree = _tree_set->find(right_blossom); |
|
| 1806 | 1690 |
|
| 1807 | 1691 |
if (left_tree == right_tree) {
|
| 1808 | 1692 |
shrinkOnEdge(e, left_tree); |
| ... | ... |
@@ -1856,7 +1740,7 @@ |
| 1856 | 1740 |
sum += _weight[(*_matching)[n]]; |
| 1857 | 1741 |
} |
| 1858 | 1742 |
} |
| 1859 |
return sum / |
|
| 1743 |
return sum / 2; |
|
| 1860 | 1744 |
} |
| 1861 | 1745 |
|
| 1862 | 1746 |
/// \brief Return the size (cardinality) of the matching. |
| ... | ... |
@@ -2233,9 +2117,6 @@ |
| 2233 | 2117 |
} |
| 2234 | 2118 |
|
| 2235 | 2119 |
void destroyStructures() {
|
| 2236 |
_node_num = countNodes(_graph); |
|
| 2237 |
_blossom_num = _node_num * 3 / 2; |
|
| 2238 |
|
|
| 2239 | 2120 |
if (_matching) {
|
| 2240 | 2121 |
delete _matching; |
| 2241 | 2122 |
} |
| ... | ... |
@@ -2991,8 +2872,8 @@ |
| 2991 | 2872 |
Value d4 = !_delta4->empty() ? |
| 2992 | 2873 |
_delta4->prio() : std::numeric_limits<Value>::max(); |
| 2993 | 2874 |
|
| 2994 |
_delta_sum = d2; OpType ot = D2; |
|
| 2995 |
if (d3 < _delta_sum) { _delta_sum = d3; ot = D3; }
|
|
| 2875 |
_delta_sum = d3; OpType ot = D3; |
|
| 2876 |
if (d2 < _delta_sum) { _delta_sum = d2; ot = D2; }
|
|
| 2996 | 2877 |
if (d4 < _delta_sum) { _delta_sum = d4; ot = D4; }
|
| 2997 | 2878 |
|
| 2998 | 2879 |
if (_delta_sum == std::numeric_limits<Value>::max()) {
|
| ... | ... |
@@ -3074,7 +2955,7 @@ |
| 3074 | 2955 |
sum += _weight[(*_matching)[n]]; |
| 3075 | 2956 |
} |
| 3076 | 2957 |
} |
| 3077 |
return sum / |
|
| 2958 |
return sum / 2; |
|
| 3078 | 2959 |
} |
| 3079 | 2960 |
|
| 3080 | 2961 |
/// \brief Return \c true if the given edge is in the matching. |
| ... | ... |
@@ -3241,4 +3122,4 @@ |
| 3241 | 3122 |
|
| 3242 | 3123 |
} //END OF NAMESPACE LEMON |
| 3243 | 3124 |
|
| 3244 |
#endif // |
|
| 3125 |
#endif //LEMON_MATCHING_H |
0 comments (0 inline)