-
Notifications
You must be signed in to change notification settings - Fork 1
/
SortColors.java
34 lines (32 loc) · 962 Bytes
/
SortColors.java
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
package com.leetcode;
public class SortColors {
//https://leetcode.com/problems/sort-colors/
class Solution {
public void sortColors(int[] nums) {
int n = nums.length;
int l = 0;
int r = n-1;
int temp;
int i=0;
while(i<=r){
if(nums[i] == 0){
// put in beginning
temp = nums[l];
nums[l] = nums[i];
nums[i] = temp;
l++;
i++;//increase as we covered all before
}else if(nums[i] == 2){
//put in last
temp = nums[r];
nums[r] = nums[i];
nums[i] = temp;
//dont do i++ as compare needed
r--;
}else if(nums[i] == 1){
i++;
}
}
}
}
}