count unique substrings

 

public class CountUniqueSubStrings {

    static class Node {

        Node children[] = new Node[26];

        boolean eow = false;

        Node() {

            for (int i = 0; i < 26; i++) {

                children[i] = null;

            }

        }

    }

    public static Node root = new Node();

    public static void insert(String word) {

        Node curr = root;

        for (int i = 0; i < word.length(); i++) {

            int idx = word.charAt(i) - 'a';

            if (curr.children[idx] == null) {

                curr.children[idx] = new Node();

            }

            curr = curr.children[idx];

        }

        curr.eow = true;

    }

    public static int findCount(Node root) {

        int count = 0;

        if (root == null) {

            return 0;

        }

        for (int i = 0; i < 26; i++) {

            if (root.children[i] != null) {

                count += findCount(root.children[i]);

            }

        }

        return count + 1;

    }

    public static void main(String[] args) {

        String str = "ababa";

        for (int i = 0; i < str.length(); i++) {

            String suffix = str.substring(i);

            insert(suffix);

        }

        System.out.println(findCount(root));

    }

}