问题起源: 最早是群里一个兄弟要用到,原意大概是有25个空位,每个空位可以选择三种不同的元素,然后就要枚举全部可能性。
第一反应就是写循环:25层循环,想想就会崩溃掉吧。所以我花了几秒钟想了如下这个算法(大学时代我研究过这个,所以几秒钟是催牛逼用的)。
算法描述如下:
1. 25个空位可以看做25个瓶子,3个元素看做3个不同颜色的球,球放在瓶子里,并且他们谦让友爱懂得排队;
2. 每次遍历都是只看瓶口的那个球的颜色;
3. 每次遍历结束的时候,都把最后一个瓶子里的球倒出一个;
4. 如果当前瓶子空了,那么前一个瓶子里的球倒出一个,并且把本瓶子重新按顺序装满;
5. 如果上一个瓶子也空了,那么继续向上找,直到找到一个不空的瓶子并倒出一个;之后把这个瓶子后面的瓶子都重新按顺序装满;
6. 第一个瓶子也空了的时候,结束。
于是在这个基础上开始写代码,花了几分钟写完,然后又花了几分钟调试,最后又花了几分钟优化了下代码风格,如下:
typedef unsigned __int64 u_int64;
const int g_nMaxCount = 15;
const int g_nMaxBall = 4;
u_char g_nData[g_nMaxCount][g_nMaxBall] = { 0 };
void ResetData( u_char * nData )
{
for( int i = 0; i < g_nMaxBall - 1; i ++ )
{
nData[i] = 'a' + i;
}
nData[g_nMaxBall - 1] = 0;
}
inline bool IsDataNull(u_char * nData)
{
return nData[0] == 0 ? true : false;
}
void PopData( u_char * nData)
{
int nTemp = nData[0];
for( int i = 0; i < g_nMaxBall - 1; i ++ )
{
nData[i] = nData[i + 1];
}
nData[g_nMaxBall - 1] = nTemp;
}
u_int64 StatTest()
{
for( int i = 0; i < g_nMaxCount; i ++ )
{
ResetData( g_nData[i] );
}
u_int64 nCount = 0;
int nIndex = g_nMaxCount - 1; //借位标识
u_int nTimeNow = ::GetTickCount();
while( nIndex >= 0 )
{
char chBuf[g_nMaxCount + 2] = { 0 };
for( int i = 0; i < g_nMaxCount; i ++ )
{
chBuf[i] = g_nData[i][0];
}
//chBuf里就是一个排列,怎么用就看你的心情了
for( int i = g_nMaxCount - 1; i >= nIndex; i -- )
{
PopData( g_nData[i] );
if( IsDataNull( g_nData[i] ) )
{
ResetData( g_nData[i] ); //还原自己
if( i == nIndex )
{
nIndex --;
PopData( g_nData[nIndex] );
break;
}
}
else
{
break;
}
}
nCount ++;
}
nTimeNow = ::GetTickCount() - nTimeNow;
<span style="white-space:pre"> </span>//因为3的25次全排列太费时,我选用了15次做例子,不信你可以试试,3的15次全排列运行结果大概是1.5秒左右(在我的机器上),25次全排列反正好几分钟没有搞定。
return nCount;
}
看到这里,如果你还在继续看的话,那么恭喜你,你要看到“数学”的魅力了。
上面这个算法是不是非常熟悉,好像总觉得哪里见过。那么我直接告诉你,还记得大明湖畔的夏雨荷吗?错了,是小学时学习珠算的“算盘”否,这个算法最好的描述就是用算盘来说啊,比“那个神马瓶子和自觉排队的球”比喻好多了。因为这个算法本质上就是一个 “三进制”的 减法操作嘛!
既然三进制减法可以做到,那么我们反过来用“加法”也是可以做到的吧,毕竟3个25次全排列总数量是固定的:就是3的25次方啊。于是经过一番酝酿,我写了如下这个
“三进制加法”
版本的C实现。
//以下为另一个枚举3的15次方的算法
const int g_nMaxCount = 15;
const int g_nMaxBall = 3;
u_char g_nData2[g_nMaxBall] = { 0 };
u_int64 StatTest()
{
u_int64 nCount = 0;
u_int nTimeNow = ::GetTickCount();
for( int i = 0; i < g_nMaxBall; i ++ )
{
g_nData2[i] = 'a' + i;
}
u_int64 nMaxCount = (u_int64)( pow( (float)g_nMaxBall, (float)g_nMaxCount ) );
for( u_int64 i = 0; i < nMaxCount; i ++ )
{
char chBuf[g_nMaxCount + 2] = { 0 };
u_int64 nData = i;
for( int j = 0; j < g_nMaxCount; j ++ )
{
chBuf[g_nMaxCount - 1 - j] = g_nData2[nData % g_nMaxBall];
nData /= g_nMaxBall; //这个除法耗时严重
}
nCount ++;
//chBuf里就是一个排列,怎么用就看你的心情了
}
nTimeNow = ::GetTickCount() - nTimeNow;
//因为3的25次全排列太费时,我选用了15次做例子,不信你可以试试,3的15次全排列运行结果大概是4.0秒左右(在我的机器上),25次全排列反正好几分钟没有搞定。
return nCount;
}
第二种方法耗时会翻翻,因为有个除法调用频繁,如果你优化这个,则能很好,希望同大家继续交流。
本博文是我发表的第二篇,第一篇被未知原因没有发出来。
我的联系方式:QQ群【MDuiLib玛德界面交流
371989312
】。
欢迎来加入“玛德界面库的世界”。