picostmatrix.java

来自「用于求解TSP(Traveling salesman problem」· Java 代码 · 共 72 行

/**
 * The Minimum1TreeCost function returns the cost of a minimum 1-tree.
 *
 * @ Author        Create/Modi     Note
 * Xiaofeng Xie    May 21, 2005
 * Xiaofeng Xie    Apr 28, 2006    MAOS-TSP Beta 1.1.002
 *
 * @ Reference:
 * [1] K. Helsgaun,"An Effective Implementation of the Lin-Kernighan Traveling
 *  Salesman Heuristic", European J Operational Research 126 (1), 106-130 (2000).
 */

package implement.TSP.knowledge;

import Global.basic.data.matrix.*;
import Global.math.*;

public class PiCostMatrix implements ISquareIMatrixEngine {
  private int precision = 100;
  private ISquareIMatrixEngine sdMatrix;
  private int[] piArray;
  int[][] piMatrix;

  public PiCostMatrix(ISquareIMatrixEngine sdMatrix) {
    this.sdMatrix = sdMatrix;
    int nodeNumber = sdMatrix.getNodeNumber();
    piMatrix = new int[nodeNumber][nodeNumber];
  }

  public void setPrecision(int precision) {
    this.precision = precision;
  }

  public ISquareIMatrixEngine getCostMatrix() {
    if (piArray==null) return sdMatrix;
    return new FullSquareIMatrix(piMatrix);
  }

  public int getPrecision() {
    return precision;
  }

  public void setPIArray(double[] extPIArray) {
    int nodeNumber = sdMatrix.getNodeNumber();
    piArray = new int[nodeNumber];
    for(int i=0; i<nodeNumber; i++) {
      piArray[i] = (int) (extPIArray[i]*precision);
    }
    for (int i=0; i<nodeNumber; i++) {
      for (int j=0; j<nodeNumber; j++) {
        piMatrix[i][j] = precision*sdMatrix.getValueAt(i, j) + piArray[i] + piArray[j];
      }
    }
  }

  public int getNodeNumber() {
    return sdMatrix.getNodeNumber();
  }

  public int getPenaltyValue() {
    if (piArray==null) return 0;
    return 2*ArrayMath.totalSum(piArray);
  }

  public int getValueAt(int i, int j) {
    if (piArray==null) return precision*sdMatrix.getValueAt(i, j);
    return piMatrix[i][j];
  }

}