ED2/Exec7_9.java at master · Thome/ED2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
package exec7_9;

/**
 *
 * @author thome pereira alves neto
 * @author matheus santos almeida
 */
import edu.princeton.cs.algs4.StdIn;
import edu.princeton.cs.algs4.TrieST;
import java.io.*;
import java.util.HashSet;
import java.util.Set;
import java.util.regex.Pattern;
import java.util.Scanner;

/**
 * alfabeto: a b … y z ? 0 1 … 8 9 (e espaço em branco)
 * ascii:   97 - 122, 63, 48 - 57, 32
 */


public class Exec7_9 {

    public static int[] a = {131,132,133,134,142,143,160,166};
    public static int[] e = {130,136,137,138,144};
    public static int[] i = {139,140,141,161,173};
    public static int[] o = {147,148,149,153,162,167};
    public static int[] u = {129,150,151,154,163};
    public static int[] c = {128,135,155};
    public static int[] y = {152,157};
    public static int[] n = {164,165};

    public static boolean contido(int[] arr, int item) {
      for (int n : arr) {
         if (item == n) {
            return true;
         }
      }
      return false;
    }

    public static String mudar(String titulo){
        StringBuilder novoTitulo = new StringBuilder();
        int aux;
        while (titulo.length() > 0) {
            aux = (int) titulo.charAt(0);
            titulo = titulo.substring(1);
            if (aux == 32 || aux > 96 && aux < 123) {
                novoTitulo.append((char)aux);
            }
            else if (aux > 64 && aux < 91) {
                aux = aux + 32;
                novoTitulo.append((char)aux);
            }
            else if (contido(a, aux))
                novoTitulo.append("a");
            else if (contido(e, aux))
                novoTitulo.append("e");
            else if (contido(i, aux))
                novoTitulo.append("i");
            else if (contido(o, aux))
                novoTitulo.append("o");
            else if (contido(u, aux))
                novoTitulo.append("u");
            else if (contido(c, aux))
                novoTitulo.append("c");
            else if (contido(y, aux))
                novoTitulo.append("y");
            else if (contido(n, aux))
                novoTitulo.append("n");
            else
                novoTitulo.append("?");

        }
        return novoTitulo.toString();
    }

    public static Set<String> withPrefix(TrieST trie, String prefix){
    	Set<String> res = new HashSet();
    	for (Object string : trie.keysWithPrefix(prefix)) {
            res.add((String)string);
	}
    	return res;
    }
    public static String longestPrefix(TrieST trie, String prefix){
    	return trie.longestPrefixOf(prefix.toLowerCase());
    }
    public static Set<String> matches(TrieST trie, String pattern){
    	Set<String> res = new HashSet();
    	for (Object string : trie.keysThatMatch(pattern)) {
            res.add((String)string);
        }
    	return res;
    }

    public static void main(String[] args) throws FileNotFoundException, IOException {
        TrieST trie = new TrieST();
        File file = new File("C:\\Users\\user\\Documents\\NetBeansProjects\\Exec7_9\\src\\exec7_9\\movies.txt");
        BufferedReader br = new BufferedReader(new FileReader(file));
        String titulo;
        String novoTitulo;
        int cont = 0;
        while ((titulo = br.readLine()) != null) {
            novoTitulo = mudar(titulo);
            trie.put(novoTitulo, cont);
            cont++;
        }

        Scanner in = new Scanner(System.in);
        while(true){
            System.out.println("Insira uma query para consulta:");
            String query = in.nextLine();

            System.out.println("(1) os títulos dos filmes que tem s como prefixo:");
            for (String s : withPrefix(trie, query)) {
                System.out.println(s);
            }

            System.out.println("(2) o mais longo prefixo de s que é um título de filme:");
            System.out.println(longestPrefix(trie, query));

            System.out.println("(3) os títulos dos filmes que casam com s quando os caracteres '.' em s são interpretados como curingas:");
            for (String s : matches(trie,query)) {
                System.out.println(s);
            }
        }


    }

}