「有紅黃藍三色小球若干排成一列,這些小球進行排序,請使用盡可能少的空間和時間」---來自微信號"九章算法"算法
思想:同色小球之間應該沒有差異,因此用1,2,3分別表示紅黃藍,使用數組ordered_balls[N]前面排紅色,後面排藍色,但黃色只有新建一個數組再排序了。代碼以下,先實現的是不超過10個數的。。。數組
#include微信
#includespa
void main()排序
{ci
int balls[10],ordered_balls[10],blue_balls[10];遍歷
int red_i=0,blue_i=0,yellow_i=0;im
cout<<"please enter the color balls(no more than 10)(1 for red balls, 2 for blue balls, 3 for yellow balls):\n";img
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)
其實編到後面想其實捷徑是否是遍歷一遍,數出紅藍黃各有多少個,就能夠給出答案了。可是這樣只有答案,卻不是實實在在地給小球排序。。。