算法学习笔记（一）-----彩色小球排序

2019独角兽企业重金招聘Python工程师标准>>>

“有红黄蓝三色小球若干排成一列，这些小球进行排序，请使用尽量少的空间和时间”---来自微信号"九章算法"

思想：同色小球之间应该没有差别，所以用1，2,3分别表示红黄蓝，使用数组ordered_balls[N]前面排红色，后面排蓝色，但黄色只有新建一个数组再排序了。代码如下，先实现的是不超过10个数的。。。

#include

#include

void main()

{

int balls[10],ordered_balls[10],blue_balls[10];

int  red_i=0,blue_i=0,yellow_i=0;

cout<<"please enter the color balls(no more than 10)(1 for red balls, 2 for blue balls, 3 for yellow balls):\n";

for (int i=0;i<10;i++)

{

cin>>balls[i];

if(balls[i]!=1&&balls[i]!=2&&balls[i]!=3)

{

cout<<"wrong number!\n";

break;

}

}

for( i=0;i<10;i++)

{

if(balls[i]==1)

ordered_balls[red_i++]=balls[i];

else if(balls[i]==2)

blue_balls[blue_i++]=balls[i];

else if(balls[i]==3)

ordered_balls[10-(yellow_i++)]=balls[i];

}

//order completed

for(i=0;i

cout<<ordered_balls[i]<<" ";

for(i=0;i

cout<<blue_balls[i]<<" ";

for(i=10;i>10-yellow_i;i--)

cout<<ordered_balls[i]<<" ";

cout<<endl;

}

这个简单粗暴的办法占用空间N(总数)+n_b（蓝色小球个数），时间上O（N+n_b）

其实编到后面想其实捷径是不是遍历一遍，数出红蓝黄各有多少个，就可以给出答案了。但是这样只有答案，却不是实实在在地给小球排序。。。

转载于:https://my.oschina.net/u/1475850/blog/211825