ArachnidRocketgum/src/net/kezvh/two/Go.java

   1 /**
   2  *
   3  */
   4 package net.kezvh.two;
   5
   6 import java.io.FileWriter;
   7 import java.util.Collection;
   8
   9 import net.kezvh.collections.text.TrieMap;
  10 import net.kezvh.collections.text.TrieSet;
  11 import net.kezvh.io.LineIterable;
  12
  13 /**
  14  * @author afflux
  15  *
  16  */
  17 public class Go {
  18
  19     /**
  20      * @param args
  21      * @throws Throwable
  22      */
  23     public static void main(final String... args) throws Throwable {
  24         final LineIterable lineIterator = new LineIterable(args[0]);
  25
  26         final TrieMap<TrieSet> adjacentWords = new TrieMap<TrieSet>();
  27         for (final String word: lineIterator)
  28             adjacentWords.put(word, new TrieSet());
  29
  30         System.out.println("read");
  31
  32         final FileWriter fileWriter = new FileWriter(args[1]);
  33
  34         int count = 0;
  35         for (final String word: adjacentWords.keySet()) {
  36             final Collection<String> similarWords = adjacentWords.getKeysSimilarTo(word);
  37             adjacentWords.get(word).addAll(similarWords);
  38             fileWriter.write(word + ": " + similarWords + "\n");
  39             fileWriter.flush();
  40             count++;
  41             if (count % 10000 == 0)
  42                 System.out.println("wrote " + count);
  43         }
  44
  45         fileWriter.close();
  46     }
  47 }