Dataset-Netflix-Java/Main.java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.*;

public class Main {

    private static void lerDadosDeArquivo(Scanner entrada, BST arvoreBST, AVL arvoreAVL) {
        int incompletecount = 0;

        arvoreBST.cleanup();
        arvoreAVL.cleanup();

        System.out.print("Insira o nome do arquivo CSV: ");
        String csvFileName = entrada.nextLine();
        try (InputStream inputStream = Main.class.getClassLoader().getResourceAsStream(csvFileName);
                BufferedReader br = new BufferedReader(new InputStreamReader(inputStream))) {
            String line;
            br.readLine();

            while ((line = br.readLine()) != null) {
                String[] fields = splitCSVLine(line);

                boolean empty = false;
                for (String field : fields) {
                    if (field.isEmpty()) {
                        empty = true;
                    }
                }
                if (empty) {
                    incompletecount++;
                } else if (fields.length == 15) {
                    ProgramaNetflix programa = new ProgramaNetflix(
                            fields[0], // id
                            fields[1], // título
                            fields[2], // showType
                            fields[3], // descrição
                            Integer.parseInt(fields[4]), // releaseYear
                            fields[5], // ageCertification
                            Integer.parseInt(fields[6]), // runtime
                            fields[7], // generos
                            fields[8], // productionCountries
                            Float.parseFloat(fields[9]), // temporadas
                            fields[10], // imdbId
                            Float.parseFloat(fields[11]), // imdbScore
                            Float.parseFloat(fields[12]), // imdbVotes
                            Float.parseFloat(fields[13]), // tmdbPopularity
                            Float.parseFloat(fields[14]) // tmdbScore
                    );

                    // Inserir o objeto nas árvores BST e AVL
                    arvoreBST.insert(programa);
                    arvoreAVL.insert(programa);
                }

            }
        } catch (IOException e) {
            System.out.println("Erro ao ler o arquivo " + csvFileName);
        }
        System.out.println(incompletecount + " linhas ignoradas devido à falta de informações.");
    }

    private static String[] splitCSVLine(String line) {
        List<String> fields = new ArrayList<>();
        boolean aspas = false;
        StringBuilder atual = new StringBuilder();
        for (char c : line.toCharArray()) {
            if (c == '"') {
                aspas = !aspas;
            } else if (c == ',' && !aspas) {
                fields.add(atual.toString().trim());
                atual.setLength(0);
            } else {
                atual.append(c);
            }
        }
        fields.add(atual.toString().trim());
        return fields.toArray(new String[0]);
    }

    public static void main(String[] args) {
        BST arvoreBST = new BST();
        AVL arvoreAVL = new AVL();
        int opcao;

        String id = null;
        String titulo;
        String show_type;
        String descricao;
        int release_year;
        String age_certification;
        int runtime;
        String generos;
        String production_countries;
        float temporadas;
        String imdb_id;
        double imdb_score;
        float imdb_votes;
        double tmdb_popularity;
        double tmdb_score;
        Scanner entrada = new Scanner(System.in);

        do {
            System.out.println("\nMenu:");
            System.out.println("1 - Ler dados do arquivo");
            System.out.println("2 - Análise de dados");
            System.out.println("3 - Inserir Programa");
            System.out.println("4 - Buscar Programa");
            System.out.println("5 - Remover Programa");
            System.out.println("6 - Exibir a altura das árvores");
            System.out.println("7 - Salvar dados em arquivo");
            System.out.println("8 - Sair");
            opcao = entrada.nextInt();
            entrada.nextLine();

            switch (opcao) {
                case 1: // FUNCIONANDO
                    lerDadosDeArquivo(entrada, arvoreBST, arvoreAVL);
                    break;
                case 2: // FUNCIONANDO
                    // 1
                    System.out.print("10 países com a melhor pontuação no IMDB\n");
                    arvoreAVL.pontuacaoMediaPorPais();

                    System.out.print("\nPressione ENTER para continuar...\n");
                    entrada.nextLine();

                    // 2
                    System.out.print("Quantidade de filmes por década\n");
                    arvoreAVL.filmesPorDecada();

                    System.out.print("\nPressione ENTER para continuar...\n");
                    entrada.nextLine();

                    // 3
                    System.out.print("Duração média de um episódio de uma série por classificação etária\n");
                    arvoreAVL.mediaDuracaoPorClassificacao();

                    System.out.print("\nPressione ENTER para continuar...\n");
                    entrada.nextLine();

                    // 4
                    System.out.print("Percentual de títulos por classificação etária\n");
                    arvoreAVL.percentualPorClassificacaoEtaria();

                    System.out.print("\nPressione ENTER para continuar...\n");
                    entrada.nextLine();

                    // 5
                    System.out.print("Países com mais títulos\n");
                    arvoreAVL.paisesComMaisTitulos();
                    break;

                case 3: // FUNCIONANDO
                    Scanner sc = new Scanner(System.in);

                    System.out.println("Insira o ID do título");
                    id = sc.nextLine();

                    System.out.println("Insira o título do título");
                    titulo = sc.nextLine();

                    System.out.println("Insira o tipo (SHOW | MOVIE)");
                    show_type = sc.nextLine();

                    System.out.println("Insira a descrição do título");
                    descricao = sc.nextLine();

                    System.out.println("Insira o ano de lançamento do título");
                    release_year = sc.nextInt();

                    System.out.println("Insira a certificação parental do título");
                    age_certification = sc.nextLine();
                    age_certification = sc.nextLine();

                    System.out.println("Insira o tempo (em minutos) do título");
                    runtime = sc.nextInt();

                    System.out.println("Insira os gêneros do título");
                    generos = sc.nextLine();
                    generos = sc.nextLine();

                    System.out.println(
                            "Insira os países de produção do título (No formato ['Country Code', 'Country Code', 'Country Code'] )");
                    production_countries = sc.nextLine();

                    System.out.println("Insira a quantidade de temporadas do título");
                    temporadas = sc.nextFloat();

                    System.out.println("Insira o imdb id do título");
                    imdb_id = sc.nextLine();
                    imdb_id = sc.nextLine();

                    System.out.println("Insira o imdb score do título");
                    imdb_score = sc.nextDouble();

                    System.out.println("Insira o imdb votes do título");
                    imdb_votes = sc.nextFloat();

                    System.out.println("Insira o tmdb popularity do título");
                    tmdb_popularity = sc.nextDouble();

                    System.out.println("Insira o tmdb_score do título");
                    tmdb_score = sc.nextDouble();

                    ProgramaNetflix programa = new ProgramaNetflix(id, titulo, show_type, descricao, release_year,
                            age_certification, runtime, generos, production_countries, temporadas, imdb_id, imdb_score,
                            imdb_votes, tmdb_popularity, tmdb_score);

                    if (arvoreAVL.search(id).isEmpty() || arvoreBST.search(id).isEmpty()) {
                        arvoreAVL.insert(programa);
                        arvoreBST.insert(programa);
                        System.out.println("Valor na Árvore AVL: " + arvoreAVL.search(id));
                        System.out.println("Valor na Árvore BST: " + arvoreBST.search(id));
                    } else {
                        System.out.println("Programa já existe em pelo menos uma árvore");
                        System.out.println("Deseja sobrescrever? (S/N)");
                        String resposta = sc.nextLine();
                        resposta = sc.nextLine();

                        if (resposta.equalsIgnoreCase("S")) {
                            arvoreAVL.remove(id);
                            arvoreBST.remove(id);
                            arvoreAVL.insert(programa);
                            arvoreBST.insert(programa);
                            System.out.println("Valor na Árvore AVL: " + arvoreAVL.search(id));
                            System.out.println("Valor na Árvore BST: " + arvoreBST.search(id));
                        }
                    }
                    break;

                case 4: // FUNCIONANDO
                    System.out.print("Informe o ID do programa: ");
                    String programaId = entrada.nextLine().trim().toLowerCase();

                    // Buscar na AVL
                    long startTimeAVL = System.nanoTime();
                    String nomeSerieAVL = arvoreAVL.search(programaId);
                    long endTimeAVL = System.nanoTime();
                    long durationAVL = (endTimeAVL - startTimeAVL);

                    // Buscar na BST
                    long startTimeBST = System.nanoTime();
                    String nomeSerieBST = arvoreBST.search(programaId);
                    long endTimeBST = System.nanoTime();
                    long durationBST = (endTimeBST - startTimeBST);

                    // Exibir as estatísticas da AVL
                    System.out.println("\nNodes tocados pela arvore AVL: " + arvoreAVL.getCount());
                    System.out.println("Tempo de execução da busca na AVL: " + durationAVL + " nanossegundos");
                    if (nomeSerieAVL.isEmpty()) {
                        System.out.println("Não encontrado na árvore AVL!");
                    } else {
                        System.out.println("Encontrado na árvore AVL: " + nomeSerieAVL);
                    }

                    // Exibir as estatísticas da BST
                    System.out.println("\nNodes tocados pela arvore BST: " + arvoreBST.getCount());
                    System.out.println("Tempo de execução da busca na BST: " + durationBST + " nanossegundos");
                    if (nomeSerieBST.isEmpty()) {
                        System.out.println("Não encontrado na árvore BST!");
                    } else {
                        System.out.println("Encontrado na árvore BST: " + nomeSerieBST);
                    }
                    break;

                case 5: // FUNCIONANDO (REMOVER DA AVL E DA BST)
                    System.out.print("Informe o ID do programa a ser removido: ");
                    String id_remove = entrada.nextLine().trim().toLowerCase();

                    // Remover da AVL
                    if (arvoreAVL.remove(id_remove)) {
                        System.out.println("Programa removido da AVL");
                    } else {
                        System.out.println("Programa não encontrado na AVL, nada removido");
                    }

                    // Remover da BST
                    if (arvoreBST.remove(id_remove)) {
                        System.out.println("Programa removido da BST");
                    } else {
                        System.out.println("Programa não encontrado na BST, nada removido");
                    }
                    break;

                case 6:
                    // Exibir a altura da árvore AVL
                    int alturaAVL = arvoreAVL.height();
                    System.out.println("Altura da Árvore AVL: " + alturaAVL);

                    // Exibir a altura da árvore BST
                    int alturaBST = arvoreBST.height();
                    System.out.println("Altura da Árvore BST: " + alturaBST);
                    break;

                case 7:
                    System.out.print("Insira o nome do arquivo: ");
                    String fileName = entrada.nextLine();

                    try (PrintWriter writer = new PrintWriter(fileName)) {

                        writer.println(
                                "id,title,type,description,release_year,age_certification,runtime,genres,production_countries,seasons,imdb_id,imdb_score,imdb_votes,tmdb_popularity,tmdb_score");
                        arvoreAVL.saveTreeToCSV(writer);

                    } catch (Exception e) {
                        // tratar erro de arquivo
                        System.out.println("Erro ao salvar o arquivo " + fileName);
                    }
                    break;

                case 8:
                    System.out.println("Saindo...\n");
                    break;

                default:
                    System.out.println("Opção inválida");
            }

        } while (opcao != 8);
        entrada.close();
    }
}