Skip to content

Instantly share code, notes, and snippets.

@ejcer
Last active August 29, 2015 14:21
Show Gist options
  • Save ejcer/5d8edd64b09488035f77 to your computer and use it in GitHub Desktop.
Save ejcer/5d8edd64b09488035f77 to your computer and use it in GitHub Desktop.
import queue
class Position(object):
def __init__(self, row, column):
self.row = row
self.column = column
def createState(stateMatrix, position):
newMatrix = list()
newMatrix = [stateMatrix[row] for row in range(len(stateMatrix))]
newMatrix[position.row][position.column] = 'X'
return State(newMatrix, position)
class State(object):
def __init__(self, matrix, curPosition):
self.curMatrix = matrix
self.curPosition = curPosition
def __hash__(self):
hash = 0
mult = 1
for row in range(len(self.curMatrix)):
for col in range(len(self.curMatrix[0])):
if(self.curMatrix[row][col] == 'X'):
hash += mult * 1
mult*=2
return hash
def __eq__(self, other):
for i, row in enumerate(other.curMatrix):
for j, col in enumerate(other.curMatrix[0]):
if not(self.curMatrix[i][j] == other.curMatrix[i][j]):
return False
return True
#adjacentStates
def getAdjacentStates(self):
#calculate a new matrix from curMatrix based on horizontal and vertical position movements
#curMatrix & curPosition
adjStates = list()
if(self.curPosition.column > 0):
# Subtract one from the column of the current position
leftPos = Position(self.curPosition.row, self.curPosition.column - 1)
adjStates.append(createState(self.curMatrix, leftPos))
if(self.curPosition.column < len(self.curMatrix[0]) - 1):
# Add one to the column of the current position
rightPos = Position(self.curPosition.row, self.curPosition.column + 1)
adjStates.append(createState(self.curMatrix, rightPos))
if(self.curPosition.row > 0):
# Subtract one from the current position's row
downPos = Position(self.curPosition.row-1, self.curPosition.column)
adjStates.append(createState(self.curMatrix, downPos))
if(self.curPosition.row < len(self.curMatrix) - 1):
# Add one to the row of the current position
upPos = Position(self.curPosition.row + 1, self.curPosition.column)
adjStates.append(createState(self.curMatrix, upPos))
#print()
return adjStates
#isFinalState
def __repr__(self):
for i, row in enumerate(self.curMatrix):
for j, col in enumerate(self.curMatrix[i]):
print(self.curMatrix[i][j], end='')
print()
n, m = list(map(int, input().split()))
startMatrix = [list(input()) for x in range(n)] #startMatrix[row][column]
startRow, startColumn = list(map(int, input().split()))
startPos = Position(startRow-1, startColumn-1)
endRow, endColumn = list(map(int, input().split()))
endPos = Position(endRow-1, endColumn-1)
initialState = createState(startMatrix, startPos)
testAdj = initialState.getAdjacentStates()
for adj in testAdj:
adj.__repr__()
print()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment