package stablemarriage;

import lib.*;

public class Marriages{
	// Woman i is married to the man current[i].
	public static final int NOT_ENGAGED = -1;

	int n;
	int[] husband;

	public Marriages(int n){
		this.n = n;
		husband = new int[n];
		for (int i = 0; i < husband.length; i++){
			husband[i] = NOT_ENGAGED;
		}
	}

	public int husband(int woman){
		return husband[woman];
	}
	
	public int wife(int m) {
		for(int w =0 ;w<n;w++){
			if(husband[w] == m) return w;
		}
		return NOT_ENGAGED;
	}

	public boolean isWomanAvailable(int woman){
		return husband[woman] == NOT_ENGAGED;
	}
	
	public boolean isManAvailable(int man){
		return wife(man) == NOT_ENGAGED;
	}

	public Maybe<Integer> freeMan(){
		boolean menStatus[] = new boolean[n];
		for (int man : husband) {
			if(man>=0) menStatus[man] = true;
		}
		for (int i=0;i<n;i++){
			if(!menStatus[i]) return new Some<Integer>(i);	
		}
		return new None<Integer>();
	}
		
//	public List<Integer> freeMen(){
//		ArrayList<Integer> freeMen = new ArrayList<Integer>();
//		boolean menStatus[] = new boolean[n];
//		for (int man : husband) {
//			if(man>=0) menStatus[man] = true;
//		}
//		for (int i=0;i<n;i++){
//			if(!menStatus[i]) freeMen.add(i);	
//		}
//		return freeMen;
//	}

	public void addMarriage(int woman, int man){
		husband[woman] = man;
	}

	public String toString() {
		StringBuilder sb = new StringBuilder("Married couples (woman + man): \n");
		for (int i = 0; i < husband.length; i++){
			sb.append(i + " + " + husband[i]+"\n");
		}
		return sb.toString();
	}
}