From 55c3d22707c0a0d6f119d16b6024a78bd1d901d2 Mon Sep 17 00:00:00 2001 From: Simon Green Date: Sun, 1 Sep 2024 18:59:00 +1000 Subject: sgreen solutions to challenge 284 --- challenge-284/sgreen/python/ch-2.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100755 challenge-284/sgreen/python/ch-2.py (limited to 'challenge-284/sgreen/python/ch-2.py') diff --git a/challenge-284/sgreen/python/ch-2.py b/challenge-284/sgreen/python/ch-2.py new file mode 100755 index 0000000000..a236c11600 --- /dev/null +++ b/challenge-284/sgreen/python/ch-2.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python3 + +import json +import sys + + +def find_index(lst: list, val: int): + # Find the position of val in lst. If it does not appear, return the + # length of the list + return lst.index(val) if val in lst else len(lst) + + +def relative_sort(list1: list, list2: list) -> list: + # Sort the first list, in the order they appear in the second list. If + # it doesn't appear in the second list, append to the end in numerical order. + return sorted(list1, key=lambda i: (find_index(list2, i), i)) + + +def main(): + lists = json.loads(sys.argv[1]) + result = relative_sort(*lists) + + # Convert to a tuple to match expected output + print(tuple(result)) + + +if __name__ == '__main__': + main() -- cgit