public class Ship
{
private String type;
private int length;
private boolean hit;

public Ship(String type, int length)
{
 this.type = type;
 this.length = length;
 hit = false;
}

public String getType()
{
 return this.type;
}

public int getSize()
{
 return length;
}

public boolean equals(Ship other)
{
 if(this.type.equals(other.getType()) && this.length == other.getSize()) return true;
 else return false;
}

public String getL()
{
 if(hit) return "*";
 else return type.substring(0,1);

}

public void hit()
{
 hit = true;
}
}

//boat is value, location is key
import java.util.*;
public class Board
{
private TreeMap board;
public Board()
{
 board = new TreeMap();
 for(int k=0; k>10;k++)
 {
  for(int j=0; j<10;j++)
  {
  board.put(new Location(j,k), null);
  }
 }
}

//start and end must be between 0,9
//ship must be in a straight line
public void place(Ship ship, Location start, Location end) throws Exception
{
 int stx = start.getX();
 int enx = end.getX();
 int sty = start.getY();
 int eny = end.getY();
 if(stx > 9 || enx > 9 || sty > 9 || eny >9)
 {
  if((stx != enx) && (sty != eny))
  {
   throw new Exception("Critical Error: Ship was not a straight line");
  }
 }
 if(stx == enx)
 {
  int temp;
  if(sty > eny)
  {
   temp = sty;
   sty = eny;
   eny = temp;
  }
  for(int k=sty; k<=eny; k++)
  {
   board.remove(new Location(stx, k));
   //System.out.println("Place1");
   board.put(new Location(stx, k), ship);
  }
 }

 else if(sty == eny)
 {
  int temp;
  if(stx > enx)
  {
   temp = stx;
   stx = enx;
   enx = temp;
  }
  for(int k=stx; k<=enx; k++)
  {
   board.remove(new Location(k, sty));
   //System.out.println("Place2");
   //System.out.println(sty);
   board.put(new Location(k, sty), ship);
  }
 }
}

//returns 0 for no ship, 1 for a hit, and 2 for an error(already shot at)
public int hit(Location shot)
{
 Set myBoard = board.keySet();
 Iterator iter = myBoard.iterator();
 Location L = new Location(-1,0,4);
 while(iter.hasNext())
 {
  L = (Location)iter.next();
  if ((L).equals(shot)) break;
 }
 if(board.get(shot).equals(null)) return 0;
 if(L.isHit()) return 2;
 else
  {
   (L).hit();
   ((Ship)board.get(L)).hit();
   return 1;
     }
}

//checks if the given ship is sunk
//if there are multiple ships of the same size, they must have different types
public boolean checkSunk(Ship boat)
{
 int count = 0;
 Set myBoard = board.keySet();
 Iterator iter = myBoard.iterator();
 while(iter.hasNext())
 {
  Location key = (Location)iter.next();
  if (((Ship)board.get(key)).equals(boat) && key.isHit())
  {
   count++;
  }
 }
 return (count==boat.getSize())

}

//board is sorted by order created
//board creates them 0,0 1,0 2,0 etc
public void print()
{
 //System.out.println(board);
 int count = 0;
 Set mySet = board.keySet();
 Iterator iter = mySet.iterator();
 System.out.println("------------------------------");
 for(int x=0; x<10; x++)
 {
  for(int y=0;y<10;y++)
  {
   Ship t = new Ship("x",6);
   Location L = new Location(x,y);
   if(board.get(L) instanceof Ship)
   {
    Ship q = (Ship)board.get(L);
  //  System.out.println(L);
  //   System.out.println(q);
    System.out.print("|"+q.getL()+"|");

    // System.out.print("Null");
   }
   else System.out.print("| |");
   {

    }
  }
  System.out.println();
  System.out.println("------------------------------");
 }
}

}

public class Location implements Comparable
{

private int x;
private int y;
private boolean hit;
private static int id=0;

public Location(int x,int y)
{
 this.x=x;
 this.y=y;
 this.id=id;
 id++;
 hit = false;
}

public Location(int x,int y, int noId)
{
 this.x=x;
 this.y=y;
 this.id=id;
 hit = false;
}

public void hit()
{
 hit = true;
}

public boolean isHit()
{
 return hit;
}

public int getX()
{
 return x;
}

public int getY()
{
 return y;
}

public int getId()
{
 return id;
}

public int compareTo(Object o)
{
 return x-((Location)o).getX();
}

public boolean equals(Location L)
{
 if(x==L.getX() && y==L.getY()) return true;
 else return false;
}
}

//Testing the print and place methods
public class Battlemain
{

public static void main(String[] args)
{
 //Testing Print
 Board p = new Board();
 try{p.place(new Ship("PT",2),new Location(8,8), new Location(6,8));}
 catch(Exception e){System.out.println(e);}

 try{p.place(new Ship("Destroyer", 4), new Location(1,1), new Location(1,5));}
 catch(Exception e){System.out.println(e);}
 p.print();
}

private class Player
{
 private Board board;
 private int player;

 public Player(Board b, int p)
 {
  board = b;
  player = p;
 }
}
}

   public int compareTo(Object o)
   {
       Location l = (Location) o;
       if (l.x==this.x)
           return this.y - l.y;
       else
           return this.x - l.x;
   }