【阿里巴巴面试题】求1,2,2,3,4,5能够组合得到的不同数

今天同事问了我一道阿里面试题，题目如下：

编程实现：有 1,2,2,3,4,5 六个数字，请输出它们所有不同的排序结果，如“322145”、“232154”等，其中 4 不能在第三位， 3 与 5 不能相连。

 

好久没碰类似题目了，感觉脑子有点生锈。想出一个比较简单的搜索实现。主要有以下两点需要注意：

1.对于两个2的处理。如果这六个数字两两不同，则每个组合都是唯一的。但出现相同两个数会导致结果减少一半。我在程序取了个巧，以1,2,3,4,5,6作为六个数字，将最终求得的结果除以2，即得到不同组合数。

2.题目要求输出所有不同的排序结果。需要将搜索得到的六位数进行替换，将6替换为2，但这会产生每个数出现2次，因此利用一个数组来计数，首次出现才输出。

 

code:

#include <cstdio> #include <cstdlib> #include <cstring> #include <iostream> using namespace std; //stack int flag[10]; //max num is 543221 int cnt[600000]; int search(int depth, int tot, string str) { if(depth+1 == tot) { //fprintf(stderr, "%s/n", str.c_str()); int num = 0; for(int i = 0; i < tot; ++i) { int k = str[i] - '0'; if(k == 6) k = 2; num = num*10 + k; } if(cnt[num] == 0) { fprintf(stderr, "%d/n", num); cnt[num] = 1; } return 1; } if(depth == -1) { int ret = 0; for(int i = 1; i <= 6; ++i) { flag[i] = 1; char tmp[10]; sprintf(tmp, "%c", '0'+i); ret += search(depth+1, tot, tmp); flag[i] = 0; } return ret; } else { int cur = str[depth]-'0'; int ret = 0; for(int i = 1; i <= 6; ++i) { if(cur == i || flag[i] == 1) continue; // 4 is not appear at 3rd position. if(depth == 1 && i == 4) continue; // 3 and 5 are not neighbours. if( (cur == 3 && i == 5) || (cur == 5 && i == 3) ) continue; flag[i] = 1; ret += search(depth+1, tot, str+char('0'+i)); flag[i] = 0; } return ret; } } int main(int argc, char **argv) { memset(flag, 0, sizeof(flag)); memset(cnt, 0, sizeof(cnt)); int ret = search(-1, 6, ""); fprintf(stderr, "%d/n", ret/2); return 0; }