题目:给定一个字符串里面只有"R" "G" "B" 三个字符,请排序,最终结果的顺序是R在前 G中 B在后。 要求:空间复杂度是O(1),且只能遍历一次字符串。
解析:本题的解法类似于快速排序partition算法,对于字符串str,利用两个下标start1,end1,start1初始时指向字符串的头部,end1初始时指向字符串的尾部,当str[start1]!='R',str[end1]!='B'时候,不能简单像原来的partition那样直接作交换,而是要根据不同的情况做不同的处理,因为字符串中还包含第三类字符'G'。
C ++代码如下:
#include<iostream>
#include<cstring>
using namespace std;
const int MAXLEN=100;//字符串的最大长度
void Set_RGB(char *str);//位置调整函数,借助了快排的partition函数思想
int main()
{
char str[MAXLEN+1];
cin>>str;//输入只包含'R','G','B'的字符串
Set_RGB(str);
cout<<str<<endl;//输出调整过的字符串
return 0;
}
void Set_RGB(char *str)
{
if(str==NULL)
return;
int len=strlen(str);
int start1,end1,start2,end2;
char tmp;
start1=start2=0;
end1=end2=len-1;
while(true)
{
while(str[start1]=='R'&&start1<=end1)
{
start1++;
if(start1>start2)
start2=start1;//保证start2始终等于或大于start1
}
while(str[end1]=='B')
{
end1--;
if(end1<end2)
end2=end1;//保证end2始终等于或小于end1
}
if(start1>end1)
break;
if(str[start1]=='B'&&str[end1]=='R')//直接交换
{
tmp=str[start1];
str[start1]=str[end1];
str[end1]=tmp;
}
else if(str[start1]=='G'&&str[end1]=='R')
{
//先交换
tmp=str[start1];
str[start1]=str[end1];
str[end1]=tmp;
while(end2>start2)//然后用end2向前扫描知道遇到第一个不是'G'的字符
{
if(str[end2]!='G')
break;
end2--;
}
if(start2<end2)//若start2<end2,则end2和end1位置的字符交换
{
tmp=str[end2];
str[end2]=str[end1];
str[end1]=tmp;
}
}
else if(str[start1]=='B'&&str[end1]=='G')
{
//先交换
tmp=str[start1];
str[start1]=str[end1];
str[end1]=tmp;
while(start2<end2)//然后用start2向前扫描知道遇到第一个不是'G'的字符
{
if(str[start2]!='G')
break;
start2++;
}
if(start2<end2)//若start2<end2,则start2和start1位置的字符交换
{
tmp=str[start2];
str[start2]=str[start1];
str[start1]=tmp;
}
}
else
{
while(start2<=end2)//分别用start2和end2从前,从后,向后,向前扫描第一个不是'G'的字符
{
if(str[start2]!='G'&&str[end2]!='G')
break;
if(str[start2]=='G')
start2++;
if(str[end2]=='G')
end2--;
}
if(start2==end2&&str[start2]=='R')//若两下标相遇,并且是字符'R',则和start1位置'G'的字符交换
{
tmp=str[start2];
str[start2]=str[start1];
str[start1]=tmp;
}
if(start2==end2&&str[end2]=='B')//若两下标相遇,并且是字符'B',则和end1位置'G'的字符交换
{
tmp=str[end2];
str[end2]=str[end1];
str[end1]=tmp;
}
if(start2<end2)//若两下标相遇,分别交换start1与start2,end1与end2位置的字符
{
tmp=str[end2];
str[end2]=str[end1];
str[end1]=tmp; tmp=str[start2];
str[start2]=str[start1];
str[start1]=tmp;
}
}
if(start2>=end2)//若start2与end2相遇,则扫描结束,保证扫描字符串一遍,因为扫描过程中start2始终在start1后面,end2始终在end1前面,
break;
} }
时间复杂度为O(len),空间复杂度为O(1).