package util.graph;

import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;

public class DynDjikstra {
    public static State FindShortest(Node start, Node target) {
        PriorityQueue<Node> openSet = new PriorityQueue<>();
        Set<Node> closedSet = new HashSet<>();
        openSet.add(start);
        while (!openSet.isEmpty()) {
            Node current = openSet.poll();
            for (Edge edge : current.edges) {
                Node end = edge.target;
                if (closedSet.contains(end)) {
                    continue;
                }
                State newState = edge.travel(current.last);
                //if (end.equals(target)) {
                    //return newState;
                //}
                if (openSet.contains(end)) {
                    if (end.last.compareTo(newState)>0) {
                        end.last = newState;
                    }
                } else {
                    openSet.add(end);
                    end.last = newState;
                }
            }
            closedSet.add(current);
            if (closedSet.contains(target)) {
                return target.last;
            }
        }

        return target.last;
    }
}

/**
 * Copyright Carma Systems, Inc. All rights reserved.
 */
package com.jessehutton.codejam.util;

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.PrintStream;

/**
 * @author Jesse Hutton <jesse.hutton@carmasys.com>
 */
public abstract class IOHandler {

  protected BufferedReader reader;
  protected PrintStream writer;

  public IOHandler(String inputFile) throws FileNotFoundException {
    this(new String[] {inputFile});
  }

  public IOHandler(String[] args) throws FileNotFoundException {
    reader = new BufferedReader(new FileReader(args[0]));
    if (args.length > 1 && args[1] != null) {
      writer = new PrintStream(new FileOutputStream(args[1]));
    } else {
      writer = System.out;
    }
  }

  public abstract String formatCase(int caseNum, Object result);

  public String line() throws IOException {
    return reader.readLine();
  }

  public void recordCase(int caseNum, Object result) {
    writer.println(formatCase(caseNum, result));
  }

  public void finish() {
    try {
      reader.close();
    } catch (IOException e) {
      System.err.println("closing reader failed.");
      e.printStackTrace();
    }

    writer.close();
  }
}
