Игра - сокровища пирата Ларри

#include <iostream>
#include <fstream>
#include <string>
#include <set>
#include <iterator>
#pragma warning(disable:4996)
 
using namespace std;
//Структура с инфорамации о тестовых данных
struct Keys
{
    
    int iKey_chest_num[2];  // кол-во ключей и сундуков, первая строка
    int *iKey_val;          //Значения начальных ключей 
    int **Chest_info = new int*[30]; //двухмерный динамический массив с информацией о самих сундуках 
                                    // Каким ключем открывается, сколько в нем ключей и какого типа они
 
};
 
 
void main()
{
    // Переменные и подключение файла, чтение данных
    int *tmp_arr = new int[40];
    int test_c = 0;
    int size = 0;
    int k = 0;
    setlocale(LC_ALL, "rus");
    ifstream fin("D-small-practice.in", ios_base::in); 
    ofstream fout("D_small_solution.in", ios_base::trunc);
    char buff[256];
    fout.close();
    fin.close();
    fin.open("D-small-practice.in");
    fout.open("D_small_solution.in");
    if (!fin.is_open())
    {
        cout << "Can't open file" << endl;
        system("Pause");
        return;
    }
    else
    {
        fin.getline(buff, 50);
        test_c = atoi(buff);
        
        cout << "Test cases = " << test_c << endl;
    
    }
    // циклично и динамично считываем данные с файла заносим в массив структур
    Keys *full_info = new Keys[test_c];
    for (k = 0; k < test_c;k++)
    {
 
        // Получаем количество начальных ключей и сундуков
        fin.getline(buff, 50);
        char * pch;
        pch = strtok(buff, " ");
        int i = 0;
        while (pch != NULL)
        {
            full_info[k].iKey_chest_num[i] = atoi(pch);
            i++;
            pch = strtok(NULL, " ");
        }
        for (int i = 0; i < 2; i++)
            cout << full_info[k].iKey_chest_num[i] << " ";
        cout << endl;
 
 
 
        //Получаем значения начальных ключей в динамический масив
        fin.getline(buff, 50);
        pch = strtok(buff, " ");
        i = 0;
        while (pch != NULL)
        {
            tmp_arr[i] = atoi(pch);
            i++;
            //cout << i << " ";
            pch = strtok(NULL, " ");
        }
        size = full_info[k].iKey_chest_num[0];
        full_info[k].iKey_val = new int[size];
        for (int i = 0; i < size; i++)
        {
            full_info[k].iKey_val[i] = tmp_arr[i];
            cout << full_info[k].iKey_val[i] << " ";
        }
        cout << endl;
 
 
 
        for (int z = 0; z < full_info[k].iKey_chest_num[1] + 1; z++)
        {
            full_info[k].Chest_info[z] = new int[];
        }
 
        
        // Получаем данные о самых сундуках, все вносится в двухмерный массив
        // full_info[k].chest_info[j]
        for (int j = 0; j < full_info[k].iKey_chest_num[1]; j++)
        {
            delete []full_info[k].Chest_info[j];
            fin.getline(buff, 50);
            pch = strtok(buff, " ");
            int i = 0;
            while (pch != NULL)
            {
                
                tmp_arr[i] = atoi(pch);
                int tmp = tmp_arr[i];                       
                pch = strtok(NULL, " ");
                i++;
            }
            
            size = tmp_arr[1] + 2;
            full_info[k].Chest_info[j] = new int[size];
            for (int i = 0; i < size; i++)
            {
                full_info[k].Chest_info[j][i] = tmp_arr[i];
                cout << full_info[k].Chest_info[j][i] << " ";
            }
            cout << endl;
 
        }
        cout << endl;
        
        
    } 
    
    
    //разбираем ключи и возможности открытия сундуков 
        k = 0;
    multiset<int> Keys;
    std::set<int>::iterator it;
    int max_count = 0, Keys_num = 0, num = 0, op_chest = 0, i_tmp = 0;
    string str;
    char c_tmp[50];
    int i = 0, j = 0;
    
    // циклично для каждого элемента массива структуры делаем разбор
    for (k = 0; k < test_c; k++)
    {
        //k = 9;
        Keys.erase(Keys.begin(),Keys.end());
        i = 0;
        j = 0;
        op_chest = 0;
        cout << "CASE# " << k + 1 << ": ";
 
        for (i = 0; i < full_info[k].iKey_chest_num[0]; i++)
        {
            Keys.insert(full_info[k].iKey_val[i]);
        }
 
        for ( i = 0; i < full_info[k].iKey_chest_num[1]; i++)
        {
            Keys_num += full_info[k].Chest_info[i][1];
        }
        Keys_num += full_info[k].iKey_chest_num[0]; // кол-во ключей в целом за цикл
 
        if (Keys_num < full_info[k].iKey_chest_num[1]) // Предпроверка, если всего сундуков больше чем ключей то все открыть не получиться
        {
            cout << "Imposible" << endl;
        }
        //Используется метод жадного обхвата, ищем максимальное кол-во ключей в каком либо сундуке,
        //который можем изначально открыть
        for (j = 0; j < full_info[k].iKey_chest_num[1]; j++)
        {
            max_count = 0;
            num = 0;
            for (i = 0; i < full_info[k].iKey_chest_num[1]; i++)
            {
                
                if (Keys.find(full_info[k].Chest_info[i][0]) != Keys.end())
                if (full_info[k].Chest_info[i][1] >= max_count)
                {
                    max_count = full_info[k].Chest_info[i][1];
                    num = i;
                }
 
            }
            // найдя максимальное кол-во ключей открываем сундук
            if (Keys.find(full_info[k].Chest_info[num][0]) != Keys.end())
            {
 
                for ( i = 2; i < full_info[k].Chest_info[num][1] + 2; i++)
                {
                    Keys.insert(full_info[k].Chest_info[num][i]);
 
                }
                it = Keys.find(full_info[k].Chest_info[num][0]);
                Keys.erase(it); // удаляем использованный ключ
                full_info[k].Chest_info[num][0] = -1; //обозначаем что сундук открыт 
                full_info[k].Chest_info[num][1] = -1; // и больше по нему не проводить поиск
                op_chest++; 
                itoa(num + 1, c_tmp, 10);
                str += c_tmp;
                str += " ";
 
            }
        // Если открыты все сундуки выводим последовательность            
        }
        if (op_chest == full_info[k].iKey_chest_num[1])
        {
            fout << str;
            cout << str;
            
        }
        else
        // Если нет, значит для данного тестового набора открыть все сундуки не возможно
        {
            fout << "Impossible";
            cout << "Impossible";
        }
        
        str = "";
        fout << endl;
        cout << endl;
    }
    // Задержка экрана, закрытие файла, удаление динамических переменных
    cout << endl;
    fout << endl;
    system("pause");
    fin.close();
    fout.close();
    for (int z = 0; z < full_info[k].iKey_chest_num[1] + 1; z++)
    {
        delete[] full_info[k].Chest_info[z];
    }
    delete [] full_info;
    delete [] tmp_arr;
 
    return;
}