Hacktober_Repo_22/CPP/sort-an-array-of-0s-1s-2s.cpp at main · py3-coder/Hacktober_Repo_22 · GitHub

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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#include<bits/stdc++.h>
using namespace std;

class Solution
{
    public:
    void sort012(int arr[], int n)
    {
        //using dutch flag method
        int low = 0;
        int mid = 0 ;
        int high = n-1;

        while(mid <= high)
        {
            if(arr[mid] == 0)
            {
                swap(arr[mid], arr[low]);
                mid++;
                low++;
            }
            else if(arr[mid] == 1)
            {

                mid++;

            }
            else
            {
                swap(arr[mid], arr[high]);
                high--;
            }
        }


        //using count
    //     int zero = 0;
    //     int one = 0;
    //     int two = 0;

    //     // Count the number of 0s, 1s and 2s in the array
    // for (int i = 0; i < n; i++) {
    //     switch (a[i]) {
    //     case 0:
    //         zero++;
    //         break;
    //     case 1:
    //         one++;
    //         break;
    //     case 2:
    //         two++;
    //         break;
    //     }
    // }

    // // Update the array
    // int i = 0;

    // // Store all the 0s in the beginning
    // while (zero > 0) {
    //     a[i++] = 0;
    //     zero--;
    // }

    // // Then all the 1s
    // while (one > 0) {
    //     a[i++] = 1;
    //     one--;
    // }

    // // Finally all the 2s
    // while (two > 0)
    // {
    //     a[i++] = 2;
    //     two--;
    // }


        // using sort function
       // return sort(a, a+n+1);
    }

};


int main() {

    int t;
    cin >> t;

    while(t--){
        int n;
        cin >>n;
        int a[n];
        for(int i=0;i<n;i++){
            cin >> a[i];
        }

        Solution ob;
        ob.sort012(a, n);

        for(int i=0;i<n;i++){
            cout << a[i]  << " ";
        }

        cout << endl;


    }
    return 0;
}