Char Prefix Tree
/*
Copyright 2009 Tomer Gabel <tomer@tomergabel.com>
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
ant-intellij-tasks project (http://code.google.com/p/ant-intellij-tasks/)
$Id: CharPrefixTree.java 106 2009-09-30 02:07:29Z tomergabel $
*/
//package com.tomergabel.util;
import java.util.*;
public class CharPrefixTree implements Collection<String> {
private static class Node {
public Map<Character, Node> children = new HashMap<Character, Node>();
}
Node root = null;
@Override
public Iterator<String> iterator() {
throw new UnsupportedOperationException();
}
@Override
public Object[] toArray() {
throw new UnsupportedOperationException();
}
@Override
public <T> T[] toArray( final T[] a ) {
throw new UnsupportedOperationException();
}
@Override
public int size() {
throw new UnsupportedOperationException();
}
@Override
public boolean isEmpty() {
return this.root != null;
}
@Override
public boolean add( final String s ) {
Node node = this.root;
if ( node == null )
node = this.root = new Node();
boolean isNew = false;
for ( int i = 0; i < s.length(); ++i ) {
final char c = s.charAt( i );
if ( node.children.containsKey( c ) )
node = node.children.get( c );
else {
isNew = true;
node.children.put( c, node = new Node() );
}
}
return isNew;
}
@Override
public boolean remove( final Object o ) {
if ( o instanceof String ) {
final String s = (String) o;
Node node = this.root;
for ( int i = 0; node != null && i < s.length() - 1; ++i )
node = node.children.get( s.charAt( i ) );
return node != null;
} else throw new ClassCastException();
}
@Override
public boolean containsAll( final Collection<?> c ) {
for ( final Object o : c )
if ( !contains( o ) )
return false;
return true;
}
@Override
public boolean addAll( final Collection<? extends String> c ) {
boolean any = false;
for ( final String s : c )
any |= add( s );
return any;
}
@Override
public boolean removeAll( final Collection<?> c ) {
boolean any = false;
for ( final Object o : c )
any |= remove( o );
return any;
}
@Override
public boolean retainAll( final Collection<?> c ) {
throw new UnsupportedOperationException();
}
@Override
public void clear() {
this.root = null;
}
@Override
public boolean contains( final Object o ) {
if ( o instanceof String ) {
final String s = (String) o;
if ( s.length() == 0 )
return !isEmpty();
Node node = this.root;
for ( int i = 0; node != null && i < s.length(); ++i )
node = node.children.get( s.charAt( i ) );
return node != null;
} else throw new ClassCastException();
}
}
Related examples in the same category