码迷,mamicode.com
首页 > 编程语言 > 详细

Sort Colors leetcode java

时间:2014-07-28 11:34:20      阅读:297      评论:0      收藏:0      [点我收藏+]

标签:style   blog   java   color   os   strong   io   for   

题目:

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue.

Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library‘s sort function for this problem.

Follow up:
A rather straight forward solution is a two-pass algorithm using counting sort.
First, iterate the array counting number of 0‘s, 1‘s, and 2‘s, then overwrite array with total number of 0‘s, then 1‘s and followed by 2‘s.

Could you come up with an one-pass algorithm using only constant space?

 

题解:

这道题就是运用指针来解决了,可以说叫3指针吧。

一个指针notred从左开始找,指向第一个不是0(红色)的位置;一个指针notblue从右开始往左找,指向第一个不是2(蓝色)的位置。

然后另一个新的指针i指向notred指向的位置,往后遍历,遍历到notred的位置。

这途中需要判断:

当i指向的位置等于0的时候,说明是红色,把他交换到notred指向的位置,然后notred++,i++。

当i指向的位置等于2的时候,说明是蓝色,把他交换到notblue指向的位置,然后notred--。

当i指向的位置等于1的时候,说明是白色,不需要交换,i++即可。

 

代码如下:

 1     public static void swap(int A[], int i, int j){
 2         int tmp = A[i];
 3         A[i]=A[j];
 4         A[j]=tmp;
 5     }
 6     
 7     public static void sortColors(int A[]) {
 8         if(A == null || A.length==0)
 9             return;
10             
11         int notred=0;
12         int notblue=A.length-1;
13          
14         while (notred<A.length&&A[notred]==0)
15             notred++;
16             
17         while (notblue>=0&&A[notblue]==2)
18             notblue--;
19          
20         int i=notred;
21         while (i<=notblue){
22             if (A[i]==0) {
23                 swap(A,i,notred);
24                 notred++;
25                 i++;
26             }else if (A[i]==2) {
27                 swap(A,i,notblue);
28                 notblue--;
29             }else
30                 i++;
31         }
32     }

 

 

Sort Colors leetcode java,布布扣,bubuko.com

Sort Colors leetcode java

标签:style   blog   java   color   os   strong   io   for   

原文地址:http://www.cnblogs.com/springfor/p/3872313.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!