Cod sursa(job #2771878)

Utilizator Gabryel9898Bizdoc Vasile Gabriel Gabryel9898 Data 29 august 2021 22:15:09
Problema Jocul Flip Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.03 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <numeric>

class Board {
private:
    std::vector<std::vector<int>> board;
public:
    explicit Board(std::vector<std::vector<int>> matrix) {
        board = std::move(matrix);
    }

    Board(const Board &p1) {
        board = p1.board;
//        board.assign(p1.board.begin(), p1.board.end());
    }

    void flip_line(int number) {
        for (auto &col: board[number]) {
            col *= -1;
        }
    }

    void flip_column(int number) {
        for (auto &row: board) {
            row[number] *= -1;
        }
    }

    void print_matrix() {
        std::cout << "\n<------------------->\n";

        for (const auto &row: board) {
            for (const auto &col: row) {
                std::cout << col << " ";
            }
            std::cout << '\n';
        }
    }

    int get_sum() {
        int sum = 0;

        for (const auto &row: board) {
            for (const auto &col: row) {
                sum += col;
            }
        }

        return sum;
    }

    int sum_line(int number) {
        int sum = 0;

        for (const auto &col: board[number]) {
            sum += col;
        }

        return sum;
    }

    int sum_col(int number) {
        int sum = 0;

        for (const auto &row: board) {
            sum += row[number];
        }

        return sum;
    }

    void print_sum() {
        std::cout << '\n' << get_sum() << '\n';
    }
};

int flip_algo() {
    std::string input_file_name("../flip.in");
    std::string output_file_name("../flip.out");

    std::ifstream input_file;
    input_file.open(input_file_name);

    if (!input_file.is_open()) {
        std::cerr << "Could not open the file: '" << input_file_name << "'" << std::endl;
        return EXIT_FAILURE;
    }

    int M, N;
    input_file >> M >> N;

    auto matrix = std::vector<std::vector<int>>(M);

    for (auto &row: matrix) {
        row = std::vector<int>(N);

        for (auto &col: row) {
            input_file >> col;
        }
    }

    input_file.close();

    auto game = new Board(matrix);

    int limit = 1 << M;

    int max_sum = game->get_sum();
    for (int i = 0; i < limit; ++i) {
//        std::cout << "new game \n";

        auto tmp_game = new Board(*game);;

        for (int j = 0; j < M && 1 << j <= i; ++j) {
            if (i & (1 << j)) {
                tmp_game->flip_line(j);
//                std::cout << j << " ";
            }
        }
//        std::cout << std::endl;

        for (int k = 0; k < N; ++k) {
            if (tmp_game->sum_col(k) < 0) {
                tmp_game->flip_column(k);
            }
        }

        int new_sum = tmp_game->get_sum();
        if (new_sum > max_sum) {
            max_sum = new_sum;
        }
    }

    std::ofstream out;
    out.open(output_file_name);

    if (!out.is_open()) {
        std::cerr << "Could not open the file: '" << output_file_name << "'" << std::endl;
        return EXIT_FAILURE;
    }

    out << max_sum;

    out.close();
    return 0;
}

int main() {
    flip_algo();
}