Feature: Classe Buscador para ser um novo motor de busca baseado em listas invertidas

gustavo-castro_Localiza · gustavo-castro_Localiza · commit eb227b2be048 · 2025-05-26T20:28:27.000-03:00
diff --git a/tp03/src/storage/structures/ListaInvertida/Buscador.java b/tp03/src/storage/structures/ListaInvertida/Buscador.java
@@ -0,0 +1,76 @@
+package tp03.src.storage.structures.ListaInvertida;
+
+import java.util.*;
+import tp03.src.storage.structures.ListaInvertida.ElementoLista;
+import tp03.src.storage.structures.ListaInvertida.ListaInvertida;
+import tp03.src.storage.structures.ListaInvertida.ListaInvertidaUtils;
+
+public class Buscador {
+
+    private ListaInvertida listaSeries;
+    private ListaInvertida listaEpisodios;
+    private ListaInvertida listaAtores;
+
+    public Buscador(ListaInvertida listaSeries, ListaInvertida listaEpisodios, ListaInvertida listaAtores) {
+        this.listaSeries = listaSeries;
+        this.listaEpisodios = listaEpisodios;
+        this.listaAtores = listaAtores;
+    }
+
+    public List<Integer> buscarSeries(String termos) throws Exception {
+        return buscar(termos, listaSeries);
+    }
+
+    public List<Integer> buscarEpisodios(String termos) throws Exception {
+        return buscar(termos, listaEpisodios);
+    }
+
+    public List<Integer> buscarAtores(String termos) throws Exception {
+        return buscar(termos, listaAtores);
+    }
+
+    private List<Integer> buscar(String termos, ListaInvertida lista) throws Exception {
+        // Normalizar e processar os termos
+        ListaInvertidaUtils utils = new ListaInvertidaUtils();
+        String[] termosProcessados = utils.extractTerms(termos);
+        Map<Integer, Float> relevancias = new HashMap<>();
+
+        for (String termo : termosProcessados) {
+            ElementoLista[] elementos = lista.read(termo);
+            if (elementos != null) {
+                for (ElementoLista elemento : elementos) {
+                    int id = elemento.getId();
+                    float tf = elemento.getFrequencia();
+                    float idf = calcularIDF(lista, termo);
+                    float tfidf = tf * idf;
+
+                    relevancias.put(id, relevancias.getOrDefault(id, 0f) + tfidf);
+                }
+            }
+        }
+
+        // Ordenar os IDs por relevância
+        List<Map.Entry<Integer, Float>> listaOrdenada = new ArrayList<>(relevancias.entrySet());
+        listaOrdenada.sort((a, b) -> Float.compare(b.getValue(), a.getValue()));
+
+        // Retornar apenas os IDs ordenados
+        List<Integer> idsOrdenados = new ArrayList<>();
+        for (Map.Entry<Integer, Float> entry : listaOrdenada) {
+            idsOrdenados.add(entry.getKey());
+        }
+
+        return idsOrdenados;
+    }
+
+    private float calcularIDF(ListaInvertida lista, String termo) throws Exception {
+        int totalDocumentos = lista.numeroEntidades();
+        ElementoLista[] elementos = lista.read(termo);
+        int documentosComTermo = (elementos != null) ? elementos.length : 0;
+
+        if (documentosComTermo == 0) {
+            return 0;
+        }
+
+        return (float) Math.log((double) totalDocumentos / documentosComTermo);
+    }
+}
diff --git a/tp03/src/storage/structures/ListaInvertida/ListaInvertidaUtils.java b/tp03/src/storage/structures/ListaInvertida/ListaInvertidaUtils.java
@@ -46,7 +46,7 @@ private static String normalize(String str) {
 
   // Remove stop words de um array de termos e retorna um novo array sem as stop
   // words
-  private String[] extractTerms(String texto) {
+  public String[] extractTerms(String texto) {
     texto = texto.toLowerCase();
     String[] termosSeparados = texto.split(" ");
     ArrayList<String> result = new ArrayList<>();
