/*
Rafay Ahmad
23I-2526
*/
#include <iostream>
using namespace std;

class Minesweeper{
    char** grid;
    int* mineLocations;
    int size;
    int mines;
    int opened;

public:

    Minesweeper(int size, int mines) : size(size), mines(mines), opened(0){
        srand(time(0));
        grid = new char*[size];
        for(int i=0;i<size;i++){
            grid[i] = new char[size];
            for(int j=0;j<size;j++) grid[i][j] = '#';
        }
        mineLocations = new int[mines];
        setMines();
    }
    ~Minesweeper(){
        for(int i=0;i<size;i++) delete[] grid[i];
        delete[] grid;
        delete[] mineLocations;
    }

    void reset(){
        for(int i=0;i<size;i++) for(int j=0;j<size;j++) grid[i][j] = '#';
        setMines();
    }
    void revealAll(){
        for(int i=0;i<mines;i++){
            grid[mineLocations[i]/7][mineLocations[i]%7] = '*';
        }
    }
    bool revealCell(int row, int col, bool checkMine){
        if(checkMine){
            for(int i=0;i<mines;i++){
                if(mineLocations[i]/size == row && mineLocations[i]%7 == col){
                    grid[row][col] = '*';
                    revealAll();
                    return false;
                }   
            }
        }
        grid[row][col] = checkSurrounding(row, col) + '0';
        opened++;
        if(grid[row][col] == '0'){
            if(row<size && col<size) revealCell(row+1, col+1, false);
            if(col<size) revealCell(row, col+1, false);
            if(row>0 && col<size) revealCell(row-1, col+1, false);
            if(row>0) revealCell(row-1, col, false);
            if(row>0 && col>0) revealCell(row-1, col-1, false);
            if(row<size && col<size) revealCell(row, col-1, false);
            if(row<size && col<size) revealCell(row+1, col-1, false);
            if(row<size && col<size) revealCell(row+1, col, false);
        }
        return true;
    }
    void flagCell(int row, int col){
        grid[row][col] = 'P';
    }
    int checkSurrounding(int row, int col){
        int count = 0;
        for(int i=0;i<mines;i++){
            if(row<size && col<size && mineLocations[i]/size == row+1 && mineLocations[i]%7 == col+1) count++;
            if(col<size && mineLocations[i]/size == row && mineLocations[i]%7 == col+1) count++;
            if(row>0 && col<size && mineLocations[i]/size == row-1 && mineLocations[i]%7 == col+1) count++;
            if(row>0 && mineLocations[i]/size == row-1 && mineLocations[i]%7 == col) count++;
            if(row>0 && col>0 && mineLocations[i]/size == row-1 && mineLocations[i]%7 == col-1) count++;
            if(col>0 && mineLocations[i]/size == row && mineLocations[i]%7 == col-1) count++;
            if(row<size && col>0 && mineLocations[i]/size == row+1 && mineLocations[i]%7 == col-1) count++;
            if(row<size && mineLocations[i]/size == row+1 && mineLocations[i]%7 == col) count++;
        }
        return count;
    }
    bool checkWin(){
        return opened == ((size*size) - mines);
    }
    void setMines(){
        for(int i=0;i<mines;i++){
            bool valid;
            do{
                valid = true;
                mineLocations[i] = rand()%(size*size);
                for(int j=0;j<i;j++){
                    if(mineLocations[j] == mineLocations[i]){
                        valid = false;
                        break;
                    }
                }
            }while(!valid);
        }
    }
    void displayGrid(){
        cout<<"  ";
        for(int i=0;i<size;i++) cout<<i<<" ";
        cout<<endl;
        for(int i=0;i<size;i++){
            cout<<i<<" ";
            for(int j=0;j<size;j++) cout<<grid[i][j]<<" ";
            cout<<endl;
        }
        cout<<endl;

        for(int i=0;i<mines;i++){
            cout<<mineLocations[i]<<" ";
        }
    }




};

int main(){

    int row, col;
    char c;
    int size, mines;
    cout<<"Enter size of the game board: ";
    cin>>size;
    cout<<"Enter no Of mines: ";
    cin>>mines;
    Minesweeper game(size, mines);

    while(!game.checkWin()){
        game.displayGrid();
        cout<<"Enter row: ";
        cin>>row;
        cout<<"Enter col: ";
        cin>>col;
        cout<<"Enter F to flag or R to reveal: ";
        cin>>c;

        if(row<size && col<size && (c == 'F' || c == 'R')){
            if(c == 'R' && !game.revealCell(row, col, true)){
                game.displayGrid();
                cout<<"You Lost\n";
                break;
            }
            else if(c == 'F') game.flagCell(row, col);
        }
        else cout<<"Invalid input\n";

    }
    

    return 0;
}