diff --git a/other/snake_water_gun.py b/other/snake_water_gun.py new file mode 100644 index 000000000000..5476e2988d9c --- /dev/null +++ b/other/snake_water_gun.py @@ -0,0 +1,72 @@ +""" +A simple implementation of the Snake, Water, Gun game. +""" + +import doctest +import random + + +def snake_water_gun(player_choice: str, computer_choice: str) -> str: + """ + Determines the winner of a Snake, Water, Gun game round. + + Args: + player_choice: The player's choice ('s' for snake, 'w' for water, 'g' for gun). + computer_choice: The computer's choice. + + Returns: + Result: "Player wins!", "Computer wins!", or "It's a draw!". + + Doctests: + >>> snake_water_gun('s', 'w') + 'Player wins!' + >>> snake_water_gun('w', 'g') + 'Player wins!' + >>> snake_water_gun('g', 's') + 'Player wins!' + >>> snake_water_gun('w', 's') + 'Computer wins!' + >>> snake_water_gun('s', 's') + "It's a draw!" + """ + if player_choice == computer_choice: + return "It's a draw!" + + if ( + (player_choice == "s" and computer_choice == "w") + or (player_choice == "w" and computer_choice == "g") + or (player_choice == "g" and computer_choice == "s") + ): + return "Player wins!" + else: + return "Computer wins!" + + +def main(): + """ + Main function to run the Snake, Water, Gun game. + """ + print("--- Snake, Water, Gun Game ---") + player_input = ( + input("Enter your choice (s for snake, w for water, g for gun): ") + .lower() + .strip() + ) + + if player_input not in ["s", "w", "g"]: + print("Invalid choice. Please choose 's', 'w', or 'g'.") + return + + choices = ["s", "w", "g"] + computer_input = random.choice(choices) + + print(f"\nYou chose: {player_input}") + print(f"Computer chose: {computer_input}\n") + + result = snake_water_gun(player_input, computer_input) + print(result) + + +if __name__ == "__main__": + doctest.testmod() # Run the doctests + main() diff --git a/strings/min_cost_string_conversion.py b/strings/min_cost_string_conversion.py index 87eb5189e16a..1915514747e9 100644 --- a/strings/min_cost_string_conversion.py +++ b/strings/min_cost_string_conversion.py @@ -1,13 +1,48 @@ """ Algorithm for calculating the most cost-efficient sequence for converting one string -into another. -The only allowed operations are ---- Cost to copy a character is copy_cost ---- Cost to replace a character is replace_cost ---- Cost to delete a character is delete_cost ---- Cost to insert a character is insert_cost +into another (Levenshtein distance). +The allowed operations are insertion, deletion, or substitution of a single character. """ +import doctest + + +def min_cost_string_conversion(str1: str, str2: str) -> int: + """ + Calculates the minimum cost (standard Levenshtein distance) to convert str1 to str2. + The cost of insertion, deletion, and substitution is 1. The cost of a copy is 0. + + This is a user-friendly wrapper around the more detailed compute_transform_tables. + + Args: + str1: The source string. + str2: The target string. + + Returns: + The minimum number of operations required to convert str1 to str2. + + Doctests: + >>> min_cost_string_conversion("apple", "apply") + 1 + >>> min_cost_string_conversion("sunday", "saturday") + 3 + >>> min_cost_string_conversion("test", "test") + 0 + >>> min_cost_string_conversion("", "") + 0 + >>> min_cost_string_conversion("kitten", "sitting") + 3 + """ + costs, _ = compute_transform_tables( + source_string=str1, + destination_string=str2, + copy_cost=0, + replace_cost=1, + delete_cost=1, + insert_cost=1, + ) + return costs[-1][-1] + def compute_transform_tables( source_string: str, @@ -18,26 +53,11 @@ def compute_transform_tables( insert_cost: int, ) -> tuple[list[list[int]], list[list[str]]]: """ - Finds the most cost efficient sequence - for converting one string into another. - - >>> costs, operations = compute_transform_tables("cat", "cut", 1, 2, 3, 3) - >>> costs[0][:4] - [0, 3, 6, 9] - >>> costs[2][:4] - [6, 4, 3, 6] - >>> operations[0][:4] - ['0', 'Ic', 'Iu', 'It'] - >>> operations[3][:4] - ['Dt', 'Dt', 'Rtu', 'Ct'] - - >>> compute_transform_tables("", "", 1, 2, 3, 3) - ([[0]], [['0']]) + Finds the most cost efficient sequence for converting one string into another + using dynamic programming with specified costs. """ - source_seq = list(source_string) - destination_seq = list(destination_string) - len_source_seq = len(source_seq) - len_destination_seq = len(destination_seq) + len_source_seq = len(source_string) + len_destination_seq = len(destination_string) costs = [ [0 for _ in range(len_destination_seq + 1)] for _ in range(len_source_seq + 1) ] @@ -47,124 +67,48 @@ def compute_transform_tables( for i in range(1, len_source_seq + 1): costs[i][0] = i * delete_cost - ops[i][0] = f"D{source_seq[i - 1]}" + ops[i][0] = f"D{source_string[i - 1]}" - for i in range(1, len_destination_seq + 1): - costs[0][i] = i * insert_cost - ops[0][i] = f"I{destination_seq[i - 1]}" + for j in range(1, len_destination_seq + 1): + costs[0][j] = j * insert_cost + ops[0][j] = f"I{destination_string[j - 1]}" for i in range(1, len_source_seq + 1): for j in range(1, len_destination_seq + 1): - if source_seq[i - 1] == destination_seq[j - 1]: + if source_string[i - 1] == destination_string[j - 1]: costs[i][j] = costs[i - 1][j - 1] + copy_cost - ops[i][j] = f"C{source_seq[i - 1]}" + ops[i][j] = f"C{source_string[i - 1]}" else: costs[i][j] = costs[i - 1][j - 1] + replace_cost - ops[i][j] = f"R{source_seq[i - 1]}" + str(destination_seq[j - 1]) + ops[i][j] = f"R{source_string[i - 1]}{destination_string[j - 1]}" if costs[i - 1][j] + delete_cost < costs[i][j]: costs[i][j] = costs[i - 1][j] + delete_cost - ops[i][j] = f"D{source_seq[i - 1]}" + ops[i][j] = f"D{source_string[i - 1]}" if costs[i][j - 1] + insert_cost < costs[i][j]: costs[i][j] = costs[i][j - 1] + insert_cost - ops[i][j] = f"I{destination_seq[j - 1]}" + ops[i][j] = f"I{destination_string[j - 1]}" return costs, ops def assemble_transformation(ops: list[list[str]], i: int, j: int) -> list[str]: """ - Assembles the transformations based on the ops table. - - >>> ops = [['0', 'Ic', 'Iu', 'It'], - ... ['Dc', 'Cc', 'Iu', 'It'], - ... ['Da', 'Da', 'Rau', 'Rat'], - ... ['Dt', 'Dt', 'Rtu', 'Ct']] - >>> x = len(ops) - 1 - >>> y = len(ops[0]) - 1 - >>> assemble_transformation(ops, x, y) - ['Cc', 'Rau', 'Ct'] - - >>> ops1 = [['0']] - >>> x1 = len(ops1) - 1 - >>> y1 = len(ops1[0]) - 1 - >>> assemble_transformation(ops1, x1, y1) - [] - - >>> ops2 = [['0', 'I1', 'I2', 'I3'], - ... ['D1', 'C1', 'I2', 'I3'], - ... ['D2', 'D2', 'R23', 'R23']] - >>> x2 = len(ops2) - 1 - >>> y2 = len(ops2[0]) - 1 - >>> assemble_transformation(ops2, x2, y2) - ['C1', 'I2', 'R23'] + Assembles the list of transformations based on the ops table. """ if i == 0 and j == 0: return [] - elif ops[i][j][0] in {"C", "R"}: + op_code = ops[i][j][0] + if op_code in {"C", "R"}: seq = assemble_transformation(ops, i - 1, j - 1) - seq.append(ops[i][j]) - return seq - elif ops[i][j][0] == "D": + elif op_code == "D": seq = assemble_transformation(ops, i - 1, j) - seq.append(ops[i][j]) - return seq - else: + else: # op_code == "I" seq = assemble_transformation(ops, i, j - 1) - seq.append(ops[i][j]) - return seq + seq.append(ops[i][j]) + return seq if __name__ == "__main__": - _, operations = compute_transform_tables("Python", "Algorithms", -1, 1, 2, 2) - - m = len(operations) - n = len(operations[0]) - sequence = assemble_transformation(operations, m - 1, n - 1) - - string = list("Python") - i = 0 - cost = 0 - - with open("min_cost.txt", "w") as file: - for op in sequence: - print("".join(string)) - - if op[0] == "C": - file.write("%-16s" % "Copy %c" % op[1]) # noqa: UP031 - file.write("\t\t\t" + "".join(string)) - file.write("\r\n") - - cost -= 1 - elif op[0] == "R": - string[i] = op[2] - - file.write("%-16s" % ("Replace %c" % op[1] + " with " + str(op[2]))) # noqa: UP031 - file.write("\t\t" + "".join(string)) - file.write("\r\n") - - cost += 1 - elif op[0] == "D": - string.pop(i) - - file.write("%-16s" % "Delete %c" % op[1]) # noqa: UP031 - file.write("\t\t\t" + "".join(string)) - file.write("\r\n") - - cost += 2 - else: - string.insert(i, op[1]) - - file.write("%-16s" % "Insert %c" % op[1]) # noqa: UP031 - file.write("\t\t\t" + "".join(string)) - file.write("\r\n") - - cost += 2 - - i += 1 - - print("".join(string)) - print("Cost: ", cost) - - file.write("\r\nMinimum cost: " + str(cost)) + doctest.testmod()