PolinaMain

package com.company;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Scanner;

public class Main {
    public static void main(String[]args) {
        Scanner in = new Scanner(System.in);
        int choice = 0;
        Set buffer;
        Set first;
        Set second;

        System.out.print("Вы можете:\n" +
                "1. Создать множества\n" +
                "2. Считать множества из файла\n" +
                "Введите желаемый вариант: ");
        choice = in.nextInt();

        if (choice == 1) {
            first = new Set();
            second = new Set();
            System.out.println("\nСоздано 2 множества");
        } else {
            first = new Set();
            second = new Set();

            try {
                BufferedReader reader = new BufferedReader(new FileReader("First.txt"));
                String line_buffer;
                while ((line_buffer = reader.readLine()) != null) {
                    first.includeElement(Integer.parseInt(line_buffer));
                }
                reader.close();

                reader = new BufferedReader(new FileReader("Second.txt"));
                while ((line_buffer = reader.readLine()) != null) {
                    second.includeElement(Integer.parseInt(line_buffer));
                }
                reader.close();
            } catch (IOException e) {
                System.out.println("Error" + e);
            }

            System.out.println("Файлы считаны!");
            first.output();
            second.output();
            System.out.println();
        }

        while (true) {
            System.out.print("\nВы можете:\n" +
                    "1. Задать пустое множество\n" +
                    "2. Включить элемент\n" +
                    "3. Исключить элемент\n" +
                    "4. Проверить элемент на вхождение\n" +
                    "5. Проверить равенство двух множеств\n" +
                    "6. Объединить множества\n" +
                    "7. Пересечь множества\n" +
                    "8. Составить разность множеств\n" +
                    "9. Записать множества в файл\n" +
                    "10. Выйти\n" +
                    "Введите желаемый вариант: ");
            choice = in.nextInt();

            switch (choice) {
                case 1:
                    System.out.print("\nНад каким множеством хотите совершить операцию?\n" +
                            "1. Над первым\n" +
                            "2. Над вторым\n" +
                            "Введите желаемый вариант: ");
                    choice = in.nextInt();

                    if (choice == 1) {
                        first.zeroSet();
                        System.out.println("\nПервое множество задано как пустое");
                    } else {
                        second.zeroSet();
                        System.out.println("\nВторое множество задано как пустое");
                    }
                    break;
                case 2:
                    System.out.print("\nНад каким множеством хотите совершить операцию?\n" +
                            "1. Над первым\n" +
                            "2. Над вторым\n" +
                            "Введите желаемый вариант: ");
                    choice = in.nextInt();
                    if (choice == 1) {
                        while (choice == 1) {
                            System.out.print("\nВведите номер элемента который хотите включить в множество: ");
                            first.includeElement(in.nextInt());

                            first.output();

                            System.out.print("\nВы можете:\n" +
                                    "1. Повторить операцию\n" +
                                    "2. Выйти в меню\n" +
                                    "Введите желаемый вариант: ");
                            choice = in.nextInt();
                        }
                    } else {
                        while (choice == 2) {
                            System.out.print("\nВведите номер элемента который хотите включить в множество: ");
                            second.includeElement(in.nextInt());

                            second.output();

                            System.out.print("Вы можете:\n" +
                                    "1. Выйти в меню\n" +
                                    "2. Повторить операцию\n" +
                                    "Введите желаемый вариант: ");
                            choice = in.nextInt();
                        }
                    }

                    break;
                case 3:
                    System.out.print("\nНад каким множеством хотите совершить операцию?\n" +
                            "1. Над первым\n" +
                            "2. Над вторым\n" +
                            "Введите желаемый вариант: ");
                    choice = in.nextInt();
                    if (choice == 1) {
                        while (choice == 1) {
                            System.out.print("\nВведите номер элемента который хотите исключить из множества: ");
                            first.excludeElement(in.nextInt());

                            first.output();

                            System.out.print("\nВы можете:\n" +
                                    "1. Повторить операцию\n" +
                                    "2. Выйти в меню\n" +
                                    "Введите желаемый вариант: ");
                            choice = in.nextInt();
                        }
                    } else {
                        while (choice == 2) {
                            System.out.print("\nВведите номер элемента который хотите исключить из множества: ");
                            second.excludeElement(in.nextInt());

                            second.output();

                            System.out.print("Вы можете:\n" +
                                    "1. Выйти в меню\n" +
                                    "2. Повторить операцию\n" +
                                    "Введите желаемый вариант: ");
                            choice = in.nextInt();
                        }
                    }

                    break;
                case 4:
                    System.out.print("\nНад каким множеством хотите совершить операцию?\n" +
                            "1. Над первым\n" +
                            "2. Над вторым\n" +
                            "Введите желаемый вариант: ");
                    choice = in.nextInt();

                    if (choice == 1) {
                        System.out.print("Введите номер проверяемого элемента: ");
                        first.checkElement(in.nextInt());
                    } else {
                        System.out.print("Введите номер проверяемого элемента: ");
                        second.checkElement(in.nextInt());
                    }
                    break;
                case 5:
                    System.out.println();
                    first.equally(second);
                    break;
                case 6:
                    buffer = new Set(first);
                    buffer.association(second);
                    buffer.output();
                    break;
                case 7:
                    buffer = new Set(first);
                    buffer.intersection(second);
                    buffer.output();
                    break;
                case 8:
                    System.out.print("\nНад каким множеством хотите совершить операцию?\n" +
                            "1. Разность первого и второго\n" +
                            "2. Разность второго и первого\n" +
                            "Введите желаемый вариант: ");
                    choice = in.nextInt();

                    if (choice == 1) {
                        first.difference(second);
                        first.output();
                    } else {
                        second.difference(first);
                        second.output();
                    }
                    break;
                case 9:
                    try {
                        FileWriter writer = new FileWriter("First.txt", false);
                        writer.write(first.toString());
                        writer.close();

                        writer = new FileWriter("Second.txt", false);
                        writer.write(second.toString());
                        writer.close();
                    } catch (IOException e) {
                        System.out.println("Error" + e);
                    }

                    System.out.println("Файлы успешно записаны!");
                    break;
                case 10:
                    System.exit(0);
                    return;
                default:
                    break;
            }
        }

    }
}