assignmentgraph.h

来自「mersad源码 03年robocup 季军 可以研究一下大家」· C头文件 代码 · 共 56 行

/*
 *  Copyright 2002-2004, Mersad Team, Allame Helli High School (NODET).
 *
 *  This program is free software, you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU Library General Public License for more details.
 *
 *  Created by: Mostafa Rokooey
 *  Released on Friday 1 April 2005 by Mersad RoboCup Team.
 *  For more information please read README file.
*/

#ifndef __ASSIGNMENT_GRAPH_H
#define __ASSIGNMENT_GRAPH_H


#define BIG 1000000


class AssignmentGraph
{
private:

	int dim;
	int **weights;
	int *answerRow;
	int *answerCol;
	int *u;
	int *v;
	int answerCost;
	bool maximumFlag;
	
public:
	AssignmentGraph(int size);
	~AssignmentGraph();

	void resetWeights();
	void setWeight(int row, int col, int weight);
	int getWeight(int row,int col);
	
	void prepareMaximum();
	void prepareMinimum();

	void solve();

	int getAnswerRow(int row);
	int getAnswerCol(int col);
};

#endif // __ASSIGNMENT_GRAPH_H