用程序去判断每个人说话的真假 - 写给新手的枚举介绍

背景

我们在以前的数学题中一定遇到过这样的问题：

A说xxxx
B说xxx
C说xxx

然后给一些条件，让你判断每个人说话的真假

这个题目是这样的：

计算机学院准备组织院篮球赛，某班有ABCDE五个同学商量组队参加，他们在讨论谁来打前锋的时候发生了争执，于是他们请了另一个班的同学J当评委，五个人PK百米速度，谁的速度最快就由谁来当前锋，其实五个同学速度相当，比赛结束时，J让他们猜猜排名情况

• A说：“E一定是第一名”
• B说：“我可能是第二名”
• C说：“A最慢”
• D说：“C不是最快的”
• E说：“D应该是第一名”

J最后说：“E肯定不是第二名或者第三名，你们几个只有获得第一名和第二名的人猜对了，你们应该知道谁最快了吧？”
编程给出五个同学的排名。

算法

习惯了通过表达式去计算一个结果的人，很难去用枚举的思维方式来处理问题。尤其对于枚举的模型很难把握。

枚举算法求解这类问题其实很简单，我尝试用以下三句话来说明：

• 把每个人说的话用一个逻辑表达式表示出来
• 枚举解空间内所有可能出现的情况
• 按照题中条件，筛选合法解

建立说话内容（断言）的模型

可以很容易想到，上述问题中A的内容“E是第一名”可以写成：

Score['E'] == 1;

关键在于，如何描述“A说”的这个过程。很容易想到，这个过程实际上是一个从字符到表达式的映射。表达式在C语言中如何作为一个变量来使用？当然是函数指针，于是可以写出下面的代码：

int guessA(){    return Score['E'] == 1;}int guessB(){    return Score['B'] == 2;}...

int (*guess[256])();    //  定义一个函数指针的数组guess['A'] = guessA;guess['B'] = guessB;...

这样的话，就可以用guess['A']()来表示A猜得正确与否

枚举解空间

这个题目较为简单，5个人的排名当然有5!种情况，可以用C++提供的全排列的库实现，也可以用深搜自己实现。这里使用康托展开逆运算得到（每次计算的时间复杂度为O(1)）

//  生成1~n的全排列中得第k项，返回值在ret中int fac[]={1,1,2,6,24,120,720,5040,40320,362880};void invKT(int n, int k, int ret[]){    k--;    int vst[8]={0}, j;    for ( int i=0; i<n; i++){        int t = k/fac[n-i-1];        for (j=1; j<=n; j++){            if (!vst[j]){                if (t==0) break;                t--;            }        }        ret[i] = j;        vst[j] = 1;        k %= fac[n-i-1];    }}

我们可以枚举得到每次的结果：

int score[N];   //  score[i]表示('A'+i)排第几名int rank[N];    //  rank[i]表示第i名是谁for ( int i=1; i<=fac[n]; i++) {    invKT(n, i, score);    for ( int j=0; j<n; j++) rank[score[i]] = 'A'+i;}

筛选

有了以上的准备工作，筛选就非常容易了。

只有第一名和第二名说的正确：

if (     guess[ rank[1] ]() == 1 &&    guess[ rank[2] ]() == 1 &&    guess[ rank[3] ]() == 0 &&    guess[ rank[4] ]() == 0 &&    guess[ rank[5] ]() == 0)

E不是第二和第三：

if ( score['E'-'A'] != 2 && score['E'-'A'] != 3 )

总代码

实际的代码为了偷懒，在一些细节上跟上面的分析不太一样，仅供参考。

#include <cstdio>#include <cstdlib>int score[6];   //  i排第几int rank[6];    //  第i名是谁void invKT(int n, int k, int ret[]);int guess(int i);int main(){    int n = 5;    for ( int i=1;i<=120;i++){        invKT(n, i, score+1);        for ( int j=1;j<=n;j++) rank[score[j]] = j;        if (guess(rank[1]) && guess(rank[2])             && guess(rank[3])==0 && guess(rank[4])==0             && guess(rank[5])==0             && score[5]!=2 && score[5]!=3) {                for ( int j=1;j<=n;j++){                    printf("%c：第%d名\n", 'A'+j-1, score[j]);                }        }    }    return 0;}int guess(int i){    return (i==1)*(score[5]==1)        ||  (i==2)*(score[2]==2)        ||  (i==3)*(score[1]==5)        ||  (i==4)*(score[3]!=1)        ||  (i==5)*(score[4]==1);}int fac[]={1,1,2,6,24,120,720,5040,40320,362880};void invKT(int n, int k, int ret[]){    k--;    int vst[8]={0}, j;    for ( int i=0; i<n; i++){        int t = k/fac[n-i-1];        for (j=1; j<=n; j++){            if (!vst[j]){                if (t==0) break;                t--;            }        }        ret[i] = j;        vst[j] = 1;        k %= fac[n-i-1];    }}