import random import sys from typing import List, Union # Global variables colors = ["BLACK", "WHITE", "RED", "GREEN", "ORANGE", "YELLOW", "PURPLE", "TAN"] color_letters = "BWRGOYPT" num_positions = 0 num_colors: int = 100 human_score = 0 computer_score = 0 def main() -> None: global colors, color_letters, num_positions, num_colors, human_score, computer_score num_colors = 100 human_score = 0 computer_score = 0 # get user inputs for game conditions print("Mastermind") print("Creative Computing Morristown, New Jersey") while num_colors > 8: num_colors = int(input("Number of colors (max 8): ")) # C9 in BASIC num_positions = int(input("Number of positions: ")) # P9 in BASIC num_rounds = int(input("Number of rounds: ")) # R9 in BASIC possibilities = num_colors**num_positions print(f"Number of possibilities {possibilities}") print("Color\tLetter") print("=====\t======") for element in range(0, num_colors): print(f"{colors[element]}\t{colors[element][0]}") current_round = 1 while current_round <= num_rounds: print(f"Round number {current_round}") num_moves = 1 guesses: List[List[Union[str, int]]] = [] turn_over = False print("Guess my combination ...") secret_combination = int(possibilities * random.random()) answer = possibility_to_color_code(secret_combination) while num_moves < 10 and not turn_over: print(f"Move # {num_moves} Guess : ") user_command = input("Guess ") if user_command == "BOARD": print_board(guesses) # 2000 elif user_command == "QUIT": # 2500 print(f"QUITTER! MY COMBINATION WAS: {answer}") print("GOOD BYE") quit() elif len(user_command) != num_positions: # 410 print("BAD NUMBER OF POSITIONS") else: invalid_letters = get_invalid_letters(user_command) if invalid_letters > "": print(f"INVALID GUESS: {invalid_letters}") else: guess_results = compare_two_positions(user_command, answer) if guess_results[1] == num_positions: # correct guess turn_over = True print(f"You guessed it in {num_moves} moves!") human_score = human_score + num_moves print_score() else: print( "You have {} blacks and {} whites".format( guess_results[1], guess_results[2] ) ) num_moves = num_moves + 1 guesses.append(guess_results) if not turn_over: # RAN OUT OF MOVES print("YOU RAN OUT OF MOVES! THAT'S ALL YOU GET!") print(f"THE ACTUAL COMBINATION WAS: {answer}") human_score = human_score + num_moves print_score() # COMPUTER TURN turn_over = False inconsistent_information = False while not turn_over and not inconsistent_information: all_possibilities = [1] * possibilities num_moves = 1 inconsistent_information = False print("NOW I GUESS. THINK OF A COMBINATION.") input("HIT RETURN WHEN READY: ") while num_moves < 10 and not turn_over and not inconsistent_information: found_guess = False possible_guess = int(possibilities * random.random()) if ( all_possibilities[possible_guess] == 1 ): # random guess is possible, use it found_guess = True else: for i in range(possible_guess + 1, possibilities): if all_possibilities[i] == 1: found_guess = True possible_guess = i break if not found_guess: for i in range(0, possible_guess): if all_possibilities[i] == 1: found_guess = True possible_guess = i break if not found_guess: # inconsistent info from user print("YOU HAVE GIVEN ME INCONSISTENT INFORMATION.") print("TRY AGAIN, AND THIS TIME PLEASE BE MORE CAREFUL.") turn_over = True inconsistent_information = True else: computer_guess = possibility_to_color_code(possible_guess) print(f"My guess is: {computer_guess}") blacks_str, whites_str = input( "ENTER BLACKS, WHITES (e.g. 1,2): " ).split(",") blacks = int(blacks_str) whites = int(whites_str) if blacks == num_positions: # Correct guess print(f"I GOT IT IN {num_moves} MOVES") turn_over = True computer_score = computer_score + num_moves print_score() else: num_moves += 1 for i in range(0, possibilities): if all_possibilities[i] == 0: # already ruled out continue possible_answer = possibility_to_color_code(i) comparison = compare_two_positions( possible_answer, computer_guess ) if (blacks != comparison[1]) or (whites != comparison[2]): all_possibilities[i] = 0 if not turn_over: # COMPUTER DID NOT GUESS print("I USED UP ALL MY MOVES!") print("I GUESS MY CPU IS JUST HAVING AN OFF DAY.") computer_score = computer_score + num_moves print_score() current_round += 1 print_score(is_final_score=True) sys.exit() # 470 def get_invalid_letters(user_command) -> str: """Makes sure player input consists of valid colors for selected game configuration.""" valid_colors = color_letters[:num_colors] invalid_letters = "" for letter in user_command: if letter not in valid_colors: invalid_letters = invalid_letters + letter return invalid_letters # 2000 def print_board(guesses) -> None: """Print previous guesses within the round.""" print("Board") print("Move\tGuess\tBlack White") for idx, guess in enumerate(guesses): print(f"{idx + 1}\t{guess[0]}\t{guess[1]} {guess[2]}") # Accepts a (decimal) number representing one permutation in the realm of possible # secret codes and returns the color code mapped to that permutation. # This algorithm is essentially converting a decimal number to a number with a # base of #num_colors, where each color code letter represents a digit in # that #num_colors base. def possibility_to_color_code(possibility: int) -> str: color_code: str = "" pos: int = num_colors ** num_positions # start with total possibilities remainder = possibility for i in range(num_positions-1, 0, -1): # process all but the last digit pos = pos // num_colors color_code += color_letters[remainder // pos] remainder = remainder % pos color_code += color_letters[remainder] # last digit is what remains return color_code # 4500 def compare_two_positions(guess: str, answer: str) -> List[Union[str, int]]: """Returns blacks (correct color and position) and whites (correct color only) for candidate position (guess) versus reference position (answer).""" increment = 0 blacks = 0 whites = 0 initial_guess = guess for pos in range(0, num_positions): if guess[pos] != answer[pos]: for pos2 in range(0, num_positions): if not ( guess[pos] != answer[pos2] or guess[pos2] == answer[pos2] ): # correct color but not correct place whites = whites + 1 answer = answer[:pos2] + chr(increment) + answer[pos2 + 1:] guess = guess[:pos] + chr(increment + 1) + guess[pos + 1:] increment = increment + 2 else: # correct color and placement blacks = blacks + 1 # THIS IS DEVIOUSLY CLEVER guess = guess[:pos] + chr(increment + 1) + guess[pos + 1:] answer = answer[:pos] + chr(increment) + answer[pos + 1:] increment = increment + 2 return [initial_guess, blacks, whites] # 5000 + logic from 1160 def print_score(is_final_score: bool = False) -> None: """Print score after each turn ends, including final score at end of game.""" if is_final_score: print("GAME OVER") print("FINAL SCORE:") else: print("SCORE:") print(f" COMPUTER {computer_score}") print(f" HUMAN {human_score}") if __name__ == "__main__": main()