Update cyclesort.py

Changing for Python 3 using exception handling for robust code
This commit is contained in:
Sayan Bandyopadhyay 2018-01-07 13:21:05 +05:30 committed by GitHub
parent 1f41a310f3
commit 5d4471d35a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -44,7 +44,13 @@ def cycle_sort(array):
# Main Code starts here
user_input = input('Enter numbers separated by a comma:\n')
if __name__ == '__main__':
try:
raw_input # Python 2
except NameError:
raw_input = input # Python 3
user_input = raw_input('Enter numbers separated by a comma:\n')
unsorted = [int(item) for item in user_input.split(',')]
n = len(unsorted)
cycle_sort(unsorted)