package template;

import java.util.ArrayList;
import java.util.Map;
import java.util.HashMap;

public class TestCase {

    private boolean isSolved;
    private Object solution;
    private Map<String, Integer> intProperties;
    private Map<String, ArrayList<Integer>> intArrayProperties;
    private Map<String, ArrayList<ArrayList<Integer>>> intArray2DProperties;
    private Map<String, Double> doubleProperties;
    private Map<String, ArrayList<Double>> doubleArrayProperties;
    private Map<String, ArrayList<ArrayList<Double>>> doubleArray2DProperties;
    private Map<String, String> stringProperties;
    private Map<String, ArrayList<String>> stringArrayProperties;
    private Map<String, ArrayList<ArrayList<String>>> stringArray2DProperties;
    private Map<String, Boolean> booleanProperties;
    private Map<String, ArrayList<Boolean>> booleanArrayProperties;
    private Map<String, ArrayList<ArrayList<Boolean>>> booleanArray2DProperties;
    private Map<String, Long> longProperties;
    private Map<String, ArrayList<Long>> longArrayProperties;
    private Map<String, ArrayList<ArrayList<Long>>> longArray2DProperties;
    private int ref;
    private double time;

    public TestCase() {
        initialise();
    }

    private void initialise() {
        isSolved = false;
        intProperties = new HashMap<>();
        intArrayProperties = new HashMap<>();
        intArray2DProperties = new HashMap<>();
        doubleProperties = new HashMap<>();
        doubleArrayProperties = new HashMap<>();
        doubleArray2DProperties = new HashMap<>();
        stringProperties = new HashMap<>();
        stringArrayProperties = new HashMap<>();
        stringArray2DProperties = new HashMap<>();
        booleanProperties = new HashMap<>();
        booleanArrayProperties = new HashMap<>();
        booleanArray2DProperties = new HashMap<>();
        longProperties = new HashMap<>();
        longArrayProperties = new HashMap<>();
        longArray2DProperties = new HashMap<>();
        ref = 0;
    }

    public void setSolution(Object o) {
        solution = o;
        isSolved = true;
    }
    
    public Object getSolution() {
        if (!isSolved) {
            Utils.die("getSolution on unsolved testcase");
        }

        return solution;
    }

    public void setRef(int i) {
        ref = i;
    }
    public int getRef() {
        return ref;
    }

    public void setTime(double d) {
        time = d;
    }
    public double getTime() {
        return time;
    }
    
    public void setInteger(String s, Integer i) {
        intProperties.put(s, i);
    }
    
    public Integer getInteger(String s) {
        return intProperties.get(s);
    }

    public void setIntegerList(String s, ArrayList<Integer> l) {
        intArrayProperties.put(s, l);
    }

    public void setIntegerMatrix(String s, ArrayList<ArrayList<Integer>> l) {
        intArray2DProperties.put(s, l);
    }

    public ArrayList<Integer> getIntegerList(String s) {
        return intArrayProperties.get(s);
    }

    public Integer getIntegerListItem(String s, int i) {
        return intArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Integer>> getIntegerMatrix(String s) {
        return intArray2DProperties.get(s);
    }
    
    public ArrayList<Integer> getIntegerMatrixRow(String s, int row) {
        return intArray2DProperties.get(s).get(row);
    }
    
    public Integer getIntegerMatrixItem(String s, int row, int column) {
        return intArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Integer> getIntegerMatrixColumn(String s, int column) {
        ArrayList<Integer> out = new ArrayList();
        
        for(ArrayList<Integer> row : intArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }


    public void setDouble(String s, Double i) {
        doubleProperties.put(s, i);
    }
    
    public Double getDouble(String s) {
        return doubleProperties.get(s);
    }

    public void setDoubleList(String s, ArrayList<Double> l) {
        doubleArrayProperties.put(s, l);
    }

    public void setDoubleMatrix(String s, ArrayList<ArrayList<Double>> l) {
        doubleArray2DProperties.put(s, l);
    }

    public ArrayList<Double> getDoubleList(String s) {
        return doubleArrayProperties.get(s);
    }

    public Double getDoubleListItem(String s, int i) {
        return doubleArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Double>> getDoubleMatrix(String s) {
        return doubleArray2DProperties.get(s);
    }
    
    public ArrayList<Double> getDoubleMatrixRow(String s, int row) {
        return doubleArray2DProperties.get(s).get(row);
    }
    
    public Double getDoubleMatrixItem(String s, int row, int column) {
        return doubleArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Double> getDoubleMatrixColumn(String s, int column) {
        ArrayList<Double> out = new ArrayList();
        
        for(ArrayList<Double> row : doubleArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    
    public void setString(String s, String t) {
        stringProperties.put(s, t);
    }
    
    public String getString(String s) {
        return stringProperties.get(s);
    }

    public void setStringList(String s, ArrayList<String> l) {
        stringArrayProperties.put(s, l);
    }

    public void setStringMatrix(String s, ArrayList<ArrayList<String>> l) {
        stringArray2DProperties.put(s, l);
    }

    public ArrayList<String> getStringList(String s) {
        return stringArrayProperties.get(s);
    }

    public String getStringListItem(String s, int i) {
        return stringArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<String>> getStringMatrix(String s) {
        return stringArray2DProperties.get(s);
    }
    
    public ArrayList<String> getStringMatrixRow(String s, int row) {
        return stringArray2DProperties.get(s).get(row);
    }
    
    public String getStringMatrixItem(String s, int row, int column) {
        return stringArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<String> getStringMatrixColumn(String s, int column) {
        ArrayList<String> out = new ArrayList();
        
        for(ArrayList<String> row : stringArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    
    public void setBoolean(String s, Boolean b) {
        booleanProperties.put(s, b);
    }
    
    public Boolean getBoolean(String s) {
        return booleanProperties.get(s);
    }

    public void setBooleanList(String s, ArrayList<Boolean> l) {
        booleanArrayProperties.put(s, l);
    }

    public void setBooleanMatrix(String s, ArrayList<ArrayList<Boolean>> l) {
        booleanArray2DProperties.put(s, l);
    }

    public ArrayList<Boolean> getBooleanList(String s) {
        return booleanArrayProperties.get(s);
    }

    public Boolean getBooleanListItem(String s, int i) {
        return booleanArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Boolean>> getBooleanMatrix(String s) {
        return booleanArray2DProperties.get(s);
    }
    
    public ArrayList<Boolean> getBooleanMatrixRow(String s, int row) {
        return booleanArray2DProperties.get(s).get(row);
    }
    
    public Boolean getBooleanMatrixItem(String s, int row, int column) {
        return booleanArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Boolean> getBooleanMatrixColumn(String s, int column) {
        ArrayList<Boolean> out = new ArrayList();
        
        for(ArrayList<Boolean> row : booleanArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    public void setLong(String s, Long b) {
        longProperties.put(s, b);
    }
    
    public Long getLong(String s) {
        return longProperties.get(s);
    }

    public void setLongList(String s, ArrayList<Long> l) {
        longArrayProperties.put(s, l);
    }

    public void setLongMatrix(String s, ArrayList<ArrayList<Long>> l) {
        longArray2DProperties.put(s, l);
    }

    public ArrayList<Long> getLongList(String s) {
        return longArrayProperties.get(s);
    }

    public Long getLongListItem(String s, int i) {
        return longArrayProperties.get(s).get(i);
    }

    public ArrayList<ArrayList<Long>> getLongMatrix(String s) {
        return longArray2DProperties.get(s);
    }
    
    public ArrayList<Long> getLongMatrixRow(String s, int row) {
        return longArray2DProperties.get(s).get(row);
    }
    
    public Long getLongMatrixItem(String s, int row, int column) {
        return longArray2DProperties.get(s).get(row).get(column);
    }

    public ArrayList<Long> getLongMatrixColumn(String s, int column) {
        ArrayList<Long> out = new ArrayList();
        
        for(ArrayList<Long> row : longArray2DProperties.get(s)) {
            out.add(row.get(column));
        }
        return out;
    }

    

    


}

package problemB.dancingwithgooglers;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.Writer;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;

public class Dancers {
	
	private static final String INPUT_FILE = "/Users/sbhadour/dev/googlecodejam/test/B-small-attempt0.in";
	
	private static final String OUTPUT_FILE = "/Users/sbhadour/dev/googlecodejam/test/dancers-output.txt";
	
	class Triplet {
		
		private int googlerId;
		
		private int valueA;
		private int valueB;
		private int valueC;
		
		public Triplet(int googlerId) {
			this.googlerId = googlerId;
		}
	}

	/**
	 * Writes the results to the output file
	 * 
	 * @param results
	 */
	public void writeResults(String results) {
		try {
			Writer writer = new BufferedWriter(new FileWriter(OUTPUT_FILE));
			try {
				writer.write(results);
			}
			catch (IOException ex) {
				ex.printStackTrace();
			}
			finally {
				writer.close();
			}
		} 
		catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * Reads the input file and returns result. 
	 * 
	 * @return
	 */
	public String getResultsForInputFile() {
		StringBuilder result = new StringBuilder();
		
		try {
			BufferedReader input = new BufferedReader(new FileReader(INPUT_FILE));
			try {
				// first line is the number of test cases
				int numTestCases = Integer.parseInt(input.readLine());
				
				for (int i = 1; i <= numTestCases; i++) {
					result.append("Case #" + i + ": ");
					StringTokenizer words = new StringTokenizer(input.readLine(), " ");
					int totalGooglers = Integer.parseInt(words.nextToken());
					int surprisingTriplets = Integer.parseInt(words.nextToken());
					int minIndivScore = Integer.parseInt(words.nextToken());
					
					List<Triplet> allTriplets = new ArrayList<Dancers.Triplet>();
					for (int j = 1; j <= totalGooglers; j++) {
						List<Triplet> googlerTriplets = findAllTriplets(j,Integer.parseInt(words.nextToken()));
						allTriplets.addAll(googlerTriplets);
					}
					
					//printTriplets(allTriplets);
					result.append(findMaxGooglersWithIndivScore(allTriplets, surprisingTriplets, minIndivScore, totalGooglers));
					
					result.append(System.getProperty("line.separator"));
				}
				
			}
			catch (NumberFormatException nex) {
				nex.printStackTrace();
			} 
			catch (IOException e) {
				e.printStackTrace();
			}
			finally {
				
			}
		} 
		catch (FileNotFoundException e) {
			e.printStackTrace();
		}
		return result.toString();
	}
	
	/**
	 * Returns unique triplets assuming individual score can differ by at most 2
	 * 
	 * @param totalScore
	 * @return
	 */
	private List<Triplet> findAllTriplets (int googlerId, int totalScore) {
		List<Triplet> triplets = new ArrayList<Triplet>();
		
		// first triplet is the easiest 
		// since there are 3 judges, divide by 3 and make the result as first 2 values and 3rd as the difference of total - sum of val1+2
		int baseVal = totalScore/3;
		Triplet triplet = new Triplet(googlerId);
		triplet.valueA = baseVal;
		triplet.valueB = baseVal;
		triplet.valueC = totalScore - baseVal - baseVal;
		
		if (triplet.valueC > 10) {
			triplet.valueA += 1;
			triplet.valueC -= 1;
		}
		
		triplets.add(triplet);
		
		// a <= c always
		if (triplet.valueA == triplet.valueC && triplet.valueA != 10 && triplet.valueA != 0) {
			Triplet triplet2 = new Triplet(googlerId);
			triplet2.valueA = triplet.valueA + 1;
			triplet2.valueB = baseVal;
			triplet2.valueC = triplet.valueC - 1;
			
			triplets.add(triplet2);
		}
		// this is the case when a < c since a !> c
		else if (triplet.valueC - triplet.valueA == 2) {
			Triplet triplet3 = new Triplet(googlerId);
			triplet3.valueB = baseVal;
			triplet3.valueA = triplet.valueA + 1;
			triplet3.valueC = triplet.valueC -1;

			triplets.add(triplet3);
		}
		
		return triplets;
	}
	
	private int findMaxGooglersWithIndivScore (List<Triplet> allTriplets, int allowedSurprises, int minIndivScore, int totalGooglers) {
		
		Set<Integer> googlersWithTripletMatch = new HashSet<Integer>();
		
		List<Triplet> surprisingTripletsWithMatch = new ArrayList<Dancers.Triplet>();
		
		for (Triplet triplet : allTriplets) {
			// filter all non matched triplets
			// only add googlers with non surprising match first
			if (triplet.valueA >= minIndivScore || triplet.valueB >= minIndivScore || triplet.valueC >= minIndivScore) {
				if (Math.abs(triplet.valueA - triplet.valueC) == 2) {
					surprisingTripletsWithMatch.add(triplet);
				}
				else {
					googlersWithTripletMatch.add(triplet.googlerId);
				}
			}
		}
		
		// first drop triplets of googlers already counted.
		List<Triplet> tempList = new ArrayList<Dancers.Triplet>(surprisingTripletsWithMatch);

		for (int i = 0; i < tempList.size(); i++) {
			Triplet trip = tempList.get(i);

			if (googlersWithTripletMatch.contains(trip.googlerId)) {
				surprisingTripletsWithMatch.remove(trip);
			}
		}

		// if the surprising triplet still bigger than allowed
		if (surprisingTripletsWithMatch.size() > allowedSurprises) {
			// filter remaining  till size match restriction achieved.
			tempList = new ArrayList<Dancers.Triplet>(surprisingTripletsWithMatch);
			for (int i = 0, prevGooglerId = 0; surprisingTripletsWithMatch.size() > allowedSurprises && i < tempList.size(); i++) {
				Triplet trip = tempList.get(i);
				
				// don't remove triplets of the same  googler.
				if (prevGooglerId != trip.googlerId) {
					surprisingTripletsWithMatch.remove(trip);
				}
				prevGooglerId = trip.googlerId;
			}
		}

		// add the remaning guys back to the list
		for (Triplet matchedTrip : surprisingTripletsWithMatch) {
			googlersWithTripletMatch.add(matchedTrip.googlerId);
		}
		
		return googlersWithTripletMatch.size();
	}
	
	public void printTriplets (List<Triplet> triplets) {
		System.out.println("Triplet size: " + triplets.size());
		for (Triplet triplet : triplets) {
			System.out.println("Tripplet for googler id: " + triplet.googlerId + " -> {" + triplet.valueA +  "," + triplet.valueB + "," + triplet.valueC + "}");
		}
	}
	
	
	/**
	 * @param args
	 */
	public static void main(String[] args) {
		Dancers probB = new Dancers();
		long startTime = System.currentTimeMillis();
		String results = probB.getResultsForInputFile();
		probB.writeResults(results);
		long endTime = System.currentTimeMillis();
		
		System.out.println("Compute Time: " + (endTime - startTime) + " milliseconds");
	}

}
