/* vim: set expandtab tabstop=4 shiftwidth=4 : */

public class Percolation
{
    private int N;
    private boolean[] opened;
    private WeightedQuickUnionUF uf;

    // create N-by-N grid, with all sites blocked
    public Percolation(int N)
    {
        this.N = N;
        int n = (N*N)+2;
        uf = new WeightedQuickUnionUF(n);
        opened = new boolean[n];
        opened[0] = true;
        opened[n-1] = true;
    }

    // open site (row i, column j) if it is not already
    public void open(int i, int j)
    {
        if (i < 1 || j < 1 || i > N || j > N)
            throw new java.lang.IndexOutOfBoundsException("check your indexes");
        int x = (i-1)*N+j;
        if (opened[x]) return;
        opened[x] = true;
        if (j > 1 && opened[x-1]) uf.union(x, x-1);
        if (j < N && opened[x+1]) uf.union(x, x+1);
        if (i == 1) {
            uf.union(0, x);
            if (opened[x+N]) uf.union(x, x+N);
        }
        else if (i == N) {
            uf.union(N*N+1, x);
            if (opened[x-N]) uf.union(x, x-N);
        } else {
            if (opened[x-N]) uf.union(x, x-N);
            if (opened[x+N]) uf.union(x, x+N);
        }
    }

    // is site (row i, column j) open?
    public boolean isOpen(int i, int j)
    {
        if (i < 1 || j < 1 || i > N || j > N)
            throw new java.lang.IndexOutOfBoundsException("check your indexes");
        return opened[(i-1)*N+j];
    }

    // is site (row i, column j) full?
    public boolean isFull(int i, int j)
    {
        if (i < 1 || j < 1 || i > N || j > N)
            throw new java.lang.IndexOutOfBoundsException("check your indexes");
        int x = (i-1)*N+j;
        return ((opened[x]) && (uf.find(x) == 0));
    }

    // does the system percolate?
    public boolean percolates()
    {
        return (uf.find(0) == uf.find(N*N+1));
    }
}