first version
This commit is contained in:
t510599
195
cli/algo.py
Normal file
195
cli/algo.py
Normal file
@@ -0,0 +1,195 @@
|
||||
from random import randrange
|
||||
|
||||
'''
|
||||
Original from secminhr
|
||||
modified by t510599
|
||||
'''
|
||||
|
||||
# init seats
|
||||
def generateSeats():
|
||||
seats = []
|
||||
numbers = [i for i in range(1, 39)]
|
||||
for i in range(6):
|
||||
seats.append([])
|
||||
for j in range(7):
|
||||
seats[len(seats)-1].append(0)
|
||||
if i == 5:
|
||||
if 0<=j<=1 or 5<=j<= 6:
|
||||
continue
|
||||
index = randrange(0, 38)
|
||||
while(numbers[index] == 0):
|
||||
index = randrange(0, 38)
|
||||
seats[i][j] = numbers[index]
|
||||
numbers[index] = 0
|
||||
return seats
|
||||
|
||||
# print seats
|
||||
def printArr(seats):
|
||||
for i in range(len(seats)):
|
||||
for j in range(len(seats[0])):
|
||||
num = seats[i][j]
|
||||
if 0 <= num <= 9:
|
||||
num = '0' + str(num)
|
||||
print(num, end=" ")
|
||||
print()
|
||||
print() # newline
|
||||
|
||||
def _arrShiftLeft(index, arr, num):
|
||||
element = arr[index]
|
||||
zero_index = []
|
||||
while 0 in arr:
|
||||
index = arr.index(0)
|
||||
zero_index += [index]
|
||||
del arr[index]
|
||||
zero_index = zero_index[::-1]
|
||||
index = arr.index(element)
|
||||
if num > index:
|
||||
num = index
|
||||
front_list = arr[:num]
|
||||
arr = arr[num:len(arr)] + front_list
|
||||
|
||||
for i in zero_index:
|
||||
arr.insert(i, 0)
|
||||
return arr
|
||||
|
||||
def _arrShiftRight(index, arr, num):
|
||||
element = arr[index]
|
||||
zero_index = []
|
||||
while 0 in arr:
|
||||
index = arr.index(0)
|
||||
zero_index += [index]
|
||||
del arr[index]
|
||||
zero_index = zero_index[::-1]
|
||||
index = arr.index(element)
|
||||
if num > len(arr) - index - 1:
|
||||
num = len(arr) - index - 1
|
||||
later_list = arr[len(arr) - num:len(arr)]
|
||||
arr = later_list + arr[:len(arr)-num]
|
||||
for i in zero_index:
|
||||
arr.insert(i, 0)
|
||||
return arr
|
||||
|
||||
|
||||
######### Actions ############
|
||||
|
||||
def shiftRight(row, col, num, seats):
|
||||
row_list = seats[row - 1]
|
||||
row_list = _arrShiftRight(col - 1, row_list, num)
|
||||
seats[row - 1] = row_list
|
||||
return seats
|
||||
|
||||
def shiftLeft(row, col, num, seats):
|
||||
row_list = seats[row - 1]
|
||||
row_list = _arrShiftLeft(col - 1, row_list, num)
|
||||
seats[row - 1] = row_list
|
||||
return seats
|
||||
|
||||
def shiftUp(row, col, num, seats):
|
||||
col_list = [seats[i][col - 1] for i in range(len(seats))]
|
||||
col_list = _arrShiftLeft(row - 1 ,col_list, num)
|
||||
for i in range(len(seats)):
|
||||
seats[i][col - 1] = col_list[i]
|
||||
return seats
|
||||
|
||||
def shiftDown(row, col, num, seats):
|
||||
col_list = [seats[i][col - 1] for i in range(len(seats))]
|
||||
col_list = _arrShiftRight(row - 1 ,col_list, num)
|
||||
for i in range(len(seats)):
|
||||
seats[i][col - 1] = col_list[i]
|
||||
return seats
|
||||
|
||||
# get list of element from left bottom to right top
|
||||
def _getUpperCross(row, col, arr):
|
||||
target_list = []
|
||||
current_row = row
|
||||
current_col = col
|
||||
highest_pos = (0, 0)
|
||||
#get all right up corner
|
||||
while current_row >= 1 and current_col <= len(arr[0]):
|
||||
target_list.insert(0, arr[current_row-1][current_col-1])
|
||||
current_row -= 1
|
||||
current_col += 1
|
||||
highest_pos = (current_row + 1, current_col - 1)
|
||||
current_row = row + 1
|
||||
current_col = col - 1
|
||||
#get all left bottom corner
|
||||
while current_row <= len(arr) and current_col >= 1:
|
||||
target_list.append(arr[current_row-1][current_col-1])
|
||||
current_row += 1
|
||||
current_col -= 1
|
||||
return (target_list, highest_pos)
|
||||
|
||||
# get list of element from left top to right bottom
|
||||
def _getLowerCross(row, col, arr):
|
||||
target_list = []
|
||||
current_row = row
|
||||
current_col = col
|
||||
highest_pos = (0, 0)
|
||||
#get all left up corner
|
||||
while current_row >= 1 and current_col >= 1:
|
||||
target_list.insert(0, arr[current_row-1][current_col-1])
|
||||
current_row -= 1
|
||||
current_col -= 1
|
||||
highest_pos = (current_row + 1, current_col + 1)
|
||||
current_row = row + 1
|
||||
current_col = col + 1
|
||||
#get all right down corner
|
||||
while current_row <= len(arr) and current_col <= len(arr[0]):
|
||||
target_list.append(arr[current_row-1][current_col-1])
|
||||
current_row += 1
|
||||
current_col += 1
|
||||
return (target_list, highest_pos)
|
||||
|
||||
def shiftRightUp(row, col, num, seats):
|
||||
# the list is follows the predicate: n.row > (n+1).row
|
||||
corner_list, highest_pos = _getUpperCross(row, col, seats)
|
||||
corner_list = _arrShiftLeft(row - highest_pos[0] ,corner_list, num)
|
||||
#fill in
|
||||
current_row = highest_pos[0]
|
||||
current_col = highest_pos[1]
|
||||
while corner_list:
|
||||
seats[current_row-1][current_col-1] = corner_list[0]
|
||||
del corner_list[0]
|
||||
current_row += 1
|
||||
current_col -= 1
|
||||
return seats
|
||||
|
||||
def shiftRightDown(row, col, num, seats):
|
||||
# the list is follows the predicate: n.row > (n+1).row
|
||||
corner_list, highest_pos = _getLowerCross(row, col, seats)
|
||||
corner_list = _arrShiftRight(row - highest_pos[0] ,corner_list, num)
|
||||
#fill in
|
||||
current_row = highest_pos[0]
|
||||
current_col = highest_pos[1]
|
||||
while corner_list:
|
||||
seats[current_row-1][current_col-1] = corner_list[0]
|
||||
del corner_list[0]
|
||||
current_row += 1
|
||||
current_col += 1
|
||||
return seats
|
||||
|
||||
def shiftLeftUp(row, col, num, seats):
|
||||
corner_list, highest_pos = _getLowerCross(row, col, seats)
|
||||
corner_list = _arrShiftLeft(highest_pos[0] - row ,corner_list, num)
|
||||
#fill in
|
||||
current_row = highest_pos[0]
|
||||
current_col = highest_pos[1]
|
||||
while corner_list:
|
||||
seats[current_row-1][current_col-1] = corner_list[0]
|
||||
del corner_list[0]
|
||||
current_row += 1
|
||||
current_col += 1
|
||||
return seats
|
||||
|
||||
def shiftLeftDown(row, col, num, seats):
|
||||
corner_list, highest_pos = _getUpperCross(row, col, seats)
|
||||
corner_list = _arrShiftRight(row - highest_pos[0] ,corner_list, num)
|
||||
#fill in
|
||||
current_row = highest_pos[0]
|
||||
current_col = highest_pos[1]
|
||||
while corner_list:
|
||||
seats[current_row-1][current_col-1] = corner_list[0]
|
||||
del corner_list[0]
|
||||
current_row += 1
|
||||
current_col -= 1
|
||||
return seats
|
||||
83
cli/seat.py
Normal file
83
cli/seat.py
Normal file
@@ -0,0 +1,83 @@
|
||||
from algo import generateSeats, printArr, shiftUp, shiftDown, shiftLeft, shiftRight, shiftLeftDown, shiftLeftUp, shiftRightDown, shiftRightUp
|
||||
from random import choice
|
||||
import sys
|
||||
import json
|
||||
|
||||
def load():
|
||||
f = open('seats.txt','r',encoding="utf-8")
|
||||
seats = json.load(f)
|
||||
f.close()
|
||||
return seats
|
||||
|
||||
def parse(seats,requirements):
|
||||
funcs = {"f": shiftUp,"b": shiftDown, "l": shiftLeft, "r": shiftRight, "fr": shiftRightUp, "br": shiftRightDown, "fl": shiftLeftUp, "bl": shiftLeftDown} # f: forward, b: backward, l:left, r: right
|
||||
name = {"f": "向前","b": "向後", "l": "向左", "r": "向右", "fr": "向右前", "br": "向右後", "fl": "向左前", "bl": "向左後"}
|
||||
order = sorted([int(n) for n in requirements.keys()])
|
||||
luckier = choice(order)
|
||||
print("幸運兒: " + luckier + "\n")
|
||||
order = order[order.index(luckier):] + order[:order.index(luckier)]
|
||||
|
||||
for req in order:
|
||||
no,direction,steps = requirements[str(req)]
|
||||
no = no
|
||||
steps = steps
|
||||
row,col = find(seats,no)
|
||||
if col != None and row != None:
|
||||
if direction in funcs.keys():
|
||||
funcs[direction](row,col,steps,seats)
|
||||
print(no, name[direction], steps)
|
||||
printArr(seats)
|
||||
print()
|
||||
return seats
|
||||
|
||||
def find(seats,no):
|
||||
for y in range(len(seats)):
|
||||
if no in seats[y]:
|
||||
col = seats[y].index(no)
|
||||
row = y
|
||||
return (row+1,col+1) # start from 0 -> start from 1
|
||||
return (None,None)
|
||||
|
||||
help_msg = '''Please provide mode!
|
||||
generate - generate a random seat
|
||||
run - apply requirements
|
||||
print - print the current seat
|
||||
debug - dry run'''
|
||||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) <= 1 or sys.argv[1] == "help":
|
||||
print(help_msg)
|
||||
sys.exit(1 if len(sys.argv) <= 1 else 0)
|
||||
else:
|
||||
mode = sys.argv[1]
|
||||
if mode == "generate":
|
||||
seats = generateSeats()
|
||||
printArr(seats)
|
||||
n = open('seats.txt','w',encoding="utf-8")
|
||||
n.write(str(seats))
|
||||
n.close()
|
||||
print("Done!")
|
||||
elif mode == "run" or mode == "debug":
|
||||
seats = load()
|
||||
print("Original:")
|
||||
printArr(seats)
|
||||
#load requirements
|
||||
requirements = dict()
|
||||
r = open('requirements.txt','r',encoding="utf-8")
|
||||
for line in r.readlines():
|
||||
no,direction,steps = line.strip().split(" ")
|
||||
requirements[no] = (int(no),direction,int(steps))
|
||||
parse(seats,requirements)
|
||||
print("Result:")
|
||||
printArr(seats)
|
||||
if mode == "run":
|
||||
#save
|
||||
n = open('seats.txt','w',encoding="utf-8")
|
||||
n.write(str(seats))
|
||||
n.close()
|
||||
print("Done!")
|
||||
elif mode == "show":
|
||||
printArr(load())
|
||||
else:
|
||||
print(help_msg)
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user