标签:
给定一个有n个对象(包括k种不同的颜色,并按照1到k进行编号)的数组,将对象进行分类使相同颜色的对象相邻,并按照1,2,...k的顺序进行排序。
样例
给出colors=[3, 2, 2, 1, 4]
,k=4
, 你的代码应该在原地操作使得数组变成[1, 2, 2, 3, 4]
解题
直接快排
class Solution { /** * @param colors: A list of integer * @param k: An integer * @return: nothing */ public void sortColors2(int[] colors, int k) { // write your code here sort(colors,0,colors.length - 1); } public void sort(int[] A,int low,int high){ if(low >= high) return ; int i = low; int j = high; int tmp = A[low]; while(i<j){ while(i<j && A[j]>tmp) j--; if(i<j){ A[i] = A[j]; i++; } while(i<j && A[i]<= tmp) i++; if(i<j){ A[j] = A[i]; j--; } } A[i] = tmp; sort(A,low,i-1); sort(A,i+1,high); } }
标记法,先统计各个颜色的次数,然后根据数组更改次数
class Solution { /** * @param colors: A list of integer * @param k: An integer * @return: nothing */ public void sortColors2(int[] colors, int k) { // write your code here int[] flag = new int[k+1]; for(int i = 0;i<colors.length;i++){ flag[colors[i]]++; } int c = 1; for(int i = 0;i<colors.length;i++){ while(flag[c]==0){// 颜色可能为0 的比较多 c++; } colors[i] = c; flag[c]--; } } }
只有三种颜色的排序 的时候,但是当有多个的时候判断情况太多。
在九章看到两个指针的方法
class Solution { /** * @param colors: A list of integer * @param k: An integer * @return: nothing */ public void sortColors2(int[] colors, int k) { int count = 0; int start = 0; int end = colors.length-1; while (count < k) { int min = Integer.MAX_VALUE; int max = Integer.MIN_VALUE; for (int i = start; i <= end; i++) { min = Math.min(min, colors[i]); max = Math.max(max, colors[i]); } int left = start; int right = end; int cur = left; while(cur <= right) { if (colors[cur] == min) { swap(left, cur, colors); cur++; left++; } else if (colors[cur] > min && colors[cur] < max) { cur++; } else { int tmp = colors[cur]; swap(cur, right, colors); right--; } } count += 2; start = left; end = right; } } void swap(int left, int right, int[] colors) { int tmp = colors[left]; colors[left] = colors[right]; colors[right] = tmp; } }
理解不透,脑子太笨。
标签:
原文地址:http://www.cnblogs.com/theskulls/p/5293491.html