#include <stdio.h>
#include <windows.h>
#include <stdlib.h>
#include <math.h>
#include <conio.h>
#include <string.h>
#include <time.h>
void food();
void show();
void move();
void turn();
void check();
void ini();
void calculate();
int movable[4] = { 0, 1, 1, 0
 
 };  //陣列的下標表示方向，0123分別表示上下左右，下同
int distance[4] = { 9999, 9999, 9999, 9999 };
int  dy[4] = { 0, 1, 0, -1 };
int  dx[4] = { -1, 0, 1, 0 };
int sum = 5;  //蛇總長度
int over = 0;
int speed;   
int foodx, foody;  //食物的座標
char map[17][17];

struct snake {
    int x, y;  //身體座標 
    int dir;  //方向（只有蛇頭的方向是有用的） 
} A[100];

void
 
 ini() {  //初始化介面
    speed = 500;
    over = 0;
    sum = 5;
    movable[0] = 0; movable[1] = 1; movable[2] = 1; movable[3] = 0;
    distance[0] = 9999; distance[1] = 9999; distance[2] = 9999; distance[3] = 9999;
    int i, j;
    for (i = 0; i < 100; i++) {  //初始化蛇頭和蛇身的資料
        A[i].dir = 0;
        A[i].x = 0
 
;
        A[i].y = 0;
    }
    A[0].x = 1; A[0].y = 1;  //地圖左上角設定一條長度為5的蛇
    A[1].x = 1; A[1].y = 2;
    A[2].x = 1; A[2].y = 3;
    A[3].x = 1; A[3].y = 4;
    A[4].x = 1; A[4].y = 5; A[4].dir = 1;
    srand(time(0));
    for (i = 0; i < 17; i++) {  //設定地圖
        for (j = 0; j < 17; j++) {
            map[i][j] = '*';
        }
    }
    for (i = 1; i < 16; i++) {
        for (j = 1; j < 16; j++) {
            map[i][j] = ' ';
        }
    }
    //map[6][5] = '*'; map[6][6] = '*'; map[6][7] = '*';
    //map[7][5] = '*'; map[7][7] = '*';
    map[A[4].x][A[4].y] = 'H';  //設定蛇頭
    for (i = 0; i < sum - 1; i++) {  //設定蛇身
        map[A[i].x][A[i].y] = 'X';
    }
    food();
    calculate();
}


void show() {  //顯示介面 
    int i, j, x, y;
    for (i = 0; i < 17; i++) {  //顯示介面
        for (j = 0; j < 17; j++) {
            printf("%c", map[i][j]);
        }
        printf("\n");
    }
    while (1) {
        Sleep(speed);  //介面重新整理速度
        turn();
        move();
        if (over) {  //設定蛇死掉後可以進行的操作 
            while (1) {
                char ch = _getch();
                if (ch == 113) {  //輸入‘q’結束 
                    return;
                }
                else if (ch == 114) {  //輸入‘r’重新開始 
                    ini();
                    break;
                }
            }
        }
        system("cls");  //清屏 
        for (i = 0; i < 17; i++) {  //重新顯示介面
            for (j = 0; j < 17; j++) {
                printf("%c", map[i][j]);
            }
            printf("\n");
        }
        calculate(); //計算並記錄蛇頭與食物距離
    }
}

void food() {  //生成食物
    int x, y;
    while (1) {
        x = (int)(15 * rand() / (RAND_MAX + 1.0));  //隨機產生一組食物座標
        y = (int)(15 * rand() / (RAND_MAX + 1.0));
        if (map[x][y] == ' ') {  //如果是空格則在該處生成食物
            map[x][y] = 'O';
            foodx = x;  //記錄食物座標 
            foody = y;
            break;
        }
    }
}


void move() {  //蛇移動
    int i, x, y;
    int t = sum;  //t記錄當前蛇總長度 
    check();  //移動前檢查按當前方向移動一步後的情況
    if (t == sum) {  //沒有吃到蘋果
        for (i = 0; i < sum - 1; i++) {
            if (i == 0) {  //蛇尾座標處變成空格，把蛇尾座標變成前一個蛇身的座標 
                map[A[i].x][A[i].y] = ' ';
                A[i].x = A[i + 1].x;
                A[i].y = A[i + 1].y;
            }
            else {  //每個蛇身座標都變為它前一個蛇身的座標
                A[i].x = A[i + 1].x;
                A[i].y = A[i + 1].y;
            }
            map[A[i].x][A[i].y] = 'X';  //把地圖上蛇身座標處的字元設定成‘X’
        }
        A[sum - 1].x = A[sum - 1].x + dx[A[sum - 1].dir];  //蛇頭按當前方向移動一格 
        A[sum - 1].y = A[sum - 1].y + dy[A[sum - 1].dir];
        map[A[sum - 1].x][A[sum - 1].y] = 'H';  //把地圖上蛇頭座標處的字元設定成‘H’
    }
    else {  //吃到蘋果（sum會加1）
        map[A[sum - 2].x][A[sum - 2].y] = 'X';  //把地圖上原蛇頭座標處的字元設定成‘X’
        A[sum - 1].x = A[sum - 2].x + dx[A[sum - 2].dir];  //新蛇頭的座標是原蛇頭沿當前方向移動一格後的座標 
        A[sum - 1].y = A[sum - 2].y + dy[A[sum - 2].dir];
        A[sum - 1].dir = A[sum - 2].dir;  //新蛇頭方向為原蛇頭的方向
        map[A[sum - 1].x][A[sum - 1].y] = 'H';  //把地圖上蛇頭座標處的字元設定成‘H’
        food();
    }
    for(i = 0; i < 4; i++) {  //記錄下能走的方向
        x = A[sum - 1].x + dx[i];
        y = A[sum - 1].y + dy[i];
        if(map[x][y] == ' ' || map[x][y] == 'O') {
            movable[i] = 1;  //能走就把對應方向的值設定為1 
        } else {
            if(x != A[0].x || y != A[0].y) {
                movable[i] = 0;  //不能走就把對應方向的值設定為0 
            } else {
                movable[i] = 1;
            }
        }
    }
}

void check() {  //檢查是否死亡或者吃到食物
    int x, y, i, j;
    x = A[sum - 1].x + dx[A[sum - 1].dir];  //記錄按當前方向移動一格後蛇頭的座標 
    y = A[sum - 1].y + dy[A[sum - 1].dir];
    if (map[x][y] == '*' || map[x][y] == 'X') {  //如果地圖上該座標處字元為‘*’或‘X’就死亡 
        if (x != A[0].x || y != A[0].y) {  //蛇尾除外 
            map[8][4] = 'G'; map[8][5] = 'A'; map[8][6] = 'M'; map[8][7] = 'E';  //輸出“GAME OVER” 
            map[8][9] = 'O'; map[8][10] = 'V'; map[8][11] = 'E'; map[8][12] = 'R';
            map[8][8] = ' ';
            system("cls");
            for (i = 0; i < 17; i++) {
                for (j = 0; j < 17; j++) {
                    printf("%c", map[i][j]);
                }
                printf("\n");
            }
            printf("Input 'r' to restart\nInput 'q' to quit\n");
            over = 1;
        }
    }
    else if (map[x][y] == 'O') {  //吃到蘋果 
        sum++;  //蛇身總長加1 
        speed = ((600 - sum * 20)>100) ? (600 - sum * 20) : 100; //速度加快 
    }
}

void turn() {  //轉彎
    int i, k = 0;
    for(i = 1; i < 4; i++) {  //找到走一步後離食物距離最短的方向
        if(distance[k] > distance[i]) {
            k = i;
        }
    }
    switch (k) {  //把把蛇頭方向改為該方向 
        case 0: A[sum - 1].dir = (A[sum - 1].dir == 2)?2:0; break;
        case 1: A[sum - 1].dir = (A[sum - 1].dir == 3)?3:1; break;
        case 2: A[sum - 1].dir = (A[sum - 1].dir == 0)?0:2; break;
        case 3: A[sum - 1].dir = (A[sum - 1].dir == 1)?1:3; break;
    }
}

void calculate() {  //計算並記錄蛇頭與食物距離
    int i = 0, x, y;
    for(i = 0; i < 4; i++) {
        if(movable[i] == 1) {  //如果該方向能走，則記錄下沿該方向走一步後與食物的距離 
            x = A[sum - 1].x + dx[i];
            y = A[sum - 1].y + dy[i];
            distance[i] = abs(foodx-x)+abs(foody-y);
        } else {  //如果不能走則把距離設定為9999 
            distance[i] = 9999;
        }
    }
}

int main() {
    printf("智慧蛇版本\n");
    printf("按任意鍵開始\n");
    char ch = _getch();
    system("cls");
    ini();
    show();
    return 0;
}