Ticket #604: 3d9bcf07f7e5.patch
File 3d9bcf07f7e5.patch, 7.6 KB (added by , 6 years ago) |
---|
-
lemon/matching.h
# HG changeset patch # User Balazs Dezso <deba@google.com> # Date 1540638048 -7200 # Sat Oct 27 13:00:48 2018 +0200 # Node ID 3d9bcf07f7e527e8efbf348576a5b504ac54dd03 # Parent 959d78f3fe0e5358e3ae072c0885f0d3f204cfe5 Paremeter to stop matching calculation when only single node is unmatched diff -r 959d78f3fe0e -r 3d9bcf07f7e5 lemon/matching.h
a b 115 115 NodeQueue _node_queue; 116 116 int _process, _postpone, _last; 117 117 118 int _node_num ;118 int _node_num, _unmatched; 119 119 120 120 private: 121 121 … … 179 179 extendOnArc(a); 180 180 } else if ((*_status)[v] == UNMATCHED) { 181 181 augmentOnArc(a); 182 --_unmatched; 182 183 return; 183 184 } 184 185 } … … 204 205 extendOnArc(b); 205 206 } else if ((*_status)[x] == UNMATCHED) { 206 207 augmentOnArc(b); 208 --_unmatched; 207 209 return; 208 210 } 209 211 } … … 228 230 extendOnArc(a); 229 231 } else if ((*_status)[v] == UNMATCHED) { 230 232 augmentOnArc(a); 233 --_unmatched; 231 234 return; 232 235 } 233 236 } … … 437 440 (*_matching)[n] = INVALID; 438 441 (*_status)[n] = UNMATCHED; 439 442 } 443 _unmatched = _node_num; 440 444 } 441 445 442 446 /// \brief Find an initial matching in a greedy way. … … 448 452 (*_matching)[n] = INVALID; 449 453 (*_status)[n] = UNMATCHED; 450 454 } 455 _unmatched = _node_num; 451 456 for (NodeIt n(_graph); n != INVALID; ++n) { 452 457 if ((*_matching)[n] == INVALID) { 453 458 for (OutArcIt a(_graph, n); a != INVALID ; ++a) { … … 457 462 (*_status)[n] = MATCHED; 458 463 (*_matching)[v] = _graph.oppositeArc(a); 459 464 (*_status)[v] = MATCHED; 465 _unmatched -= 2; 460 466 break; 461 467 } 462 468 } … … 479 485 (*_matching)[n] = INVALID; 480 486 (*_status)[n] = UNMATCHED; 481 487 } 488 _unmatched = _node_num; 482 489 for(EdgeIt e(_graph); e!=INVALID; ++e) { 483 490 if (matching[e]) { 484 491 … … 491 498 if ((*_matching)[v] != INVALID) return false; 492 499 (*_matching)[v] = _graph.direct(e, false); 493 500 (*_status)[v] = MATCHED; 501 502 _unmatched -= 2; 494 503 } 495 504 } 496 505 return true; … … 498 507 499 508 /// \brief Start Edmonds' algorithm 500 509 /// 501 /// This function runs the original Edmonds' algorithm. 510 /// This function runs the original Edmonds' algorithm. If the 511 /// \c decomposition parameter is set to false, then the Gallai-Edmonds 512 /// decomposition is not computed. 502 513 /// 503 514 /// \pre \ref init(), \ref greedyInit() or \ref matchingInit() must be 504 515 /// called before using this function. 505 void startSparse() { 506 for(NodeIt n(_graph); n != INVALID; ++n) { 516 void startSparse(bool decomposition = true) { 517 int _unmatched_limit = decomposition ? 0 : 1; 518 for(NodeIt n(_graph); _unmatched > _unmatched_limit; ++n) { 507 519 if ((*_status)[n] == UNMATCHED) { 508 520 (*_blossom_rep)[_blossom_set->insert(n)] = n; 509 521 _tree_set->insert(n); 510 522 (*_status)[n] = EVEN; 523 --_unmatched; 511 524 processSparse(n); 512 525 } 513 526 } … … 517 530 /// for dense graphs 518 531 /// 519 532 /// This function runs Edmonds' algorithm with a heuristic of postponing 520 /// shrinks, therefore resulting in a faster algorithm for dense graphs. 533 /// shrinks, therefore resulting in a faster algorithm for dense graphs. If 534 /// the \c decomposition parameter is set to false, then the Gallai-Edmonds 535 /// decomposition is not computed. 521 536 /// 522 537 /// \pre \ref init(), \ref greedyInit() or \ref matchingInit() must be 523 538 /// called before using this function. 524 void startDense() { 525 for(NodeIt n(_graph); n != INVALID; ++n) { 539 void startDense(bool decomposition = true) { 540 int _unmatched_limit = decomposition ? 0 : 1; 541 for(NodeIt n(_graph); _unmatched > _unmatched_limit; ++n) { 526 542 if ((*_status)[n] == UNMATCHED) { 527 543 (*_blossom_rep)[_blossom_set->insert(n)] = n; 528 544 _tree_set->insert(n); 529 545 (*_status)[n] = EVEN; 546 --_unmatched; 530 547 processDense(n); 531 548 } 532 549 } … … 536 553 /// \brief Run Edmonds' algorithm 537 554 /// 538 555 /// This function runs Edmonds' algorithm. An additional heuristic of 539 /// postponing shrinks is used for relatively dense graphs 540 /// (for which <tt>m>=2*n</tt> holds). 541 void run() { 556 /// postponing shrinks is used for relatively dense graphs (for which 557 /// <tt>m>=2*n</tt> holds). If the \c decomposition parameter is set to 558 /// false, then the Gallai-Edmonds decomposition is not computed. 559 void run(bool decomposition = true) { 542 560 if (countEdges(_graph) < 2 * countNodes(_graph)) { 543 561 greedyInit(); 544 startSparse( );562 startSparse(decomposition); 545 563 } else { 546 564 init(); 547 startDense( );565 startDense(decomposition); 548 566 } 549 567 } 550 568 -
test/matching_test.cc
diff -r 959d78f3fe0e -r 3d9bcf07f7e5 test/matching_test.cc
a b 134 134 mat_test.startSparse(); 135 135 mat_test.startDense(); 136 136 mat_test.run(); 137 mat_test.startSparse(false); 138 mat_test.startDense(false); 139 mat_test.run(false); 137 140 138 141 const_mat_test.matchingSize(); 139 142 const_mat_test.matching(e); … … 402 405 graphReader(graph, lgfs). 403 406 edgeMap("weight", weight).run(); 404 407 408 int size; 405 409 bool perfect; 406 410 { 407 411 MaxMatching<SmartGraph> mm(graph); 408 412 mm.run(); 409 413 checkMatching(graph, mm); 414 size = mm.matchingSize(); 410 415 perfect = 2 * mm.matchingSize() == countNodes(graph); 411 416 } 412 417 413 418 { 419 MaxMatching<SmartGraph> mm(graph); 420 mm.init(); 421 mm.startSparse(); 422 checkMatching(graph, mm); 423 check(size == mm.matchingSize(), "Inconsistent matching size"); 424 } 425 426 { 427 MaxMatching<SmartGraph> mm(graph); 428 mm.init(); 429 mm.startDense(); 430 checkMatching(graph, mm); 431 check(size == mm.matchingSize(), "Inconsistent matching size"); 432 } 433 434 { 435 MaxMatching<SmartGraph> mm(graph); 436 mm.greedyInit(); 437 mm.startSparse(); 438 checkMatching(graph, mm); 439 check(size == mm.matchingSize(), "Inconsistent matching size"); 440 } 441 442 { 443 MaxMatching<SmartGraph> mm(graph); 444 mm.greedyInit(); 445 mm.startDense(); 446 checkMatching(graph, mm); 447 check(size == mm.matchingSize(), "Inconsistent matching size"); 448 } 449 450 { 451 MaxMatching<SmartGraph> mm(graph); 452 mm.run(false); 453 check(size == mm.matchingSize(), "Inconsistent max cardinality matching"); 454 } 455 456 { 457 MaxMatching<SmartGraph> mm(graph); 458 mm.init(); 459 mm.startSparse(false); 460 check(size == mm.matchingSize(), "Inconsistent matching size"); 461 } 462 463 { 464 MaxMatching<SmartGraph> mm(graph); 465 mm.init(); 466 mm.startDense(false); 467 check(size == mm.matchingSize(), "Inconsistent matching size"); 468 } 469 470 { 471 MaxMatching<SmartGraph> mm(graph); 472 mm.greedyInit(); 473 mm.startSparse(false); 474 check(size == mm.matchingSize(), "Inconsistent matching size"); 475 } 476 477 { 478 MaxMatching<SmartGraph> mm(graph); 479 mm.greedyInit(); 480 mm.startDense(false); 481 check(size == mm.matchingSize(), "Inconsistent matching size"); 482 } 483 484 { 414 485 MaxWeightedMatching<SmartGraph> mwm(graph, weight); 415 486 mwm.run(); 416 487 checkWeightedMatching(graph, weight, mwm);