-
Notifications
You must be signed in to change notification settings - Fork 8
/
sort o,1,and 2.c
64 lines (56 loc) · 1.07 KB
/
sort o,1,and 2.c
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
// C program to sort an array with 0, 1 and 2
// in a single pass
#include <stdio.h>
/* Function to swap *a and *b */
void swap(int* a, int* b);
// Sort the input array, the array is assumed to
// have values in {0, 1, 2}
void sort012(int a[], int arr_size)
{
int lo = 0;
int hi = arr_size - 1;
int mid = 0;
// Iterate till all the elements
// are sorted
while (mid <= hi) {
switch (a[mid]) {
// If the element is 0
case 0:
swap(&a[lo++], &a[mid++]);
break;
// If the element is 1
case 1:
mid++;
break;
// If the element is 2
case 2:
swap(&a[mid], &a[hi--]);
break;
}
}
}
/* UTILITY FUNCTIONS */
void swap(int* a, int* b)
{
int temp = *a;
*a = *b;
*b = temp;
}
/* Utility function to print array arr[] */
void printArray(int arr[], int arr_size)
{
int i;
for (i = 0; i < arr_size; i++)
printf("%d ", arr[i]);
}
/* driver program to test */
int main()
{
int arr[] = { 0, 1, 1, 0, 1, 2, 1, 2, 0, 0, 0, 1 };
int arr_size = sizeof(arr) / sizeof(arr[0]);
int i;
sort012(arr, arr_size);
printArray(arr, arr_size);
getchar();
return 0;
}