path: root/mbbsd/guess.c

/* $Id$ */
#include "bbs.h"
#define LOGPASS BBSHOME "/etc/winguess.log"

static int
check_data(const char *str)
{
    int             i, j;

    if (strlen(str) != 4)
    return -1;
    for (i = 0; i < 4; i++)
    if (str[i] < '0' || str[i] > '9')
        return -1;
    for (i = 0; i < 4; i++)
    for (j = i + 1; j < 4; j++)
        if (str[i] == str[j])
        return -1;
    return 1;
}

static char    *
get_data(char data[5], int count)
{
    while (1) {
    getdata(6, 0, "輸入四位數字(不重複): ", data, 5, LCECHO);
    if (check_data(data) == 1)
        break;
    }
    return data;
}

static int
guess_play(const char *data, const char *answer, int count)
{
    int             A_num = 0, B_num = 0;
    int             i, j;

    for (i = 0; i < 4; i++) {
    if (data[i] == answer[i])
        A_num++;
    for (j = 0; j < 4; j++)
        if (i == j)
        continue;
        else if (data[i] == answer[j]) {
        B_num++;
        break;
        }
    }
    if (A_num == 4)
    return 1;
    move(count + 8, 55);
    prints("%s => " ANSI_COLOR(1;32) "%dA %dB" ANSI_RESET, data, A_num, B_num);
    return 0;
}

static int
result(int correct, int number)
{
    char            a = 0, b = 0, i, j;
    char            n1[5], n2[5];

    snprintf(n1, sizeof(n1), "%04d", correct);
    snprintf(n2, sizeof(n2), "%04d", number);
    for (i = 0; i < 4; i++)
    for (j = 0; j < 4; j++)
        if (n1[(int)i] == n2[(int)j])
        b++;
    for (i = 0; i < 4; i++)
    if (n1[(int)i] == n2[(int)i]) {
        b--;
        a++;
    }
    return 10 * a + b;
}

static int
legal(int number)
{
    char            i, j;
    char            temp[5];

    snprintf(temp, sizeof(temp), "%04d", number);
    for (i = 0; i < 4; i++)
    for (j = i + 1; j < 4; j++)
        if (temp[(int)i] == temp[(int)j])
        return 0;
    return 1;
}

static void
initcomputer(char flag[])
{
    int             i;

    for (i = 0; i < 10000; i++)
    if (legal(i))
        flag[i] = 1;
    else
        flag[i] = 0;
}

static int
computer(int correct, int total, char flag[], int n[])
{
    int             guess;
    static int      j;
    int             k, i;
    char            data[5];

    if (total == 1) {
    do {
        guess = random() % 10000;
    } while (!legal(guess));
    } else
    guess = n[random() % j];
    k = result(correct, guess);
    if (k == 40) {
    move(total + 8, 25);
    snprintf(data, sizeof(data), "%04d", guess);
    prints("%s => 猜中了!!", data);
    return 1;
    } else {
    move(total + 8, 25);
    snprintf(data, sizeof(data), "%04d", guess);
    prints("%s => " ANSI_COLOR(1;32) "%dA %dB" ANSI_RESET, data, k / 10, k % 10);
    }
    j = 0;
    for (i = 0; i < 10000; i++)
    if (flag[i]) {
        if (result(i, guess) != k)
        flag[i] = 0;
        else
        n[j++] = i;
    }
    return 0;
}

static void
Diff_Random(char *answer)
{
    register int    i = 0, j, k;

    while (i < 4) {
    k = random() % 10 + '0';
    for (j = 0; j < i; j++)
        if (k == answer[j])
        break;
    if (j == i) {
        answer[j] = k;
        i++;
    }
    }
    answer[4] = 0;
}

int
guess_main(void)
{
    char            data[5];
    char            computerwin = 0, youwin = 0;
    int             count = 0, c_count = 0;
    char            ifcomputer[2];
    char            answer[5];
    char            yournum[5];
    const int max_guess = 10;

    // these variables are not very huge, no need to use malloc
    // to prevent heap allocation.
    char        flag[10000];
    int         n[1500];

    setutmpmode(GUESSNUM);
    clear();
    showtitle("猜數字", BBSName);

    Diff_Random(answer);
    move(2, 0);
    clrtoeol();

    getdata(4, 0, "您要和電腦比賽嗎? <Y/n>[y]:",
        ifcomputer, sizeof(ifcomputer), LCECHO);
    *ifcomputer = (*ifcomputer == 'n') ? 0 : 1;

    if (ifcomputer[0]) {
    do {
        getdata(5, 0, "請輸入您要讓電腦猜的數字: ",
            yournum, sizeof(yournum), LCECHO);
    } while (!legal(atoi(yournum)));
    move(8, 25);
    outs("電腦猜");
    initcomputer(flag);
    }
    move(8, 55);
    outs("你猜");
    while (((!computerwin || !youwin) && count < max_guess && (ifcomputer[0])) 
        || (!ifcomputer[0] && count < max_guess && !youwin)) {
    if (!computerwin && ifcomputer[0]) {
        ++c_count;
        if (computer(atoi(yournum), c_count, flag, n))
        computerwin = 1;
    }
    move(20, 55);
    prints("第 %d/%d 次機會 ", count + 1, max_guess);
    if (!youwin) {
        ++count;
        if (guess_play(get_data(data, count), answer, count))
        youwin = 1;
    }
    }
    move(17, 33);
    if (ifcomputer[0]) {
    if (count > c_count) {
        outs("  你輸給電腦了");
    } else if (count < c_count) {
        outs("真厲害, 讓你猜到囉");
    } else {
        prints("真厲害, 和電腦打成平手了");
    }
    pressanykey();
    return 1;
    }
    if (youwin) {
    if (count < 5) {
        outs("真厲害!");
    } else if (count > 5) {
        outs("唉, 太多次才猜出來了");
    } else {
        outs("五次猜出來, 還可以~");
        move(18, 35);
        clrtoeol();
    }
    pressanykey();
    return 1;
    }
    move(17, 32);
    prints("嘿嘿 標準答案是 %s ", answer);
    move(18, 32);
    outs("下次再來吧");
    pressanykey();
    return 1;
}