40 lines
2.0 KiB
C
40 lines
2.0 KiB
C
|
///
|
||
|
|
// Hungarian.h: Header file for Class HungarianAlgorithm.
|
||
|
|
//
|
||
|
|
// This is a C++ wrapper with slight modification of a hungarian algorithm implementation by Markus Buehren.
|
||
|
|
// The original implementation is a few mex-functions for use in MATLAB, found here:
|
||
|
|
// http://www.mathworks.com/matlabcentral/fileexchange/6543-functions-for-the-rectangular-assignment-problem
|
||
|
|
//
|
||
|
|
// Both this code and the orignal code are published under the BSD license.
|
||
|
|
// by Cong Ma, 2016
|
||
|
|
//
|
||
|
|
///
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
#ifndef HUNGARIAN_H
|
||
|
|
#define HUNGARIAN_H
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
|
||
|
|
class HungarianAlgorithm
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
HungarianAlgorithm();
|
||
|
|
~HungarianAlgorithm();
|
||
|
|
float Solve(std::vector<std::vector<float> >& DistMatrix, std::vector<int>& Assignment);
|
||
|
|
|
||
|
|
private:
|
||
|
|
void assignmentoptimal(int* assignment, float* cost, float* distMatrix, int nOfRows, int nOfColumns);
|
||
|
|
void buildassignmentvector(int* assignment, bool* starMatrix, int nOfRows, int nOfColumns);
|
||
|
|
void computeassignmentcost(int* assignment, float* cost, float* distMatrix, int nOfRows);
|
||
|
|
void step2a(int* assignment, float* distMatrix, bool* starMatrix, bool* newStarMatrix, bool* primeMatrix, bool* coveredColumns, bool* coveredRows, int nOfRows, int nOfColumns, int minDim);
|
||
|
|
void step2b(int* assignment, float* distMatrix, bool* starMatrix, bool* newStarMatrix, bool* primeMatrix, bool* coveredColumns, bool* coveredRows, int nOfRows, int nOfColumns, int minDim);
|
||
|
|
void step3(int* assignment, float* distMatrix, bool* starMatrix, bool* newStarMatrix, bool* primeMatrix, bool* coveredColumns, bool* coveredRows, int nOfRows, int nOfColumns, int minDim);
|
||
|
|
void step4(int* assignment, float* distMatrix, bool* starMatrix, bool* newStarMatrix, bool* primeMatrix, bool* coveredColumns, bool* coveredRows, int nOfRows, int nOfColumns, int minDim, int row, int col);
|
||
|
|
void step5(int* assignment, float* distMatrix, bool* starMatrix, bool* newStarMatrix, bool* primeMatrix, bool* coveredColumns, bool* coveredRows, int nOfRows, int nOfColumns, int minDim);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
#endif
|