-
Notifications
You must be signed in to change notification settings - Fork 0
/
Sort 0 1 2
68 lines (59 loc) · 1.98 KB
/
Sort 0 1 2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
You are given an integer array/list(ARR) of size N. It contains only 0s, 1s and 2s. Write a solution to sort this array/list in a 'single scan'.
'Single Scan' refers to iterating over the array/list just once or to put it in other words, you will be visiting each element in the array/list just once.
Note:
You need to change in the given array/list itself. Hence, no need to return or print anything.
Input format :
The first line contains an Integer 't' which denotes the number of test cases or queries to be run. Then the test cases follow.
First line of each test case or query contains an integer 'N' representing the size of the array/list.
Second line contains 'N' single space separated integers(all 0s, 1s and 2s) representing the elements in the array/list.
Output Format :
For each test case, print the sorted array/list elements in a row separated by a single space.
Output for every test case will be printed in a separate line.
Constraints :
1 <= t <= 10^2
0 <= N <= 10^5
Time Limit: 1 sec
Sample Input 1:
1
7
0 1 2 0 2 0 1
Sample Output 1:
0 0 0 1 1 2 2
Sample Input 2:
2
5
2 2 0 1 1
7
0 1 2 0 1 2 0
Sample Output 2:
0 1 1 2 2
0 0 0 1 1 2 2
_________________________________________________
solution is here
def sort012(arr):
n = len(arr)
low = 0 # Pointer for 0s
mid = 0 # Pointer for 1s
high = n - 1 # Pointer for 2s
while mid <= high:
if arr[mid] == 0:
arr[mid], arr[low] = arr[low], arr[mid]
low += 1
mid += 1
elif arr[mid] == 1:
mid += 1
else:
arr[mid], arr[high] = arr[high], arr[mid]
high -= 1
# Read the number of test cases
t = int(input())
for _ in range(t):
# Read the size of the array
n = int(input())
# Read the array elements
arr = list(map(int, input().split()))
# Sort the array using the Dutch National Flag algorithm
sort012(arr)
# Print the sorted array
print(*arr)
______________________________________________________________