classSolution{ publicvoidsortColors(int[] nums){ int n = nums.length; int a = 0, b = 0, c = 0; for (int i = 0; i < n; i++) { if (nums[i] == 0) { a = a + 1; } elseif (nums[i] == 1) { b = b + 1; } elseif (nums[i] == 2) { c = c + 1; } } for (int i = 0; i <= a - 1; i++) { nums[i] = 0; } for (int i = a; i <= a + b - 1; i++) { nums[i] = 1; } for (int i = a + b; i <= a + b + c - 1; i++) { nums[i] = 2; } for (int i = 0; i < n; i++) { System.out.print(nums[i]); } }
publicstaticvoidmain(String[] args){ solution s = new solution(); int[] b = { 2, 0, 2, 1, 1, 0 }; s.sortColors(b); }